告诉基类的方法在派生类中使用更多限制类型提示？

Question

情景

假设我有一个通用的Store类，它实现了各种方法来检索StoreObjects。为了填充存储，它定义了一个抽象方法load_object。你知道吗

然后我创建一个CarStore。我从Store派生并覆盖load_object方法以返回Car对象。你知道吗

现在的问题是如何为此添加类型提示。首先是代码：

from typing import Dict
import weakref
import abc


class StoreObject:
    pass


class Car(StoreObject):
    def __init__(self, color: str):
        self.color = color  # type: str


class Store(abc.ABC):
    def __init__(self):
        self._cache = weakref.WeakValueDictionary()  # type: weakref.WeakValueDictionary[int, StoreObject]

    def get(self, index: int) -> StoreObject:
        try:
            return self._cache[index]
        except KeyError:
            obj = self.load_object(index)
            self._cache[index] = obj
            return obj

    @abc.abstractmethod
    def load_object(self, index: int) -> StoreObject:
        raise NotImplementedError


class CarStore(Store):
    def load_object(self, index: int) -> Car:
        if index < 100:
            return Car("red")
        else:
            return Car("blue")


store = CarStore()
car = store.get(10)
print("Your car color is", car.color)

类型检查错误

问题出现在以下行中：

print("Your car color is", car.color)

这里PyCharm给出了以下警告：

Unresolved attribute reference 'color' for class 'StoreObject'

Mypy给出以下错误：

development/storetyping.py:39: error: "StoreObject" has no attribute "color"

而且PyCharm代码完成显然不包括name的store.get(10).?方法。你知道吗

问题

如何键入基类以便PyCharm和mypy可以成功地检查此代码？你知道吗

有没有办法参数化Store中的类型，以便在创建CarStore时，我可以告诉它在注释中使用Car而不是StoreObject？你知道吗

Answer 1

您的类型检查工作正常；get在CarStore中没有被重写，因此它的注释继续指定它返回StoreObject。如果要更改注释，必须在CarStore中重新定义get，例如添加：

def get(self, index: int) -> Car:
    return typing.cast(Car, super().get(index))

确保import typing可以访问cast（或者使用不合格的cast并将其添加到from typing import Dict导入中）。你知道吗

为了避免运行时性能开销，您只能基于if typing.TYPE_CHECKING:测试有条件地定义get（当静态检查器分析代码时返回True，运行代码时返回False），因此get重载实际上在运行时没有定义。你知道吗