python中输入提示_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, 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, str]) -> Union[MyEnum, 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)

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

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