Python泛型与继承：如何为返回对应子类实例的父类方法正确标注类型

看起来你遇到的是Python类型标注里很典型的「父类方法返回子类关联类型」的场景，其实我们可以通过带绑定的泛型TypeVar加上类变量的类型注解来完美解决——既不用重写子类的方法，也能保持正确的继承关系，让类型检查器准确推断返回类型。

先看修改后的完整代码，完全满足你的需求：

from typing import TypeVar, Generic, Type

class A:
    def __init__(self, a:int) -> None:
        self.a:int = a

class SubA1(A): ...
class SubA2(A): ...

# 定义一个绑定到A的TypeVar，只能接受A或它的子类作为类型参数
Aclass = TypeVar('Aclass', bound=A)

class B(Generic[Aclass]):
    # 标注_otherClass是对应Aclass的类型对象，默认值为A
    _otherClass: Type[Aclass] = A

    def __init__(self, b:int) -> None:
        self.b:int = b

    def getOther(self) -> Aclass:
        return self._otherClass(a = self.b)

class SubB1(B[SubA1]):
    _otherClass = SubA1

class SubB2(B[SubA2]):
    _otherClass = SubA2

关键改动说明： #

1. 带绑定的TypeVar：Aclass = TypeVar('Aclass', bound=A) 限定了这个泛型参数只能是A或者它的子类，避免了传入无关类型的风险，同时让类型检查器能识别子类的继承关系。
2. 类变量的类型标注：把_otherClass标注为Type[Aclass]，明确告诉类型检查器：这个类变量是对应泛型参数的类对象，而不是实例。
3. 泛型类的默认行为：给B类的_otherClass设置默认值为A，这样当你直接实例化B(0)时，类型检查器会自动推断它是B[A]，getOther()的返回类型就是A，解决了你第一种尝试里B本身类型推断错误的问题。
4. 方法返回类型绑定泛型：getOther的返回类型直接写Aclass，这样子类继承时只需要在类定义里指定B[SubA1]，类型检查器会自动把泛型参数替换成SubA1，完全不需要重写方法。

验证期望行为 #

现在测试你想要的场景，不管是B本身还是它的子类，类型检查器都能准确推断返回类型：

b0 = B(0)
b1 = SubB1(1)
b2 = SubB2(2)

b0.getOther()  # 返回A(0)，类型检查器识别为A
b1.getOther()  # 返回SubA1(1)，类型检查器识别为SubA1
b2.getOther()  # 返回SubA2(2)，类型检查器识别为SubA2

# 继承关系也完全正常
assert isinstance(b0, B)  # 成立
assert isinstance(b1, B)  # 成立
assert isinstance(b2, B)  # 成立

为什么之前的尝试有问题？ #

• 第一种泛型方案：没有给B类的泛型参数设置默认绑定，直接实例化B时类型检查器会把泛型参数当成Any，导致返回类型推断错误。
• 第二种多重继承方案：Python的MRO机制会优先选择靠前的父类方法，而且泛型参数无法通过多重继承正确覆盖，自然会出现类型推断错误。
• 第三种分离基类方案：直接破坏了SubB1、SubB2和B的继承关系，不符合你的核心要求。

这个方案在mypy、pyright等主流类型检查器里都能正常工作，完全满足你“只在父类定义一次方法”的需求。

备注：内容来源于stack exchange，提问作者Francesco