Python中如何正确为tzinfo对象指定类型提示？

问题根源 #

Python标准库中，datetime.astimezone()返回的datetime对象的tzinfo属性被静态标注为tzinfo | None——哪怕你运行时打印类型是datetime.timezone，mypy也会严格遵循静态类型定义做检查，因此直接指定timezone类型会触发类型不兼容错误。

解决方案 #

1. 先做None检查（推荐，安全可靠） #

先验证tzinfo不为None，mypy会自动缩小类型范围：

from datetime import datetime, timezone

dt = datetime.now().astimezone()
if dt.tzinfo is not None:
    current_tz: timezone = dt.tzinfo
    # 后续代码中mypy会认可current_tz为timezone类型
else:
    # 根据业务场景处理None情况，比如抛出异常
    raise ValueError("无法获取有效时区信息")

2. 使用类型断言（适用于确定tzinfo不会为None的场景） #

通过cast明确告知mypy类型，但运行时若为None会引发潜在问题：

from datetime import datetime, timezone
from typing import cast

current_tz = cast(timezone, datetime.now().astimezone().tzinfo)

3. 用assert做类型验证 #

通过assert语句让mypy推断类型，同时运行时也能做校验：

from datetime import datetime, timezone

current_tz = datetime.now().astimezone().tzinfo
assert isinstance(current_tz, timezone), "tzinfo必须是timezone实例"
# 断言后mypy会认为current_tz是timezone类型

内容的提问来源于stack exchange，提问作者johnukfr