# Copyright (C) 2015 Chintalagiri Shashank
#
# This file is part of Tendril.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
This file is part of tendril
See the COPYING, README, and INSTALL files for more information
"""
import arrow.arrow
from dateutil.relativedelta import relativedelta
from .unitbase import NumericalUnitBase
from .unitbase import Percentage
from decimal import Decimal
from numbers import Number
import re
[docs]class Frequency(NumericalUnitBase):
_regex_std = re.compile(r"^(?P<numerical>[\d]+\.?[\d]*)\s?(?P<order>[mkMG]?Hz)(?P<residual>)$") # noqa
_ostrs = ['mHz', 'Hz', 'kHz', 'MHz', 'GHz']
_dostr = 'Hz'
_allow_nOr = False
def __rdiv__(self, other):
if isinstance(other, Percentage):
other = other.value
if isinstance(other, Number):
if not isinstance(other, Decimal):
other = Decimal(other)
return TimeSpan(other / self._value)
return NotImplemented
def __rtruediv__(self, other):
return self.__rdiv__(other)
[docs]class DateSpan(NumericalUnitBase):
_regex_std = re.compile(r"^(?P<numerical>[-+]?[\d]+\.?[\d]*)\s?(?P<order>(y(ea)?r?)|(d(ay)?))s?(?P<residual>)$") # noqa
_orders = [('year', lambda x: x * 365),
('yr', lambda x: x * 365),
('y', lambda x: x * 365),
('day', 1), ('d', 1)]
_rorders = [('year', 365),
('day', 1)]
_dostr = 'day'
_allow_nOr = False
_pluralize_ostr = True
_separate_unit = True
[docs]class TimeSpan(NumericalUnitBase):
_regex_std = re.compile(r"^(?P<numerical>[-+]?[\d]+\.?[\d]*)\s?(?P<order>[fpnum]?s)(?P<residual>)$") # noqa
_ostrs = ['fs', 'ps', 'ns', 'us', 'ms', 's']
_dostr = 's'
_allow_nOr = False
def __init__(self, value):
if isinstance(value, TimeDelta):
if value.years != 0 or value.months != 0:
raise ValueError(
'TimeDelta instances used to construct TimeSpan instances'
'cannot have non-zero years and months.'
)
value = value.microseconds / Decimal('1000000.0') + value.seconds + \
value.minutes * 60 + value.hours * 3600 + \
value.days * 3600 * 24
elif not isinstance(value, Number) and not isinstance(value, (str, bytes)):
raise TypeError(
"Can't construct timespan object from {0}".format(value)
)
super(TimeSpan, self).__init__(value)
def __repr__(self):
if self._value >= 60:
return repr(self.timedelta)
else:
return super(TimeSpan, self).__repr__()
@property
def timedelta(self):
seconds = int(self._value)
microseconds = int((self._value-seconds) * 1000000)
rd = relativedelta(seconds=seconds)
return TimeDelta(
years=rd.years,
months=rd.months,
days=rd.days,
hours=rd.hours,
minutes=rd.minutes,
seconds=rd.seconds,
microseconds=microseconds
)
def __rdiv__(self, other):
if isinstance(other, Percentage):
other = other.value
if isinstance(other, Number):
if not isinstance(other, Decimal):
other = Decimal(other)
return Frequency(other / self._value)
return NotImplemented
def __rtruediv__(self, other):
return self.__rdiv__(other)
[docs]class TimeStamp(arrow.arrow.Arrow):
def __sub__(self, other):
if isinstance(other, TimeDelta):
return self.clone().replace(years=-other.years,
months=-other.months,
days=-other.days,
hours=-other.hours,
minutes=-other.minutes,
seconds=-other.seconds,
microseconds=-other.microseconds)
elif isinstance(other, TimeStamp):
return TimeDelta(years=self.year-other.year,
months=self.month-other.month,
days=self.day-other.day,
hours=self.hour-other.hour,
minutes=self.minute-other.minute,
seconds=self.second-other.second,
microseconds=self.microsecond-other.microsecond)
else:
return NotImplemented
def __add__(self, other):
if isinstance(other, TimeDelta):
return self.clone().replace(years=other.years,
months=other.months,
days=other.days,
hours=other.hours,
minutes=other.minutes,
seconds=other.seconds,
microseconds=other.microseconds)
elif isinstance(other, TimeStamp):
# return self.clone().replace(years=other.year,
# months=other.month,
# days=other.day,
# hours=other.hour,
# minutes=other.minute,
# seconds=other.second,
# microseconds=other.microsecond)
raise ValueError
else:
return NotImplemented
[docs]class TimeDelta(object):
def __init__(self, years=0, months=0, days=0, hours=0, minutes=0,
seconds=0, microseconds=0):
self.years = years
self.months = months
self.days = days
self.hours = hours
self.minutes = minutes
self.seconds = seconds
self.microseconds = microseconds
def __sub__(self, other):
if isinstance(other, TimeDelta):
return TimeDelta(
years=self.years - other.years,
months=self.months - other.months,
days=self.days - other.days,
hours=self.hours - other.hours,
minutes=self.minutes - other.minutes,
seconds=self.seconds - other.seconds,
microseconds=self.microseconds - other.microseconds
)
else:
return NotImplemented
def __add__(self, other):
if isinstance(other, TimeDelta):
return TimeDelta(
years=self.years + other.years,
months=self.months + other.months,
days=self.days + other.days,
hours=self.hours + other.hours,
minutes=self.minutes + other.minutes,
seconds=self.seconds + other.seconds,
microseconds=self.microseconds + other.microseconds
)
else:
return NotImplemented
def __repr__(self):
return ':'.join(self.trim())
[docs] def trim(self):
parts = ['years', 'months', 'days', 'hours', 'minutes', 'seconds', 'microseconds']
walker = 0
value = getattr(self, parts[walker])
while not value and walker < len(parts) - 1:
walker += 1
value = getattr(self, parts[walker])
rv = {}
while walker < len(parts):
rv[parts[walker]] = getattr(self, parts[walker])
walker += 1
if not rv['microseconds']:
rv.pop('microseconds')
return rv
@property
def timespan(self):
return TimeSpan(self)
timestamp_factory = arrow.ArrowFactory(TimeStamp)
