import datetime import calendar from django.utils.timezone import utc as timezone_utc class TimezoneNotUTCException(Exception): pass def verify_UTC(dt: datetime.datetime) -> None: if dt.tzinfo is None or dt.tzinfo.utcoffset(dt) != timezone_utc.utcoffset(dt): raise TimezoneNotUTCException("Datetime %s does not have a UTC timezone." % (dt,)) def convert_to_UTC(dt: datetime.datetime) -> datetime.datetime: if dt.tzinfo is None: return dt.replace(tzinfo=timezone_utc) return dt.astimezone(timezone_utc) def floor_to_hour(dt: datetime.datetime) -> datetime.datetime: verify_UTC(dt) return datetime.datetime(*dt.timetuple()[:4]) \ .replace(tzinfo=timezone_utc) def floor_to_day(dt: datetime.datetime) -> datetime.datetime: verify_UTC(dt) return datetime.datetime(*dt.timetuple()[:3]) \ .replace(tzinfo=timezone_utc) def ceiling_to_hour(dt: datetime.datetime) -> datetime.datetime: floor = floor_to_hour(dt) if floor == dt: return floor return floor + datetime.timedelta(hours=1) def ceiling_to_day(dt: datetime.datetime) -> datetime.datetime: floor = floor_to_day(dt) if floor == dt: return floor return floor + datetime.timedelta(days=1) def timestamp_to_datetime(timestamp: float) -> datetime.datetime: return datetime.datetime.fromtimestamp(float(timestamp), tz=timezone_utc) def datetime_to_timestamp(dt: datetime.datetime) -> int: verify_UTC(dt) return calendar.timegm(dt.timetuple())