| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- try:
- import pytz
- except ModuleNotFoundError:
- pytz = None
- try:
- import zoneinfo
- except ModuleNotFoundError:
- zoneinfo = None
- def _get_tzinfo(tzenv: str):
- """Get the tzinfo from `zoneinfo` or `pytz`
- :param tzenv: timezone in the form of Continent/City
- :return: tzinfo object or None if not found
- """
- if pytz:
- try:
- return pytz.timezone(tzenv)
- except pytz.UnknownTimeZoneError:
- pass
- else:
- try:
- return zoneinfo.ZoneInfo(tzenv)
- except ValueError as ve:
- # This is somewhat hacky, but since _validate_tzfile_path() doesn't
- # raise a specific error type, we'll need to check the message to be
- # one we know to be from that function.
- # If so, we pretend it meant that the TZ didn't exist, for the benefit
- # of `babel.localtime` catching the `LookupError` raised by
- # `_get_tzinfo_or_raise()`.
- # See https://github.com/python-babel/babel/issues/1092
- if str(ve).startswith("ZoneInfo keys "):
- return None
- except zoneinfo.ZoneInfoNotFoundError:
- pass
- return None
- def _get_tzinfo_or_raise(tzenv: str):
- tzinfo = _get_tzinfo(tzenv)
- if tzinfo is None:
- raise LookupError(
- f"Can not find timezone {tzenv}. \n"
- "Timezone names are generally in the form `Continent/City`.",
- )
- return tzinfo
- def _get_tzinfo_from_file(tzfilename: str):
- with open(tzfilename, 'rb') as tzfile:
- if pytz:
- return pytz.tzfile.build_tzinfo('local', tzfile)
- else:
- return zoneinfo.ZoneInfo.from_file(tzfile)
|
