Mypy异常：装饰器确保全局变量非空，仅带参数函数报Union-attr错误

先还原你的代码场景（基于问题描述的典型实现）：

from typing import Optional, Callable, TypeVar

T = TypeVar('T')

class Server:
    def __init__(self):
        self.ip = "127.0.0.1"

server: Optional[Server] = None

def ensure_initialized(func: Callable[..., T]) -> Callable[..., T]:
    def wrapper(*args, **kwargs) -> T:
        global server
        if server is None:
            server = Server()
        return func(*args, **kwargs)
    return wrapper

@ensure_initialized
def hello_server() -> None:
    print(server)  # Mypy未报错

@ensure_initialized
def add_server(data: str) -> str:
    return f"{data}@{server.ip}"  # Mypy报错：Item "None" of "Server | None" has no attribute "ip"

差异核心原因 #

Mypy是静态类型检查工具，它不会执行代码，只会基于变量的声明类型和静态代码结构做分析：

1. 无参数的hello_server：你只是直接使用server变量（比如打印它），而None本身是合法的Python值，可以被正常打印或传递——Mypy认为这种操作对Server | None类型的变量是安全的，因此不会报错。
2. 带参数的add_server：你尝试访问server.ip，但None对象并没有ip属性。Mypy会遍历Server | None联合类型的所有分支，发现当server为None时，这个操作会触发运行时错误，因此抛出union-attr错误。

关键问题在于：Mypy无法识别装饰器ensure_initialized的运行时逻辑——它不知道装饰器会确保server被初始化。对Mypy来说，server的类型始终是Optional[Server]，所以会严格按照这个类型做检查。

解决方法 #

要让Mypy认可装饰器的初始化保证，有几种常见方案：

• 添加断言明确类型：在使用server前加上断言，告诉Mypy此时server一定不为None：

  @ensure_initialized
  def add_server(data: str) -> str:
      assert server is not None
      return f"{data}@{server.ip}"
  

• 使用cast强制类型转换：

  from typing import cast
  
  @ensure_initialized
  def add_server(data: str) -> str:
      typed_server = cast(Server, server)
      return f"{data}@{typed_server.ip}"
  

• 用TypeGuard标记装饰器逻辑：如果装饰器是通用的初始化工具，可以通过TypeGuard让Mypy识别它对变量类型的保证（适合复杂场景）。

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