Python typing如何根据类变量动态定义类属性类型

可以实现。注意静态类型检查器（包括mypy、pyright）不会在检查阶段执行代码逻辑，因此无法根据子类重写的field_cls类变量值自动推导field的类型，必须通过Python类型系统原生支持的泛型机制声明两者的类型关联，该方案无额外依赖，可完美通过mypy --strict检查，也能被VSCode等IDE正确识别类型。

实现步骤 #

• 定义绑定到基类A的类型变量T，约束合法的类型范围只能是A及其子类
• 将基类UseA声明为支持泛型的类，把field字段的类型标注为类型变量T
• 子类继承泛型基类时，在基类名后的方括号中传入当前子类绑定的具体字段类型，无需重复编写field的注解，类型检查器会自动完成类型推导

修正后可通过严格类型检查的完整代码 #

from dataclasses import dataclass
from typing import ClassVar, Generic, TypeVar


@dataclass
class A:
    a: str


@dataclass
class B(A):
    b: str


# 定义绑定到A的类型变量，约束泛型参数必须是A的子类
T = TypeVar("T", bound=A)


@dataclass
class UseA(Generic[T]):
    field_cls: ClassVar[type[A]] = A
    field: T  # 用类型变量标注，实际类型由子类传入的泛型参数决定


# 继承时指定泛型参数为B，此时field的类型会自动推导为B
class UseB(UseA[B]):
    field_cls = B

    def print_b(self) -> None:
        # 类型检查器可正确识别self.field为B类型，访问b属性无报错
        print(self.field.b)

效果验证 #

• 对上述代码执行mypy --strict检查，不会抛出任何类型错误
• IDE中可正确识别UseB.field为B类型，提供b属性的自动补全、类型提示
• 后续新增其他A的子类（例如C继承自A）时，仅需编写class UseC(UseA[C])即可自动绑定field类型为C，无需修改基类逻辑
• 基类可直接实例化使用：UseA[A](field=A(a="test"))，此时field类型为A，和原有逻辑完全兼容

内容的提问来源于stack exchange，提问作者Anton M.