Python如何定义父类所有子类联合的类型提示，无需手动枚举？

你将返回值标注为Animal时，静态类型检查器会认为返回值可能是任意Animal类的实例，包含不属于Cat子类的情况，因此插入List[Cat]时会触发类型不兼容警告。以下是不需要手动编写全子类Union的解决方案：

方法1：使用@overload重载标注（适用于condition参数可静态推导的场景） #

如果你的condition参数是枚举、Literal字面量这类可以在静态检查阶段确定值的类型，可以用重载给每种condition对应返回类型，无需声明全量Union：

from typing import overload, Literal

@overload
def creator(condition: Literal[1]) -> Animal: ...

@overload
def creator(condition: Literal[2]) -> Cat: ...

@overload
def creator(condition: Literal[3]) -> HouseCat: ...

def creator(condition):
    if condition == 1:
        return Animal()
    elif condition == 2:
        return Cat()
    elif condition == 3:
        return HouseCat()

调用时传入固定值的condition，IDE会自动推导对应返回类型，比如传condition=2时返回值自动识别为Cat，插入List[Cat]不会触发警告。

方法2：运行时类型窄化（适用于condition运行时才能确定的场景） #

如果condition的值只有运行时才能获取，静态阶段无法推导，可以在使用返回值前加isinstance判断，类型检查器会自动窄化对象类型：

my_list: list[Cat] = []
obj = creator(...)
if isinstance(obj, Cat):
    my_list.append(obj) # 无警告，类型检查器已识别obj为Cat类型

这个方法不需要修改creator函数的标注，还能避免运行时的类型错误。

方法3：使用绑定父类的TypeVar（适用于通用返回场景） #

如果你不需要精确到具体子类，只需要告知类型检查器返回值是Animal的某个子类，可以用带bound参数的TypeVar：

from typing import TypeVar

T = TypeVar('T', bound=Animal)

def creator(condition) -> T:
    # 原有逻辑保持不变

如果使用场景需要接收明确的Cat类型，还是需要配合类型窄化或重载使用，避免误传Animal实例到Cat列表中。

注意：不要尝试用运行时遍历子类的方法自动生成Union，静态类型检查器（包括PyCharm内置的类型检查工具）不会识别运行时生成的类型标注，仅会读取代码中静态声明的标注内容，这类方法对类型提示没有作用。

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