给出以下课程:
from typing import AnyStr
class A(object):
def __init__(self, param):
# type: (AnyStr) -> None
self.a = param # type: AnyStr
我得到以下输出:
$ mypy . -v
LOG: Mypy version 0.521
LOG: Build finished in 1.199 seconds with 10 modules, 2076 types, and 2 errors
test.py:8: error: Incompatible types in assignment (expression has type "str", variable has type "AnyStr")
test.py:8: error: Incompatible types in assignment (expression has type "bytes", variable has type "AnyStr"
为什么这个赋值操作会给出不兼容的类型?
答案 0 :(得分:1)
我不是mypy的专家,但是通过一些侦探工作,我想我已经想到了这一点。
如果将AnyStr
传递给函数,这似乎工作正常,但当变量类型为AnyStr
时失败。例如,这似乎工作正常:
from typing import AnyStr
def f(a):
# type: (AnyStr) -> AnyStr
return a
if __name__ == "__main__":
print(f('cat'))
print(f(b'dog'))
但这失败了:
from typing import AnyStr
c = 3 # type: AnyStr
错误:
mypy_anystr.py:3: error: Invalid type "typing.AnyStr"
这是有道理的,因为来自the documentation的AnyStr
的想法是 str
或bytes
,但它必须在给定函数调用的范围内一致。他们为AnyStr
用法提供的示例是:
def concat(a, b):
#type: (AnyStr, AnyStr) -> AnyStr
return a + b
concat('one', 'two') # OK
concat(b'three', b'four') # OK
concat('five', b'six') # Error
当然,除非AnyStr
是全局的(并且上面的示例显示它不是),否则将变量分配到原始AnyStr
变量的范围之外(例如全局或类的属性)没有意义,这可能是它失败的原因。我怀疑错误信息可能会更清楚。
根据您实际想要完成的内容,这里有一些解决方案。如果您在str
和bytes
之间真的不可知,那么您可以使用Union[Text, bytes]
:
输入import Union,Text,AnyStr
class A:
def __init__(self, a):
#type: (AnyStr) -> None
self.param = a # type: Union[Text, bytes]
请注意,在这种情况下,我在输入中使用了AnyStr
,但在这种情况下,它等同于Union[Text, bytes]
,因为只有一个参数。或者,如果您实际做关心参数是str
还是bytes
,您可以只需AnyStr
并将其转换为您想要的版本:
from typing import Union, Text, AnyStr
from six import binary_type
class A:
def __init__(self, a):
#type: (AnyStr) -> None
if isinstance(a, binary_type):
b = a.decode() # type: Text
else:
b = a
self.param = b # type: Text
请注意,如果a
在奇怪的区域设置或其他内容中进行编码,则可能会出现问题,因此请注意,如果您尝试主动解码bytes
个对象,这是一个简化的示例和YMMV。