如何在Python中用派生自联合类型的别名约束泛型类型

要实现你想要的扩展性和类型检查效果，需要调整泛型的定义方式，同时确保模式匹配覆盖所有possibleTypeForCombine中的类型，让mypy能准确检测遗漏的分支。

步骤1：正确定义绑定类型别名的TypeVar #

不要直接在函数泛型参数里写T: possibleTypeForCombine，而是用typing.TypeVar将T绑定到你的类型别名，这样mypy能更好地跟踪泛型的具体类型：

from typing import TypeVar, assert_never

possibleTypeForCombine = int | str
T = TypeVar("T", bound=possibleTypeForCombine)

步骤2：重构combine函数 #

用绑定后的T作为泛型参数，同时在模式匹配中确保每个分支的类型推断正确，最后添加assert_never分支来强制覆盖所有类型：

def combine(a: T, b: T) -> T:
    match a:
        case int():
            return a + b
        case str():
            return a + " " + b
        case _:
            # 当possibleTypeForCombine添加新类型时，mypy会在这里报错
            assert_never(a)

为什么原来的写法会报错？ #

你之前的combine2函数中，T: possibleTypeForCombine的约束只是说明T是该联合的子类型，但mypy无法确定在case int()分支中，b的类型也是int（因为T可能是整个联合类型），所以会触发操作符不兼容和返回类型不匹配的错误。而用bound=possibleTypeForCombine的TypeVar，mypy能明确知道a和b的类型是同一具体类型（int或str），从而正确推断分支中的类型。

扩展性验证 #

当你向possibleTypeForCombine添加新类型时，比如：

possibleTypeForCombine = int | str | list[str]

mypy会立刻在assert_never(a)处报错，提示你需要补充对应的模式匹配分支，完全符合你的需求。

内容的提问来源于stack exchange，提问作者Pierre-olivier Gendraud