try: import pytz except ModuleNotFoundError: pytz = None import zoneinfo 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 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)