def get_prev_start_and_next_end(self, dt_start, dt_end):
start = Delorean(dt_start, 'UTC')
end = Delorean(dt_end, 'UTC')
if start.next_year(1) == end:
prev_start = start.last_year(1)
next_end = end.next_year(1)
elif start.next_month(1) == end:
prev_start = start.last_month(1)
next_end = end.next_month(1)
elif start.next_day(1) == end:
prev_start = start.last_day(1)
next_end = end.next_day(1)
else:
diff = dt_end - dt_start
prev_start = dt_start - diff
next_end = dt_end + diff
# Returning here since we already have datetimes in this case
return prev_start, next_end
return prev_start.datetime, next_end.datetime
class UtcDatetime(object):
"""Class for manage datetimes. The timezone is always internally in UTC. Its used for unify serialization
and ease complexity of manage datetimes with timezones. Always use this class for datetime management"""
def __init__(self, the_datetime, the_timezone):
"""the_datetime must be a datetime.datetime
the_timezone is a String identifing timezone: EX: 'Europe/Madrid' 'UTC' See: all_timezones"""
the_timezone = self._get_timezone_parameter(the_timezone)
self._delorean = Delorean(datetime=the_datetime,
timezone=the_timezone).shift("UTC")
self._delorean.truncate(
'second') # Truncate to second, its the precission of serialize
@staticmethod
def get_all_timezones():
"""All timezones available"""
return all_timezones
@staticmethod
def get_current_utc_datetime():
"""Always call this method to get the current datetime.
Return UrcDateTime"""
delorean = Delorean()
return UtcDatetime(delorean.datetime, "UTC")
def datetime_in_timezone(self, the_timezone):
"""Gets the UTC timezone """
the_timezone = self._get_timezone_parameter(the_timezone)
tmp_delorean = Delorean(datetime=self._delorean.datetime)
return tmp_delorean.shift(the_timezone).datetime
def advance_in_time(self, periodicity):
"""Give us the future UtcDatetime traveling in time adding periodicity to self date"""
if periodicity.period == "YEAR":
return UtcDatetime(
self._delorean.next_year(periodicity.frequency).datetime,
"UTC")
elif periodicity.period == "MONTH":
return UtcDatetime(
self._delorean.next_month(periodicity.frequency).datetime,
"UTC")
elif periodicity.period == "DAY":
return UtcDatetime(
self._delorean.next_day(periodicity.frequency).datetime, "UTC")
def back_in_time(self, periodicity):
"""Give us the past UtcDatetime traveling in time substracting periodicity to self date"""
if periodicity.period == "YEAR":
return UtcDatetime(
self._delorean.last_year(periodicity.frequency).datetime,
"UTC")
elif periodicity.period == "MONTH":
return UtcDatetime(
self._delorean.last_month(periodicity.frequency).datetime,
"UTC")
elif periodicity.period == "DAY":
return UtcDatetime(
self._delorean.last_day(periodicity.frequency).datetime, "UTC")
@property
def to_iso8601(self):
return self.datetime_utc.strftime('%Y-%m-%dT%H:%M:%S.%fZ')
@property
def datetime_utc(self):
"""datetime object in UTC"""
return self._delorean.datetime
@property
def date_utc(self):
"""date object in UTC"""
return self._delorean.date
@staticmethod
def deserialize(data):
"""deserialize model"""
if data is None:
return None
return UtcDatetime(data, "UTC")
def serialize(self):
"""serialize model
NOTE: It serialize for pymongo datetime compatibility not for a string based interface like REST"""
return self._delorean.datetime
def __eq__(self, other):
"""equal method"""
if self is other:
return True
return isinstance(other, self.__class__) \
and self._delorean == other._delorean
def __ne__(self, other):
"""not equal method"""
return not self.__eq__(other)
def __lt__(self, other):
"""< operation"""
return self._delorean.datetime < other._delorean.datetime
def __le__(self, other):
"""<= operation"""
return self._delorean.datetime <= other._delorean.datetime
def __gt__(self, other):
"""> operation"""
return self._delorean.datetime > other._delorean.datetime
def __ge__(self, other):
""">= operation"""
return self._delorean.datetime >= other._delorean.datetime
def __sub__(self, other):
"""Returns a timedelta"""
return self._delorean.datetime - other._delorean.datetime
def _get_timezone_parameter(self, the_timezone):
"""Gets a valid timezone parameter or raise"""
if the_timezone == "PST": # Very common
return "PST8PDT"
if the_timezone not in self.get_all_timezones():
raise ValueError("%s is not a valid timezone" % the_timezone)
return the_timezone
def __repr__(self):
return "%s %s" % (self._delorean.naive(), self._delorean.timezone())
class UtcDatetime(object):
"""Class for manage datetimes. The timezone is always internally in UTC. Its used for unify serialization
and ease complexity of manage datetimes with timezones. Always use this class for datetime management"""
def __init__(self, the_datetime, the_timezone):
"""the_datetime must be a datetime.datetime
the_timezone is a String identifing timezone: EX: 'Europe/Madrid' 'UTC' See: all_timezones"""
the_timezone = self._get_timezone_parameter(the_timezone)
self._delorean = Delorean(datetime=the_datetime, timezone=the_timezone).shift("UTC")
self._delorean.truncate('second') # Truncate to second, its the precission of serialize
@staticmethod
def get_all_timezones():
"""All timezones available"""
return all_timezones
@staticmethod
def get_current_utc_datetime():
"""Always call this method to get the current datetime.
Return UrcDateTime"""
delorean = Delorean()
return UtcDatetime(delorean.datetime, "UTC")
def datetime_in_timezone(self, the_timezone):
"""Gets the UTC timezone """
the_timezone = self._get_timezone_parameter(the_timezone)
tmp_delorean = Delorean(datetime=self._delorean.datetime)
return tmp_delorean.shift(the_timezone).datetime
def advance_in_time(self, periodicity):
"""Give us the future UtcDatetime traveling in time adding periodicity to self date"""
if periodicity.period == "YEAR":
return UtcDatetime(self._delorean.next_year(periodicity.frequency).datetime, "UTC")
elif periodicity.period == "MONTH":
return UtcDatetime(self._delorean.next_month(periodicity.frequency).datetime, "UTC")
elif periodicity.period == "DAY":
return UtcDatetime(self._delorean.next_day(periodicity.frequency).datetime, "UTC")
def back_in_time(self, periodicity):
"""Give us the past UtcDatetime traveling in time substracting periodicity to self date"""
if periodicity.period == "YEAR":
return UtcDatetime(self._delorean.last_year(periodicity.frequency).datetime, "UTC")
elif periodicity.period == "MONTH":
return UtcDatetime(self._delorean.last_month(periodicity.frequency).datetime, "UTC")
elif periodicity.period == "DAY":
return UtcDatetime(self._delorean.last_day(periodicity.frequency).datetime, "UTC")
@property
def to_iso8601(self):
return self.datetime_utc.strftime('%Y-%m-%dT%H:%M:%S.%fZ')
@property
def datetime_utc(self):
"""datetime object in UTC"""
return self._delorean.datetime
@property
def date_utc(self):
"""date object in UTC"""
return self._delorean.date
@staticmethod
def deserialize(data):
"""deserialize model"""
if data is None:
return None
return UtcDatetime(data, "UTC")
def serialize(self):
"""serialize model
NOTE: It serialize for pymongo datetime compatibility not for a string based interface like REST"""
return self._delorean.datetime
def __eq__(self, other):
"""equal method"""
if self is other:
return True
return isinstance(other, self.__class__) \
and self._delorean == other._delorean
def __ne__(self, other):
"""not equal method"""
return not self.__eq__(other)
def __lt__(self, other):
"""< operation"""
return self._delorean.datetime < other._delorean.datetime
def __le__(self, other):
"""<= operation"""
return self._delorean.datetime <= other._delorean.datetime
def __gt__(self, other):
"""> operation"""
return self._delorean.datetime > other._delorean.datetime
def __ge__(self, other):
""">= operation"""
return self._delorean.datetime >= other._delorean.datetime
def __sub__(self, other):
"""Returns a timedelta"""
return self._delorean.datetime - other._delorean.datetime
def _get_timezone_parameter(self, the_timezone):
"""Gets a valid timezone parameter or raise"""
if the_timezone == "PST": # Very common
return "PST8PDT"
if the_timezone not in self.get_all_timezones():
raise ValueError("%s is not a valid timezone" % the_timezone)
return the_timezone
def __repr__(self):
return "%s %s" % (self._delorean.naive(), self._delorean.timezone())
class DeloreanTests(TestCase):
def setUp(self):
date1 = datetime(2013, 1, 3, 4, 31, 14, 148546)
self.do = Delorean(datetime=date1, timezone="UTC")
def test_truncation_hour(self):
"""
Tests that truncate always works
"""
self.do.truncate('hour')
self.assertEqual(self.do.naive(), datetime(2013, 1, 3, 4, 0))
def test_truncation_second(self):
"""
Tests that truncate always works
"""
self.do.truncate('second')
self.assertEqual(self.do.naive(), datetime(2013, 1, 3, 4, 31, 14, 0))
def test_truncation_minute(self):
"""
Tests that truncate always works
"""
self.do.truncate('minute')
self.assertEqual(self.do.naive(), datetime(2013, 1, 3, 4, 31, 0, 0))
def test_truncation_day(self):
"""
Tests that truncate always works
"""
self.do.truncate('day')
self.assertEqual(self.do.naive(), datetime(2013, 1, 3, 0, 0, 0, 0))
def test_truncation_month(self):
"""
Tests that truncate always works
"""
self.do.truncate('month')
self.assertEqual(self.do.naive(), datetime(2013, 1, 1, 0, 0, 0, 0))
def test_truncation_year(self):
"""
Tests that truncate always works
"""
self.do.truncate('year')
self.assertEqual(self.do.naive(), datetime(2013, 1, 1, 0, 0, 0, 0))
def test_date(self):
self.assertEqual(self.do.date, date(2013, 1, 3))
def test_datetime(self):
self.assertEqual(self.do.naive(), datetime(2013, 1, 3, 4, 31, 14, 148546))
def test_naive(self):
dt1 = Delorean()
dt_naive = dt1.naive()
self.assertEqual(dt_naive.tzinfo, None)
def test_naive_timezone(self):
dt1 = Delorean(timezone="US/Eastern").truncate('second').naive()
dt2 = Delorean().truncate('second').naive()
self.assertEqual(dt2, dt1)
self.assertEqual(dt1.tzinfo, None)
def test_localize(self):
dt = datetime.today()
utc = timezone("UTC")
dt = localize(dt, "UTC")
self.assertEqual(dt.tzinfo, utc)
def test_failure_truncation(self):
self.assertRaises(ValueError, self.do.truncate, "century")
def test_normalize(self):
dt1 = Delorean()
dt2 = Delorean(timezone="US/Eastern")
dt1.truncate('minute')
dt2.truncate('minute')
dt_normalized = normalize(dt1.datetime, "US/Eastern")
self.assertEqual(dt2.datetime, dt_normalized)
def test_normalize_failure(self):
naive_datetime = datetime.today()
self.assertRaises(ValueError, normalize, naive_datetime, "US/Eastern")
def test_localize_failure(self):
dt1 = localize(datetime.utcnow(), "UTC")
self.assertRaises(ValueError, localize, dt1, "UTC")
def test_timezone(self):
utc = timezone('UTC')
do_timezone = Delorean().timezone()
self.assertEqual(utc, do_timezone)
def test_datetime_timezone_default(self):
do = Delorean()
do.truncate('minute')
dt1 = datetime_timezone(UTC)
self.assertEqual(dt1.replace(second=0, microsecond=0), do.datetime)
def test_datetime_timezone(self):
do = Delorean(timezone="US/Eastern")
do.truncate("minute")
dt1 = datetime_timezone(tz="US/Eastern")
self.assertEqual(dt1.replace(second=0, microsecond=0), do.datetime)
def test_parse(self):
do = parse('Thu Sep 25 10:36:28 BRST 2003')
dt1 = utc.localize(datetime(2003, 9, 25, 10, 36, 28))
self.assertEqual(do.datetime, dt1)
def test_parse_with_utc_year_fill(self):
do = parse('Thu Sep 25 10:36:28')
dt1 = utc.localize(datetime(2013, 9, 25, 10, 36, 28))
self.assertEqual(do.datetime, dt1)
def test_parse_with_timezone_year_fill(self):
do = parse('Thu Sep 25 10:36:28')
dt1 = utc.localize(datetime(2013, 9, 25, 10, 36, 28))
self.assertEqual(do.datetime, dt1)
self.assertEqual(do._tz, "UTC")
def test_move_namedday(self):
dt_next = datetime(2013, 1, 4, 4, 31, 14, 148546, tzinfo=utc)
dt_next_2 = datetime(2013, 1, 11, 4, 31, 14, 148546, tzinfo=utc)
dt_last = datetime(2012, 12, 28, 4, 31, 14, 148546, tzinfo=utc)
dt_last_2 = datetime(2012, 12, 21, 4, 31, 14, 148546, tzinfo=utc)
d_obj_next = self.do.next_friday()
d_obj_next_2 = self.do.next_friday(2)
d_obj_last = self.do.last_friday()
d_obj_last_2 = self.do.last_friday(2)
self.assertEqual(dt_next, d_obj_next.datetime)
self.assertEqual(dt_last, d_obj_last.datetime)
self.assertEqual(dt_next_2, d_obj_next_2.datetime)
self.assertEqual(dt_last_2, d_obj_last_2.datetime)
def test_move_namedday_function(self):
dt_next = datetime(2013, 1, 4, 4, 31, 14, 148546, tzinfo=utc)
dt_last = datetime(2012, 12, 28, 4, 31, 14, 148546, tzinfo=utc)
d_obj_next = move_datetime_namedday(self.do.datetime, 'next', 'friday')
d_obj_last = move_datetime_namedday(self.do.datetime, 'last', 'friday')
self.assertEqual(dt_next, d_obj_next)
self.assertEqual(dt_last, d_obj_last)
def test_move_week(self):
dt_next = datetime(2013, 1, 10, 4, 31, 14, 148546, tzinfo=utc)
dt_next_2 = datetime(2013, 1, 17, 4, 31, 14, 148546, tzinfo=utc)
dt_last = datetime(2012, 12, 27, 4, 31, 14, 148546, tzinfo=utc)
dt_last_2 = datetime(2012, 12, 20, 4, 31, 14, 148546, tzinfo=utc)
d_obj_next = self.do.next_week()
d_obj_next_2 = self.do.next_week(2)
d_obj_last = self.do.last_week()
d_obj_last_2 = self.do.last_week(2)
self.assertEqual(dt_next, d_obj_next.datetime)
self.assertEqual(dt_last, d_obj_last.datetime)
self.assertEqual(dt_next_2, d_obj_next_2.datetime)
self.assertEqual(dt_last_2, d_obj_last_2.datetime)
def test_move_week_function(self):
dt_next = datetime(2013, 1, 10, 4, 31, 14, 148546, tzinfo=utc)
dt_last = datetime(2012, 12, 27, 4, 31, 14, 148546, tzinfo=utc)
d_obj_next = move_datetime_week(self.do.datetime, 'next', 1)
d_obj_last = move_datetime_week(self.do.datetime, 'last', 1)
self.assertEqual(dt_next, d_obj_next)
self.assertEqual(dt_last, d_obj_last)
def test_move_month(self):
dt_next = datetime(2013, 2, 3, 4, 31, 14, 148546, tzinfo=utc)
dt_next_2 = datetime(2013, 3, 3, 4, 31, 14, 148546, tzinfo=utc)
dt_last = datetime(2012, 12, 3, 4, 31, 14, 148546, tzinfo=utc)
dt_last_2 = datetime(2012, 11, 3, 4, 31, 14, 148546, tzinfo=utc)
d_obj_next = self.do.next_month()
d_obj_next_2 = self.do.next_month(2)
d_obj_last = self.do.last_month()
d_obj_last_2 = self.do.last_month(2)
self.assertEqual(dt_next, d_obj_next.datetime)
self.assertEqual(dt_last, d_obj_last.datetime)
self.assertEqual(dt_next_2, d_obj_next_2.datetime)
self.assertEqual(dt_last_2, d_obj_last_2.datetime)
def test_move_month_function(self):
dt_next = datetime(2013, 2, 3, 4, 31, 14, 148546, tzinfo=utc)
dt_last = datetime(2012, 12, 3, 4, 31, 14, 148546, tzinfo=utc)
d_obj_next = move_datetime_month(self.do.datetime, 'next', 1)
d_obj_last = move_datetime_month(self.do.datetime, 'last', 1)
self.assertEqual(dt_next, d_obj_next)
self.assertEqual(dt_last, d_obj_last)
def test_move_year(self):
dt_next = datetime(2014, 1, 3, 4, 31, 14, 148546, tzinfo=utc)
dt_next_2 = datetime(2015, 1, 3, 4, 31, 14, 148546, tzinfo=utc)
dt_last = datetime(2012, 1, 3, 4, 31, 14, 148546, tzinfo=utc)
dt_last_2 = datetime(2011, 1, 3, 4, 31, 14, 148546, tzinfo=utc)
d_obj_next = self.do.next_year()
d_obj_next_2 = self.do.next_year(2)
d_obj_last = self.do.last_year()
d_obj_last_2 = self.do.last_year(2)
self.assertEqual(dt_next, d_obj_next.datetime)
self.assertEqual(dt_last, d_obj_last.datetime)
self.assertEqual(dt_next_2, d_obj_next_2.datetime)
self.assertEqual(dt_last_2, d_obj_last_2.datetime)
def test_move_year_function(self):
dt_next = datetime(2014, 1, 3, 4, 31, 14, 148546, tzinfo=utc)
dt_last = datetime(2012, 1, 3, 4, 31, 14, 148546, tzinfo=utc)
d_obj_next = move_datetime_year(self.do.datetime, 'next', 1)
d_obj_last = move_datetime_year(self.do.datetime, 'last', 1)
self.assertEqual(dt_next, d_obj_next)
self.assertEqual(dt_last, d_obj_last)
def test_range_count(self):
"""
tests the range method with count used
"""
count = list(stops(DAILY, count=5))
self.assertEqual(len(count), 5)
def test_range_with_start(self):
dates1 = []
for do in stops(DAILY, count=5, start=datetime.utcnow()):
do.truncate('minute')
dates1.append(do)
do = Delorean().truncate('minute')
dates2 = []
for x in range(5):
dates2.append(do.next_day(x))
self.assertEqual(dates1, dates2)
def test_range_with_start_and_stop(self):
dates1 = []
tomorrow = datetime.utcnow() + timedelta(days=1)
for do in stops(DAILY, start=datetime.utcnow(), until=tomorrow):
do.truncate('minute')
dates1.append(do)
do = Delorean().truncate('minute')
dates2 = []
for x in range(2):
dates2.append(do.next_day(x))
self.assertEqual(dates1, dates2)
def test_range_with_interval(self):
dates1 = []
for do in stops(DAILY, interval=2, count=3, start=datetime.utcnow()):
do.truncate('minute')
dates1.append(do)
do = Delorean().truncate('minute')
dates2 = []
for x in range(6):
if (x % 2) == 0:
dates2.append(do.next_day(x))
self.assertEqual(dates1, dates2)
def test_delorean_failure(self):
dt = datetime.utcnow()
dt = utc.localize(dt)
self.assertRaises(DeloreanInvalidDatetime, Delorean, datetime=dt)
def test_delorean_with_datetime(self):
dt = datetime.utcnow()
d = Delorean(datetime=dt, timezone=UTC)
dt = utc.localize(dt)
self.assertEqual(dt, d._dt)
self.assertEqual(UTC, d._tz)
def test_delorean_with_timezone(self):
dt = datetime.utcnow()
d = Delorean(datetime=dt, timezone=UTC)
d = d.shift("US/Eastern")
dt = utc.localize(dt)
dt = est.normalize(dt)
self.assertEqual(dt, d._dt)
self.assertEqual(est, timezone(d._tz))
def test_delorean_with_only_timezone(self):
dt = datetime.utcnow()
dt = utc.localize(dt)
dt = est.normalize(dt)
dt = dt.replace(second=0, microsecond=0)
d = Delorean(timezone="US/Eastern")
d.truncate('minute')
self.assertEqual(est, timezone(d._tz))
self.assertEqual(dt, d._dt)
def testparse_with_timezone(self):
d1 = parse("2011/01/01 00:00:00 -0700")
d2 = datetime(2011, 1, 1, 7, 0)
d2 = utc.localize(d2)
self.assertEqual(d2, d1.datetime)
self.assertEqual(utc, timezone(d1._tz))
def test_shift_failure(self):
self.assertRaises(DeloreanInvalidTimezone, self.do.shift, "US/Westerrn")
def test_epoch(self):
unix_time = self.do.epoch()
self.assertEqual(unix_time, 1357187474.148546)
def test_epoch_creation(self):
do = epoch(1357187474.148546)
self.assertEqual(self.do, do)