python – 如何键入提示函数,其中返回类型取决于参数的输入类型

发布时间：2020-12-16 22:28:23 所属栏目：Python 来源：网络整理

导读：假设我有一个将Python数据类型转换为Postgres数据类型的函数,如下所示： def map_type(input): if isinstance(input,int): return MyEnum(input) elif isinstance(input,str): return MyCustomClass(str) 我可以输入提示： def map_type(input: Union[int,st

假设我有一个将Python数据类型转换为Postgres数据类型的函数,如下所示：

def map_type(input):
    if isinstance(input,int):
        return MyEnum(input)
    elif isinstance(input,str):
        return MyCustomClass(str)

我可以输入提示：

def map_type(input: Union[int,str]) -> Union[MyEnum,MyCustomClass]: ...

但是,即使它是正确的,下面的代码也无法进行类型检查：

myvar = map_type('foobar')
print(myvar.property_of_my_custom_class)

完整示例(工作代码,但类型提示中的错误)：

from typing import Union
from enum import Enum


class MyEnum(Enum):
    VALUE_1 = 1
    VALUE_2 = 2


class MyCustomClass:

    def __init__(self,value: str) -> None:
        self.value = value

    @property
    def myproperty(self) -> str:
        return 2 * self.value


def map_type(value: Union[int,MyCustomClass]:

    if isinstance(value,int):
        return MyEnum(value)
    elif isinstance(value,str):
        return MyCustomClass(value)
    raise TypeError('Invalid input type')


myvar1 = map_type(1)
print(myvar1.value,myvar1.name)

myvar2 = map_type('foobar')
print(myvar2.myproperty)

我知道我可以将映射拆分为两个函数,但目标是具有泛型类型映射函数.

我也在考虑使用类和多态,但是我如何键入提示最顶级的类方法呢？因为它们的输出类型取决于具体的实例类型.

最佳答案

这正是070??00的用途.

简而言之,您执行以下操作：

from typing import overload

# ...snip...

@overload
def map_type(value: int) -> MyEnum: ...

@overload
def map_type(value: str) -> MyCustomClass: ...

def map_type(value: Union[int,MyCustomClass]:
    if isinstance(value,str):
        return MyCustomClass(value)
    raise TypeError('Invalid input type')

现在,当你执行map_type(3)时,mypy会理解返回类型是MyEnum.

在运行时,实际运行的唯一功能是最后一个 – 前两个被完全覆盖并被忽略.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!