0ca02aad1f
* refactor: accept tz name strings in ietf.utils.timezone methods * refactor: use explicitly tz-aware math for iana-changes-updates script * chore: remove unused "local_timezone_to_utc()" method helper
77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
import datetime
|
|
|
|
from typing import Union
|
|
from zoneinfo import ZoneInfo
|
|
|
|
from django.conf import settings
|
|
from django.utils import timezone
|
|
|
|
|
|
# Timezone constants - tempting to make these settings, but changing them will
|
|
# require code changes.
|
|
#
|
|
# Default time zone for deadlines / expiration dates.
|
|
DEADLINE_TZINFO = ZoneInfo('PST8PDT')
|
|
|
|
# Time zone for dates from the RPC. This value is baked into the timestamps on DocEvents
|
|
# of type="published_rfc" - see Document.pub_date() and ietf.sync.refceditor.update_docs_from_rfc_index()
|
|
# for more information about how that works.
|
|
RPC_TZINFO = ZoneInfo('PST8PDT')
|
|
|
|
|
|
def _tzinfo(tz: Union[str, datetime.tzinfo, None]):
|
|
"""Helper to convert a tz param into a tzinfo
|
|
|
|
Accepts Defaults to UTC.
|
|
"""
|
|
if tz is None:
|
|
return datetime.timezone.utc
|
|
elif isinstance(tz, datetime.tzinfo):
|
|
return tz
|
|
else:
|
|
return ZoneInfo(tz)
|
|
|
|
|
|
def make_aware(dt, tz):
|
|
"""Assign timezone to a naive datetime
|
|
|
|
Helper to deal with both pytz and zoneinfo type time zones. Can go away when pytz is removed.
|
|
"""
|
|
tzinfo = _tzinfo(tz)
|
|
if hasattr(tzinfo, 'localize'):
|
|
return tzinfo.localize(dt) # pytz-style
|
|
else:
|
|
return dt.replace(tzinfo=tzinfo) # zoneinfo- / datetime.timezone-style
|
|
|
|
|
|
def datetime_from_date(date, tz=None):
|
|
"""Get datetime at midnight on a given date"""
|
|
# accept either pytz or zoneinfo tzinfos until we get rid of pytz
|
|
return make_aware(datetime.datetime(date.year, date.month, date.day), _tzinfo(tz))
|
|
|
|
|
|
def datetime_today(tz=None):
|
|
"""Get a timezone-aware datetime representing midnight today
|
|
|
|
For use with datetime fields representing a date.
|
|
"""
|
|
return timezone.now().astimezone(_tzinfo(tz)).replace(hour=0, minute=0, second=0, microsecond=0)
|
|
|
|
|
|
def date_today(tz=None):
|
|
"""Get the date corresponding to the current moment
|
|
|
|
Note that Dates are not themselves timezone aware.
|
|
"""
|
|
return timezone.now().astimezone(_tzinfo(tz)).date()
|
|
|
|
|
|
def time_now(tz=None):
|
|
"""Get the "wall clock" time corresponding to the current moment
|
|
|
|
The value returned by this data is a Time with no tzinfo attached. (Time
|
|
objects have only limited timezone support, even if tzinfo is filled in,
|
|
and may not behave correctly when daylight savings time shifts are relevant.)
|
|
"""
|
|
return timezone.now().astimezone(_tzinfo(tz)).time()
|