如何为Python类动态添加的属性配置类型提示消除mypy报错

mypy是静态类型检查工具，检查逻辑发生在代码运行前，不会执行你在__init__中循环setattr、运行时修改__annotations__的逻辑，因此无法感知动态挂载到实例上的属性，才会抛出属性不存在的错误。运行时修改类注解的方式对mypy完全无效，因为mypy只会读取代码中静态定义的注解，不会读取运行时动态变更的注解值。

根据你对类型提示精度的要求，可以选择以下两种方案：

方案1：入参字典结构可提前定义时，获得完整类型提示 #

如果调用Foo时传入的字典结构在编码阶段是明确的（比如示例中的FOO固定包含bar: int字段），可以结合TypedDict、泛型和交叉类型实现精确的类型推断：

from typing import Any, Dict, TypeVar, Generic, cast
from typing_extensions import TypedDict

# 按实际入参结构定义TypedDict，每个不同的入参结构对应一个TypedDict
class FooInput(TypedDict):
    bar: int

FOO: FooInput = {"bar": 42}

T = TypeVar("T", bound=Dict[str, Any])
class Foo(Generic[T]):
    def __init__(self, params: T):
        for name, value in params.items():
            setattr(self, name, value)

# 用cast告诉mypy返回的实例同时具备Foo类和入参字典的结构
foo = cast("Foo[FooInput] & FooInput", Foo(FOO))
foo.bar  # mypy可正确识别bar为int类型，无报错

方案2：键名完全动态无法预知时，关闭未知属性检查 #

如果传入的字典键名完全动态，编码阶段无法确定具体字段，可以通过给类添加__getattr__方法的类型注解，让mypy允许未知属性访问：

from typing import Any, Dict

FOO = {"bar": 42}

class Foo:
    def __init__(self, params: Dict[str, Any]):
        for name, value in params.items():
            setattr(self, name, value)
    
    # 该注解会让mypy将所有未显式定义的属性访问视为合法，返回类型为Any
    def __getattr__(self, name: str) -> Any:
        # 运行时兜底逻辑，和mypy检查无关
        raise AttributeError(f"'Foo' object has no attribute '{name}'")

foo = Foo(FOO)
foo.bar  # mypy不会抛出属性不存在的错误

该方案下mypy不会校验动态属性的存在性，也不会推断属性的具体类型，适合完全动态的使用场景。

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