diff --git a/stubs/tzlocal/METADATA.toml b/stubs/tzlocal/METADATA.toml index 3f6d82f31..d57dde4a4 100644 --- a/stubs/tzlocal/METADATA.toml +++ b/stubs/tzlocal/METADATA.toml @@ -1,2 +1,2 @@ -version = "4.3" +version = "5.0" requires = ["types-pytz"] diff --git a/stubs/tzlocal/tzlocal/__init__.pyi b/stubs/tzlocal/tzlocal/__init__.pyi index 81c2351ad..2d118c02f 100644 --- a/stubs/tzlocal/tzlocal/__init__.pyi +++ b/stubs/tzlocal/tzlocal/__init__.pyi @@ -1,4 +1,7 @@ from pytz import BaseTzInfo +from tzlocal.utils import assert_tz_offset + +__all__ = ["get_localzone", "get_localzone_name", "reload_localzone", "assert_tz_offset"] def reload_localzone() -> None: ... def get_localzone() -> BaseTzInfo: ... diff --git a/stubs/tzlocal/tzlocal/utils.pyi b/stubs/tzlocal/tzlocal/utils.pyi index aedc34ac5..e56ead683 100644 --- a/stubs/tzlocal/tzlocal/utils.pyi +++ b/stubs/tzlocal/tzlocal/utils.pyi @@ -1,16 +1,4 @@ import datetime -import sys -import pytz - -if sys.version_info >= (3, 9): - import zoneinfo - - class ZoneInfoNotFoundError(pytz.UnknownTimeZoneError, zoneinfo.ZoneInfoNotFoundError): ... - -else: - class ZoneInfoNotFoundError(pytz.UnknownTimeZoneError): ... - -def get_system_offset() -> int: ... def get_tz_offset(tz: datetime.tzinfo) -> int: ... -def assert_tz_offset(tz: datetime.tzinfo) -> None: ... +def assert_tz_offset(tz: datetime.tzinfo, error: bool = True) -> None: ...