Python：如何按值编写bool参数的tying.overload修饰符 [英] Python: How to write typing.overload decorator for bool arguments by value

问题描述

我尝试询问的示例代码如下。
Internet上的任何示例都不会尝试这样重载参数值。
其中一个参数是布尔值，我希望重载基于布尔值而不是通常的参数类型的方法。

from typing import overload, Union

@overload
def myfunc(arg:bool=True)-> str: ...

@overload
def myfunc(arg:bool=False)-> int: ...

def myfunc(arg:bool)->Union[int, str]:
    if arg: return "something"
    else: return 0

上述示例代码中的重载代码是否正确？
您能给出一个提到这种重载的示例/博客/源代码吗，因为我在Python文档和pep-484

中找不到任何东西

我发现一种可能的方法是使用latest python docs (since python v3.8)中使用的typing.Literal

from typing import overload, Union, Literal

@overload
def myfunc(arg:Literal[True])-> str: ...

@overload
def myfunc(arg:Literal[False])-> int: ...

def myfunc(arg:bool)->Union[int, str]:
    if arg: return "something"
    else: return 0

但我还不能迁移到Python3.8，因为我正在处理仍在Python3.6上的产品代码，很快最多会升级到3.7。
因此，我仍然在寻找如何在python3.6中实现这一点的答案

推荐答案

安装typing-extensions模块，该模块包含各种类型构造的官方后端口。然后，执行：

from typing import overload, Union

# typing_extensions defines Literal for Python 3.7 and earlier, but
# re-exports it from 'typing' for later versions of Python.
from typing_extensions import Literal

@overload
def myfunc(arg: Literal[True]) -> str: ...

@overload
def myfunc(arg: Literal[False]) -> int: ...

@overload
def myfunc(arg: bool) -> Union[str, int]: ...

def myfunc(arg: bool) -> Union[int, str]:
    if arg: return "something"
    else: return 0

请参阅mypy docs on literal types中的第一个示例，了解我在bool上包含第三个重载的原因。

这篇关于Python：如何按值编写bool参数的tying.overload修饰符的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！