mypy AnyStr在简单的示例中给出了不兼容的类型

时间:2017-07-27 15:39:44

标签: python mypy

给出以下课程:

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"

为什么这个赋值操作会给出不兼容的类型?

1 个答案:

答案 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 documentationAnyStr的想法是 strbytes ,但它必须在给定函数调用的范围内一致。他们为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变量的范围之外(例如全局或类的属性)没有意义,这可能是它失败的原因。我怀疑错误信息可能会更清楚。

解决方案

根据您实际想要完成的内容,这里有一些解决方案。如果您在strbytes之间真的不可知,那么您可以使用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。