如何获取Python特化泛型类Z的特化参数值？

给定以下泛型类结构：

from typing import Generic, TypeVar

T = TypeVar("T")
T1 = TypeVar("T1")
T2 = TypeVar("T2")

class X(Generic[T1, T2]):
    x: T1
    y: T2

class Y(Generic[T], X[float, T]):
    pass

class Z(Y[int]):
    pass

我们需要获取最终特化类Z对应的基类X的实际类型参数（此场景下为(float, int)）。另一个场景：

class Y(Generic[T], X[T, float]):
    pass

class Z(Y[int]):
    pass

应返回(int, float)。

解决方案 #

可以通过Python的typing模块工具结合递归遍历继承链的方式实现，核心思路是回溯泛型类的特化过程，替换类型变量为实际传入的类型：

from typing import Generic, TypeVar, get_args, get_origin

# 定义泛型类（复用问题中的代码）
T = TypeVar("T")
T1 = TypeVar("T1")
T2 = TypeVar("T2")

class X(Generic[T1, T2]):
    x: T1
    y: T2

class Y(Generic[T], X[float, T]):
    pass

class Z(Y[int]):
    pass

# 第二个测试场景
class Y2(Generic[T], X[T, float]):
    pass

class Z2(Y2[int]):
    pass

def get_target_base_args(cls, target_base):
    """
    获取目标泛型基类在当前类中的实际特化参数
    :param cls: 最终特化的类
    :param target_base: 要查找的泛型基类（如X）
    :return: 目标基类的实际参数元组，未找到则返回None
    """
    # 遍历类的泛型特化基类
    for base in cls.__orig_bases__:
        base_origin = get_origin(base)
        # 当前基类就是目标基类的特化实例
        if base_origin is target_base:
            return get_args(base)
        # 当前基类是另一个泛型类，递归查找
        elif base_origin is not None and issubclass(base_origin, Generic):
            # 递归获取泛型基类中目标基类的参数
            inner_args = get_target_base_args(base_origin, target_base)
            if inner_args is not None:
                # 获取当前基类的类型参数映射（类型变量→实际类型）
                base_type_args = get_args(base)
                type_var_map = dict(zip(base_origin.__parameters__, base_type_args))
                # 替换递归结果中的类型变量为实际类型
                return tuple(type_var_map.get(arg, arg) for arg in inner_args)
    return None

# 测试第一个场景
print(get_target_base_args(Z, X))  # 输出: (float, int)
# 测试第二个场景
print(get_target_base_args(Z2, X))  # 输出: (int, float)

工作原理 #

1. __orig_bases__属性：存储了类继承的泛型特化基类（而非原始泛型模板），是实现回溯的核心依据。
2. get_origin与get_args：get_origin获取泛型特化实例对应的原始泛型类，get_args获取特化时传入的类型参数。
3. 递归替换类型变量：当遇到中间泛型类（如Y）时，先递归查找该类中目标基类的参数，再用当前类传入的实际类型替换其中的类型变量（如把Y中的T替换为int）。

注意事项 #

• 该实现依赖CPython的__orig_bases__属性，属于实现细节，但在Python标准库typing及主流类型检查器（如mypy）中均支持。
• 若继承链中存在多个相同的目标泛型基类，需额外逻辑区分目标实例，当前代码假设目标基类仅出现一次。

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