blob: cb421057d4551e83fe50d51887226b03cbf72a99 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
# -*- coding: utf-8 -*-
"""Extra classes for Construct."""
__author__ = 'Diego Elio Pettenò'
__email__ = 'flameeyes@flameeyes.eu'
__copyright__ = 'Copyright © 2018, Diego Elio Pettenò'
__license__ = 'MIT'
import datetime
import construct
class Timestamp(construct.Adapter):
"""Adapter for converting datetime object into timestamps.
Take two parameters: the subcon object to output the resulting timestamp as,
and an optional epoch offset to the UNIX Epoch.
"""
__slots__ = ["epoch"]
def __init__(self, subcon, epoch=0):
super(Timestamp, self).__init__(subcon)
self.epoch = epoch
def _encode(self, obj, context):
assert isinstance(obj, datetime.datetime)
epoch_date = datetime.datetime.utcfromtimestamp(self.epoch)
delta = obj - epoch_date
return int(delta.total_seconds())
def _decode(self, obj, context):
return datetime.datetime.utcfromtimestamp(obj + self.epoch)
|
