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

Question

我想问的示例代码如下。
None 互联网上的示例尝试这样重载参数值。
其中一个参数是一个 bool 值，我想重载一个基于 bool 值而不是通常的参数类型的方法。

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

上面示例代码中的重载代码是否正确？
你能给出一个提到这种重载的 example/blog/source 吗，因为我在 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 上包含第三个重载。

Answer 2

为默认（关键字）参数构建正确的、@overloading 可以实现如下：

from typing import overload, Union, Literal

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

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

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

第一个重载（使用 Literal[True]）用于类型检查函数调用 myfunc(True) 和 myfunc()，而使用第二个重载（使用 Literal[False]）用于类型检查函数调用 myfunc(False).

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

Python: How to write typing.overload decorator for bool arguments by value

python

overloading

mypy

static-typing

pyright