从功能返回类型中缩小类型

Question

我的函数可能会返回两种不同类型的值，例如union [str，int]。返回值的类型可以从函数参数确定。因此，在每个通话中，我知道应该返回哪种类型。但是，Mypy抱怨说union [str，int]不能分配给更严格的类型，例如str。有没有一种方法可以允许更严格的类型分配，而无需完全关闭签名（例如＃＃type：imploore注释）？

例如，如果我具有此功能：

from typing import Union


def change_value(action: str, value: int) -> Union[str, int]
    if action == "stringify":
        return str(value)
    return value

stringified_value: str
stringified_value = change_value("stringify", 10)

Mypy会出现此错误：

Expression of type "str | int" cannot be assigned to declared type "str"

我想通过以某种方式告诉Mypy来删除Mypy错误，那么在这种情况下，在这种情况下，缩小缩小始终是正确的。

Answer 1

可以使用Overloads进行：

from typing import Literal, Union, overload

@overload
def change_value(action: Literal['stringify'], value: int) -> str: ...
@overload
def change_value(action: Literal['whatever_else'], value: int) -> int: ...

def change_value(action: str, value: int) -> Union[str, int]:
    if action == "stringify":
        return str(value)
    return value

stringified_value: str = change_value("stringify", 10)

”还有一个附加的好处：您明确地说Stringify和whywing_else是两个可能的操作，然后呼叫thats_value（'Stingify'，2）>将被类型的检查器拒绝（注意到错字？如果没有，则很难调试否则）。