如何使用相同的过载签名键入多个类似的Python函数

Question

我有两个非常相似的功能，第二个函数将第一个函数调用（在实际代码中，此后执行其他操作）。

两者都采用两个可以采用多种类型的参数，两者都只有某些组合有意义，因此我使用过载来表明（实际代码也将增加无效组合）。 请参阅以下示例：

from typing import overload

@overload
def x(k: int, m: float) -> int:
    ...

@overload
def x(k: str, m: bytes) -> str:
    ...

def x(k: int | str, m: float | bytes) -> int | str:
    return k


@overload
def y(k: int, m: float) -> int:
    ...

@overload
def y(k: str, m: bytes) -> str:
    ...

def y(k: int | str, m: float | bytes) -> int | str:
    return x(k, m)

现在Mypy抱怨说：

error: No overload variant of "x" matches argument type "Union[int, str]"  [call-overload]
note: Possible overload variants:
note:     def x(k: int, m: float) -> int
note:     def x(k: str, m: bytes) -> str

键入此类问题的正确方法是什么？

Answer 1

我认为您已经在Python的Overload中达到了限制，虽然

def y(k: int | str, m: float | bytes) -> int | str:

推荐写这篇文章，但这并不是真的。它太宽了，我不相信有任何方式。您将必须在y的实现签名和调用x之间的某个时刻再次缩小类型。您可以做isInstance检查，但是由于要花费一些东西，我个人只是＃type：impanore [call-overload] this。