Python 3.12中Generic[T]类方法的隐式类型获取问题

一、实现MyClass[int].make_class()目标API #

当调用MyClass[int].make_class()时，传入类方法的cls参数已经是具体化的泛型类（即绑定了int的MyClass子类），可直接通过其__args__属性提取类型参数。

完整代码 #

from typing import Generic, TypeVar

T = TypeVar('T')

class MyClass(Generic[T]):
    kind: type[T]

    def __init__(self) -> None:
        # 从具体化类的__args__中获取类型并绑定到实例
        self.kind = self.__class__.__args__[0]

    @classmethod
    def make_class(cls) -> "MyClass[T]":
        # 提取当前具体化类的类型参数
        t_type = cls.__args__[0]
        print(f"I am type {t_type}!")
        return cls()

测试验证 #

>>> MyClass[int].make_class()
I am type <class 'int'>!
<__main__.MyClass[int] object at 0x...>
>>> MyClass[int].make_class().kind is int
True

说明 #

• 具体化泛型类的__args__属性会存储绑定的类型参数（单参数时为长度1的元组），多参数泛型可按索引提取对应类型。
• 类变量kind在实例化时绑定到实例，确保每个实例持有正确的类型引用。

二、尝试实现MyClass.make_class[int]()目标API #

Python 3.12暂不支持直接给类方法添加泛型索引语法（[int]），但可通过元类+包装器的hack方案模拟该行为：

完整代码 #

from typing import Generic, TypeVar

T = TypeVar('T')

class GenericMeta(type):
    def __getattr__(self, name):
        if name == 'make_class':
            # 返回支持索引调用的包装器
            class MakeClassWrapper:
                def __getitem__(self, t_type):
                    print(f"I am type {t_type}!")
                    instance = MyClass[t_type]()
                    return instance
            return MakeClassWrapper()
        return super().__getattr__(name)

class MyClass(Generic[T], metaclass=GenericMeta):
    kind: type[T]

    def __init__(self) -> None:
        self.kind = self.__class__.__args__[0]

测试验证 #

>>> MyClass.make_class[int]()
I am type int!
<__main__.MyClass[int] object at 0x...>
>>> MyClass.make_class[int]().kind is int
True

说明 #

• 利用元类的__getattr__方法，当访问MyClass.make_class时返回支持__getitem__的包装器，从而实现[int]索引语法。
• 该方案属于语法糖hack，本质是通过包装器创建并返回具体化的MyClass[int]实例。

注意事项 #

1. 泛型擦除特性：Python泛型采用擦除式实现，仅具体化泛型类（如MyClass[int]）会保留__args__等类型参数信息，普通泛型类（MyClass）本身无运行时类型参数。
2. 多参数泛型适配：如果是多参数泛型（如MyClass[T, U]），只需调整__args__的索引即可提取对应类型。
3. 类型提示兼容性：上述代码均符合Python 3.12的泛型语法规范，类型检查器（如mypy）可正常识别类型信息。

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