Question

假设以下代码：

from typing import Union


def invert(value: Union[str, int]) -> Union[int, str]:
    if isinstance(value, str):
        return int(value)
    elif isinstance(value, int):
        return str(value)
    else:
        raise ValueError("value must be 'int' or 'str'")

很容易看出，str输入会导致int输出，反之亦然。有没有办法指定返回类型，以便它编码这种反向关系？

Answer 1

目前还没有一种自然的方法可以在Python中指定条件类型提示。

也就是说，在您的特定情况下，您可以使用overloads来表达您要做的事情：

from typing import overload, Union

# Body of overloads must be empty

@overload
def invert(value: str) -> int:
    pass

@overload
def invert(value: int) -> str:
    pass

# Implementation goes last, without an overload.
# Adding type hints here are optional -- if they
# exist, the function body is checked against the
# provided hints.
def invert(value: Union[int, str]) -> Union[int, str]:
    if isinstance(value, str):
        return int(value)
    elif isinstance(value, int):
        return str(value)
    else:
        raise ValueError("value must be 'int' or 'str'")

有没有办法在Python中指定条件类型提示？

1 个答案: