mirror of
https://github.com/codl/forget
synced 2024-12-15 09:58:17 +01:00
61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
from datetime import timedelta
|
|
|
|
SCALES = [
|
|
('minutes', timedelta(minutes=1)),
|
|
('hours', timedelta(hours=1)),
|
|
('days', timedelta(days=1)),
|
|
('weeks', timedelta(days=7)),
|
|
('months', timedelta(days=30)),
|
|
('years', timedelta(days=365)),
|
|
]
|
|
|
|
def decompose_interval(attrname):
|
|
scales = [scale[1] for scale in SCALES]
|
|
scales.reverse()
|
|
|
|
def decorator(cls):
|
|
scl_name = '{}_scale'.format(attrname)
|
|
sig_name = '{}_significand'.format(attrname)
|
|
|
|
@property
|
|
def scale(self):
|
|
|
|
if getattr(self, attrname) == timedelta(0):
|
|
return timedelta(minutes=1)
|
|
|
|
for m in scales:
|
|
if getattr(self, attrname) % m == timedelta(0):
|
|
return m
|
|
|
|
return timedelta(minutes=1)
|
|
|
|
@scale.setter
|
|
def scale(self, value):
|
|
if(type(value) != timedelta):
|
|
value = timedelta(seconds=float(value))
|
|
setattr(self, attrname, getattr(self, sig_name) * value)
|
|
|
|
@property
|
|
def significand(self):
|
|
return int(getattr(self, attrname) / getattr(self, scl_name))
|
|
|
|
@significand.setter
|
|
def significand(self, value):
|
|
if type(value) == str and value.strip() == '':
|
|
value = 0
|
|
|
|
try:
|
|
value = int(value)
|
|
assert value >= 0
|
|
except (ValueError, AssertionError) as e:
|
|
raise ValueError("Incorrect time interval", e)
|
|
setattr(self, attrname, value * getattr(self, scl_name))
|
|
|
|
|
|
setattr(cls, scl_name, scale)
|
|
setattr(cls, sig_name, significand)
|
|
|
|
return cls
|
|
|
|
return decorator
|