Python:如何按值编写bool参数的tying.overload修饰符
问题描述
我尝试询问的示例代码如下。
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上包含第三个重载的原因。
相关文章