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
34
35
36
37
38
39
|
# -*- coding: utf-8 -*-
"""
chemspipy.utils
~~~~~~~~~~~~~~~
Miscellaneous utility functions.
"""
from __future__ import print_function
from __future__ import unicode_literals
from __future__ import division
import datetime
import functools
def memoized_property(fget):
"""Decorator to create memoized properties."""
attr_name = '_{}'.format(fget.__name__)
@functools.wraps(fget)
def fget_memoized(self):
if not hasattr(self, attr_name):
setattr(self, attr_name, fget(self))
return getattr(self, attr_name)
return property(fget_memoized)
def timestamp(ts):
"""Create a datetime object from a timestamp string."""
fmt = '%Y-%m-%dT%H:%M:%S.%f' if '.' in ts else '%Y-%m-%dT%H:%M:%S'
return datetime.datetime.strptime(ts, fmt)
def duration(ts):
"""Create a timedelta object from a duration string."""
fmt = '%H:%M:%S.%f' if '.' in ts else '%H:%M:%S'
dt = datetime.datetime.strptime(ts, fmt)
return datetime.timedelta(hours=dt.hour, minutes=dt.minute, seconds=dt.second, microseconds=dt.microsecond)
|
