mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 14:03:30 +00:00
365fe0b3d5
Fixes #2665. Regenerated by tabbott with `lint --fix` after a rebase and change in parameters. Note from tabbott: In a few cases, this converts technical debt in the form of unsorted imports into different technical debt in the form of our largest files having very long, ugly import sequences at the start. I expect this change will increase pressure for us to split those files, which isn't a bad thing. Signed-off-by: Anders Kaseorg <anders@zulip.com>
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import calendar
|
|
import datetime
|
|
|
|
|
|
class TimezoneNotUTCException(Exception):
|
|
pass
|
|
|
|
def verify_UTC(dt: datetime.datetime) -> None:
|
|
if dt.tzinfo is None or dt.tzinfo.utcoffset(dt) != datetime.timezone.utc.utcoffset(dt):
|
|
raise TimezoneNotUTCException(f"Datetime {dt} does not have a UTC timezone.")
|
|
|
|
def convert_to_UTC(dt: datetime.datetime) -> datetime.datetime:
|
|
if dt.tzinfo is None:
|
|
return dt.replace(tzinfo=datetime.timezone.utc)
|
|
return dt.astimezone(datetime.timezone.utc)
|
|
|
|
def floor_to_hour(dt: datetime.datetime) -> datetime.datetime:
|
|
verify_UTC(dt)
|
|
return datetime.datetime(*dt.timetuple()[:4], tzinfo=datetime.timezone.utc)
|
|
|
|
def floor_to_day(dt: datetime.datetime) -> datetime.datetime:
|
|
verify_UTC(dt)
|
|
return datetime.datetime(*dt.timetuple()[:3], tzinfo=datetime.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=datetime.timezone.utc)
|
|
|
|
def datetime_to_timestamp(dt: datetime.datetime) -> int:
|
|
verify_UTC(dt)
|
|
return calendar.timegm(dt.timetuple())
|