Function parse_timezone

tortoise/timezone.py:26–40

(zone: str)

def parse_timezone(zone: str) -> tzinfo:
    if zone.upper() == "UTC":
        return ZoneInfo("UTC")
    try:
        return ZoneInfo(zone)
    except ZoneInfoNotFoundError as e:
        words = zone.split("/")
        # Compatible with `pytz.timezone`:
        #   US/central -> US/Central
        #   Europe/moscow -> Europe/Moscow
        #   asia/ShangHai -> Asia/Shanghai
        styled = "/".join([i if i.isupper() else i.title() for i in words])
        if styled != zone:
            return ZoneInfo(styled)
        raise e


@functools.cache