如何向mypy说明关联类型重载函数调用的合法性？

问题根源在于：Mypy无法自动推断foo实现函数里的x和y是配对的合法类型组合（即float对应str、list对应tuple），它会默认考虑所有可能的交叉组合（比如float+tuple），而bar没有对应的重载，因此抛出类型不兼容错误。

以下是两种可行的解决方式：

方法1：添加类型检查分支（类型守卫） #

通过isinstance明确参数类型的合法组合分支，让Mypy识别每个分支内的参数类型是合法配对的：

from typing import overload, Union


@overload
def foo(x: float, y: str):
    ...

@overload
def foo(x: list, y: tuple):
    ...

def foo(x: Union[float, list], y: Union[str, tuple]):
    if isinstance(x, float) and isinstance(y, str):
        bar(x, y)
    elif isinstance(x, list) and isinstance(y, tuple):
        bar(x, y)
    else:
        # 根据重载定义，外部不会传入非法组合，这里做兜底处理
        raise TypeError("Unsupported argument combination")


@overload
def bar(x: float, y: str):
    ...

@overload
def bar(x: list, y: tuple):
    ...

def bar(x: Union[float, list], y: Union[str, tuple]):
    print(x)
    print(y)

方法2：用关联TypeVar绑定类型对 #

定义一组关联的TypeVar，约束参数只能以合法的组合出现，让Mypy直接识别类型的绑定关系：

from typing import overload, TypeVar

# 定义两个关联的TypeVar，对应合法的参数类型对
X = TypeVar("X", float, list)
Y = TypeVar("Y", str, tuple)

@overload
def foo(x: float, y: str):
    ...

@overload
def foo(x: list, y: tuple):
    ...

# 用关联TypeVar标注实现函数参数
def foo(x: X, y: Y):
    bar(x, y)


@overload
def bar(x: float, y: str):
    ...

@overload
def bar(x: list, y: tuple):
    ...

def bar(x: X, y: Y):
    print(x)
    print(y)

进阶：Python 3.12+ 用类型别名定义合法参数组合 #

如果使用Python 3.12及以上版本，可以用类型别名更清晰地定义合法的参数组合：

from typing import overload, TypeAlias

# 定义合法的参数类型对
FloatStrPair: TypeAlias = tuple[float, str]
ListTuplePair: TypeAlias = tuple[list, tuple]
ValidParams: TypeAlias = FloatStrPair | ListTuplePair

@overload
def foo(x: float, y: str):
    ...

@overload
def foo(x: list, y: tuple):
    ...

def foo(*args: ValidParams):
    bar(*args)


@overload
def bar(x: float, y: str):
    ...

@overload
def bar(x: list, y: tuple):
    ...

def bar(*args: ValidParams):
    print(args[0])
    print(args[1])

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