如何在Python中实现C++风格的跨泛型参数复用类型提示

核心思路 #

模拟C++中泛型结构体内部暴露类型别名的逻辑，在Python泛型类Foo内部预先声明关联的类型参数，再将Bar/Baz的泛型参数绑定到Foo类型，直接从传入的Foo特化类型中提取内部关联类型即可。

代码实现 #

from typing import TypeVar, Generic, get_args

# 定义Foo的泛型参数
T = TypeVar("T")
class Foo(Generic[T]):
    # 模拟C++的`using t = T`，对外暴露关联类型
    T = T

# 定义Bar的泛型参数，绑定为Foo的子类型
FooT = TypeVar("FooT", bound=Foo)
class Bar(Generic[FooT]):
    def fn(self) -> FooT.T:
        # 运行时获取实际特化的类型参数（可选，仅当运行时需要用到类型时写）
        foo_specialized_type = self.__orig_class__.__args__[0]
        actual_t_type = get_args(foo_specialized_type)[0]
        # 你的业务逻辑
        return actual_t_type()  # 示例返回，根据需求修改

# 用法完全符合预期
MyFoo = Foo[int]
MyBar = Bar[MyFoo]
x = MyBar().fn() # 类型检查器（mypy/pyright）可正确识别x为int类型

多参数泛型适配 #

如果你的Foo是多参数泛型，仅需调整Foo内部的关联类型声明即可，后续Bar/Baz无需重复声明多组泛型参数：

R = TypeVar("R")
S = TypeVar("S")
T = TypeVar("T")
class Foo(Generic[R, S, T]):
    # 暴露所有关联类型
    R = R
    S = S
    T = T

FooT = TypeVar("FooT", bound=Foo)
class Bar(Generic[FooT]):
    def fn(self) -> tuple[FooT.R, FooT.S, FooT.T]:
        # 运行时拿参数的逻辑同理，get_args返回三个参数的元组
        ...

# 特化时仅需写一次多参数，后续Bar/Baz直接复用
MyFoo = Foo[str, int, float]
MyBar = Bar[MyFoo]
res = MyBar().fn() # 类型检查器可识别res为tuple[str, int, float]

兼容说明 #

• 静态类型检查：上述写法在mypy、pyright中均可以正常推导类型，符合PEP 484规范
• 运行时支持：__orig_class__属性为Python 3.9+版本Generic类默认提供，低版本Python可安装typing_extensions并使用其中的Generic实现兼容

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