Python：如何按值编写bool参数的types.overload装饰器

Question

我想问的示例代码如下。互联网上的所有示例均未尝试重载参数值。参数之一是布尔值，我想重载基于布尔值而不是通常的参数类型的方法。

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

中找不到任何内容

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

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

但是我还不能继续使用python 3.8，因为我正在开发仍在python 3.6上的生产代码，因此很快充其量只能升级到3.7。因此，我仍在寻找如何在python 3.6中实现此目标的答案

Answer 1

安装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中包含了第三个重载。