Python3.8环境下泛型类如何获取TypeVar替换后的实际类型

问题根因 #

Python 3.8 原生typing模块的get_args方法无法直接从泛型别名（如A[Model]）或其实例中提取泛型参数，你遇到的空元组返回属于版本特性导致的问题。

解决方案 #

方案一：重写__class_getitem__存储泛型参数（无第三方依赖，推荐） #

通过重写泛型类的类下标魔法方法，将传入的实际类型绑定到生成的泛型子类属性中，兼容类方法、实例方法调用场景：

from typing import Generic, TypeVar
from pydantic import BaseModel

T = TypeVar("T", bound=BaseModel)
some_data_model = ... # 你的数据源对象

class A(Generic[T]):
    # 用于存储实际泛型参数的类属性
    _generic_type: type[BaseModel] = None

    def __class_getitem__(cls, item):
        # 调用父类逻辑生成标准泛型子类
        generic_sub_cls = super().__class_getitem__(item)
        # 存储实际传入的T类型
        generic_sub_cls._generic_type = item
        return generic_sub_cls

    # 实例方法实现
    def some_method(self) -> T:
        # 直接读取存储的泛型类型调用方法
        return self._generic_type.from_orm(some_data_model)

    # 类方法实现（按需二选一即可）
    @classmethod
    def some_classmethod(cls) -> T:
        return cls._generic_type.from_orm(some_data_model)

# 测试代码
class Model(BaseModel):
    pass
# 类方法调用
A[Model].some_classmethod()
# 实例方法调用
A[Model]().some_method()
# 断言可正常执行
assert Model.from_orm.called_once

方案二：使用兼容库typing-inspect（改动最小） #

如果你不想修改泛型类的基础逻辑，可以安装第三方兼容库处理低版本Python的泛型参数提取：

1. 安装依赖：

pip install typing-inspect

2. 修改方法实现：

from typing import Generic, TypeVar
from pydantic import BaseModel
from typing_inspect import get_args

T = TypeVar("T", bound=BaseModel)
some_data_model = ...

class A(Generic[T]):
    @classmethod
    def some_classmethod(cls) -> T:
        # 从__orig_bases__中提取泛型参数
        return get_args(cls.__orig_bases__[0])[0].from_orm(some_data_model)

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