mypy 错误：向抽象方法添加类型时无法使用抽象属性实例化抽象类

Question

我有一个类，我想为其添加类型提示，如下所示：

import yaml

class TestClass(dict):

    @classmethod
    def load(cls, fname) -> "TestClass":
        return cls(yaml.safe_load(""))

    @property
    @abc.abstractmethod
    def test(self):
        raise

当我在仅包含此类的模块上运行 mypy 时，我收到以下错误消息：

error: Cannot instantiate abstract class 'TestClass' with abstract attribute 'test'

根据我从其他帖子中了解到的情况，这与在执行加载方法时实例化“测试”方法有关。有没有办法单独使用 typehints 来解决这个问题，还是我需要在这里调整我的代码？

Answer 1

mypy 告诉你的是TestClass.load(...) 会失败，因为它会尝试创建抽象类TestClass 的实例。

我们可以通过要求cls 可以用任何yaml.safe_load 返回的东西来解决这个问题（我在这里假设dict），并返回一个TestClass 的实例：

from typing import Callable
import abc
import yaml

class TestClass(dict):

    @classmethod
    def load(cls: Callable[[dict], TestClass], fname) -> "TestClass":
        return cls(yaml.safe_load(""))

    @property
    @abc.abstractmethod
    def test(self):
        raise

请注意，TestClass.load("foo") 现在也将通过类型检查。这在某种程度上是公平的，因为它在运行时也很好，直到您在创建的实例上调用test()。我认为这可能是 mypy 如何实现协议的限制。