Python中如何将__init_subclass__生成的子类属性用作类型提示

问题核心 #

你遇到的问题本质是Python静态类型检查器无法识别运行时通过__init_subclass__动态绑定的类属性，所以直接用self.a_type作为参数类型标注不会生效。

现有方案优化 #

你想到的泛型实现已经是当前Python类型系统下的标准最优解法，仅需要补充几个细节就能让类型校验更严谨：

• 给类型变量添加上界约束，确保所有传入的a_type都符合基类接口约定，类型检查器可以在基类阶段就校验方法调用的合法性
• 在基类中显式声明a_type的类属性类型，避免类型检查器报属性未定义的错误

优化后完整代码如下：

from abc import ABC
from typing import Generic, TypeVar

# 假设所有合法的a_type都继承自该基类
class A(ABC):
    def thing(self) -> None: ...

# 给类型变量添加上界约束
AT = TypeVar('AT', bound=A)

class IFoo(Generic[AT], ABC):
    # 显式声明类属性类型
    a_type: type[AT]

    def __init_subclass__(cls, a_type: type[AT], **kwargs):
        cls.a_type = a_type
        super().__init_subclass__(**kwargs)

    def foo(self, a: AT) -> None:
        # 此处类型检查器可以直接识别thing方法，无需依赖子类实现
        a.thing()

# 具体实现示例
class ConcreteA(A):
    def thing(self) -> None:
        print("执行逻辑")

class ConcreteFoo(IFoo[ConcreteA], a_type=ConcreteA):
    pass

可选替代方案 #

如果你的Python版本 >= 3.11，也可以用typing.Self实现，不过该方案的类型校验严谨度低于泛型方案：

from abc import ABC
from typing import Self

class A(ABC):
    def thing(self) -> None: ...

class IFoo(ABC):
    a_type: type[A]

    def foo(self: Self, a: Self.a_type) -> None:
        a.thing()

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