mypy检查打字。与Python 3.7支持协议

Question

我有一个需要支持Python 3.7的项目，但是我想使用 typing.protocol ，在3.8中添加。为了支持3.7，我有一个较小的后备代码，该代码仅使用 object ：

import typing

_Protocol = getattr(typing, 'Protocol', object)

class Foo(_Protocol):
    def bar(self) -> int:
        pass

这一切正如我所期望的。问题是，在运行mypy时，我会收到以下错误：

test.py:5: error: Variable "test._Protocol" is not valid as a type
test.py:5: note: See https://mypy.readthedocs.io/en/stable/common_issues.html#variables-vs-type-aliases
test.py:5: error: Invalid base class "_Protocol"

链接”变量vs类型别名该错误消息中的部分表示我应该注释 _protocol with ：typing.type [object] 无效）或使用 typing.typealias （直到Python 3.9才能使用）。

如何向Mypy表示 _protocol 有效作为一种类型？

---

我尝试的另一个解决方法是在：

if sys.version_info >= (3, 8):
    _Protocol = typing.Protocol
else:
    _Protocol = object

但是，这以相同的错误结束。

Answer 1

使用 typing_extensions 和 typing.type_checking to Import typing_extensions 仅在键入键入代码时。

import typing

if typing.TYPE_CHECKING:
    from typing_extensions import Protocol
else:
    Protocol = object


class Foo(Protocol):
    def bar(self) -> int:
        ...

---

typing_extensions python版本并使用 typing.protocol 版本＆gt; = 3.8 ：

# 3.8+
if hasattr(typing, 'Protocol'):
    Protocol = typing.Protocol
# 3.7
else:
    ...

Answer 2

Mypy似乎可以正确理解协议作为导入语句的一部分：

if sys.version_info >= (3, 8):
    from typing import Protocol as _Protocol
else:
    _Protocol = object

class Foo(_Protocol):
    def bar(self) -> int: ...

Mypy仍然会标记 foo 作为Python 3.7中的类型规范，因此您需要使用类似的解决方法来使用它：

if sys.version_info >= (3, 8):
    _Foo = Foo
else:
    _Foo = typing.Any

def bar(x: _Foo):
    pass

值得注意的是，Mypy do 不了解更多聪明的东西，例如 _foo = foo = foo如果sys.version_info＆gt; =（3，8）否则打字.yany - 您必须使用简单的格式。