假设我有一个将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)
最佳答案
这正是07000的用途.
简而言之,您执行以下操作：
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.
在运行时,实际运行的唯一功能是最后一个 – 前两个被完全覆盖并被忽略.