python - 抽象类和字典的 mypy 问题

Question

太好了，请考虑以下代码。

from abc import ABC, abstractmethod


class Interface(ABC):
    @abstractmethod
    def method(self) -> None:
        pass


class A(Interface):
    def method(self) -> None:
        pass


class B(Interface):
    def method(self) -> None:
        pass


mapping = {'A': A, 'B': B}


# does NOT pass mypy checks
def create_map(param: str) -> Interface:
    if param in mapping:
        return mapping[param]()
    else:
        raise NotImplementedError()

# passes mypy checks
def create_if(param: str) -> Interface:
    if param == 'A':
        return A()
    elif param == 'B':
        return B()
    else:
        raise NotImplementedError()

出于某种原因，create_if 通过了所有 mypy 类型检查，但 create_map 没有。这两个函数的 reveal_type 是 'def (param: builtins.str) -> test.Interface'。

我得到的错误与我试图直接实例化一个抽象类一样，这很奇怪 considering this reference for mypy .

error: Cannot instantiate abstract class 'Interface' with abstract attribute 'method'

另外，如果我现在 mapping = {'A': A} (即删除 'B': B) create_map通过。

有人可以解释一下吗？

Answer 1

显然我缺少的只是映射的类型注释。

mapping: Mapping[str, Type[Interface]] = {'A': A, 'B': B}

谢谢@jonafato

归功于 mypy 问题 1843和 3048用于显示此语法。