在3.5中键入__exit__会在运行时失败,但会出现类型错误

时间:2018-04-21 19:50:43

标签: python python-3.x python-3.5 contextmanager mypy

__exit__的正确类型签名是什么?我有以下内容:

from types import TracebackType
from typing import Optional, Type

class Foo:
    def __enter__(self) -> 'Foo':
        return self

    def __exit__(self, exc_type: Optional[Type[BaseException]],
                 exc_value: Optional[BaseException],
                 traceback: Optional[TracebackType]) -> bool:
        return False

在最近的mypy(0.560)上,这个类型为--strict(我对此签名有一定的信心,因为我从类型的内部偷走了它)。

当使用python 3.6运行此脚本时,正如预期的那样,没有任何反应。但是当使用3.5.2运行时,我们得到一个例外:

Traceback (most recent call last):
  File "/home/student/mypy_test/test.py", line 4, in <module>
    class Foo: #(ContextManager['Foo']):
  File "/home/student/mypy_test/test.py", line 8, in Foo
    def __exit__(self, exc_type: Optional[Type[BaseException]],
  File "/usr/lib/python3.5/typing.py", line 649, in __getitem__
    return Union[arg, type(None)]
  File "/usr/lib/python3.5/typing.py", line 552, in __getitem__
    dict(self.__dict__), parameters, _root=True)
  File "/usr/lib/python3.5/typing.py", line 512, in __new__
    for t2 in all_params - {t1} if not isinstance(t2, TypeVar)):
  File "/usr/lib/python3.5/typing.py", line 512, in <genexpr>
    for t2 in all_params - {t1} if not isinstance(t2, TypeVar)):
  File "/usr/lib/python3.5/typing.py", line 1077, in __subclasscheck__
    if super().__subclasscheck__(cls):
  File "/home/student/.local/share/virtualenvs/sf_cs328-crowdsourced-QAuuIxFA/lib/python3.5/abc.py", line 225, in __subclasscheck__
    for scls in cls.__subclasses__():
TypeError: descriptor '__subclasses__' of 'type' object needs an argument

如果在异常消失之前删除参数,我们会发现问题类型是第一个:exc_type: Optional[Type[BaseException]]

注意:如果类型签名不匹配(使用mypy运行时)要让它投诉,您需要将class Foo:更改为class Foo(ContextManager['Foo'])。我没有在代码段中执行此操作,因为Python 3.5.2中的typing缺少CoroutineAwaitableContextManager等(这是LTS版本中的版本)旧的发行版)。我在这里写了一个解决方法:https://stackoverflow.com/a/49952293/568785。所以我想,完全可重复的例子是:

# Workaround for ContextManager missing in 3.5.2 typing
from typing import Any, TypeVar, TYPE_CHECKING
try:
    from typing import ContextManager
except ImportError:
    class _ContextManager:
        def __getitem__(self, index: Any) -> None:
            return type(object())

    if not TYPE_CHECKING:
        ContextManager = _ContextManager()

# The actual issue:
from types import TracebackType
from typing import Optional, Type

class Foo:
    def __enter__(self) -> 'Foo':
        return self

    def __exit__(self, exc_type: Optional[Type[BaseException]],
                 exc_value: Optional[BaseException],
                 traceback: Optional[TracebackType]) -> bool:
        return False

我已经验证,继承ContextManager在Python 3.5.2中运行仍然会产生错误(所以这个异常不是这个hack的产物,它是3.5.2运行时输入库的产物不喜欢__exit__)的签名。

据推测,这是Python 3.5的typing库中的另一个错误。有没有理智的解决方法?

1 个答案:

答案 0 :(得分:0)

我现在正在使用的丑陋解决方法是使用TYPE_CHECKING来确定我是否应该伪造该类型:

from typing import Type, TYPE_CHECKING

if TYPE_CHECKING:
    BaseExceptionType = Type[BaseException]
else:
    BaseExceptionType = bool # don't care, as long is it doesn't error

然后你可以这样做:

def __exit__(self, exc_type: Optional[BaseExceptionType],
             exc_value: Optional[BaseException],
             traceback: Optional[TracebackType]) -> bool:
    return False

验证了Python 3.5.2和mypy 0.560。

这当然打破了任何RTTI,但AFAIK RTTI是PEP的一部分,它没有降落(实验性地)直到3.6或3.7。尽管如此,这确实会打破typing.get_type_hints()