前端之家收集整理的这篇文章主要介绍了
python-Mypy：具有函数字典的不兼容类型(对象) ，
前端之家小编觉得挺不错的，现在分享给大家，也给大家做个参考。
如下代码：
from typing import Union
def a() -> int:
return 1
def b() -> str:
return 'a'
def c(arg: str = 'a') -> Union[int,str]:
return {'a': a,'b': b}[arg]()
触发以下mypy异常：
error: Incompatible return value type (got "object",expected "Union[int,str]")
一种解决方法是使用：
return a() if arg == 'a' else b()
在这种情况下,Mypy不会抱怨,但是如果有两个以上的函数,则字典语法仍然很有用.
有没有解决的办法,还是Mypy错误？
最佳答案
我认为问题在于您没有声明字典的允许类型.尽管在您的
代码中很清楚字典中只有两种类型的
输出,但是从打字的角度来看,没有什么可以阻止向其
添加另一个
函数d()的.
您可以尝试以下解决此问题的方法：
from typing import Union,Dict,Callable
output_dictionary : Dict[str,Union[Callable[[],int],Callable[[],str]]] = {'a': a,'b': b}
def c(arg: str = 'a') -> Union[int,str]:
return output_dictionary[arg]()
