有没有办法用 mypy 覆盖 python 中继承的类属性类型？

Question

我编写了一个defaultdict可以default_factory使用 key 作为参数进行调用的子类。

from collections import defaultdict
from typing import TypeVar, Any, Callable, Generic

K = TypeVar('K')


class keydefaultdict(defaultdict, Generic[K]):
    ''' Drop-in replacement for defaultdict accepting key as argument '''

    default_factory: Callable[[], Any] | Callable[[K], Any] | None

    def __missing__(self, key: K) -> Any:
        if self.default_factory is None:
            raise KeyError(key)
        else:
            try:
                ret = self[key] = self.default_factory(key)
            except TypeError:  # try no-key signature
                ret = self[key] = self.default_factory()
            # if failed, let the error propagate as usual

            return ret

mypy抱怨default_factory类型提示：

赋值中的类型不兼容（表达式的类型为“Callable[[], Any] | Callable[[K], Any] | None”，基类“defaultdict”将类型定义为“Optional[Callable[[], Any]]” ）

有什么办法可以覆盖类型吗？mypy也在这一行上抱怨self.default_factory(...)- 参数太多（或参数太少）以及实例化该字典的地方（不兼容的类型）：

data = keydefaultdict(lambda key: [key])

“keydefaultdict”的参数 1 具有不兼容的类型“Callable[[Any], list[Any]]”；预期“可选[可调用[[]，任何]]”

Answer 1

这是预期的行为，因为您所描述的内容违反了里氏替换原则。有关更多详细信息，请参阅mypy 文档。

由于这个原因，用作defaultdict子类是一个坏主意。但是，如果您确实想解决这个问题（不推荐），您可以使用# type: ignore[override]，如下所示：

default_factory: Callable[[], Any] | Callable[[K], Any] | None # type: ignore[override]

如果您需要更多信息，请在mypy 文档中对此进行更详细的描述。