如何为泛型集合中的重写方法指定类型提示？

要解决你的问题，核心是用Python的**泛型（Generic）**来改造抽象基类，让它既能约束为Foo及其子类，又能让子类重写方法时获得精准的类型提示。下面是具体的实现步骤和示例：

1. 引入必要的类型工具 #

先从typing模块导入泛型相关的工具，这是实现类型约束的基础：

from abc import ABCMeta, abstractmethod
from typing import List, TypeVar, Generic, NoReturn

2. 定义泛型类型变量 #

声明一个绑定到Foo的类型变量，这样后续仓库类可以灵活支持Foo的任意子类：

# 定义类型变量T，约束为Foo或其子类
T = TypeVar("T", bound="Foo")

class Foo(object):
    pass  # 简单数据存储对象

3. 改造抽象基类为泛型类 #

让BaseRepository继承Generic[T]，并把所有集合项相关的类型提示替换为T，这样它就变成了一个可复用的泛型模板：

class BaseRepository(Generic[T], metaclass=ABCMeta):
    # 现在_items的类型是List[T]，T会在子类中被具体类型替换
    _items: List[T] = []

    @classmethod
    def add_item(cls, item: T) -> None:
        """便捷类方法：添加项，类型提示约束为T类型"""
        cls._items.append(item)

    @abstractmethod
    def get_item(self, item_id: int) -> T:
        """抽象方法：子类必须重写，返回T类型的项"""
        ...

4. 子类继承时指定具体类型并写类型提示 #

当你定义Foo的子类（比如Bar），并创建对应的仓库类时，只需要指定泛型参数为Bar，重写方法时就能自动获得准确的类型提示：

class Bar(Foo):
    # Foo的子类，扩展自定义属性
    def __init__(self, item_id: int, name: str):
        self.item_id = item_id
        self.name = name

class BarRepository(BaseRepository[Bar]):
    def get_item(self, item_id: int) -> Bar:
        """重写抽象方法，返回具体的Bar类型"""
        for item in self._items:
            if item.item_id == item_id:
                return item
        raise ValueError(f"Item with id {item_id} not found")

关键说明 #

• 泛型的核心价值：通过Generic[T]，我们让BaseRepository变成了一个"模板类"，子类可以通过BaseRepository[具体类型]绑定到特定实体类，所有涉及T的地方都会自动替换为具体类型。
• 静态检查支持：像mypy这类工具会自动验证子类重写方法的参数、返回值是否符合泛型约束，提前规避类型错误。
• IDE友好性：编写子类方法时，IDE会自动补全T对应具体类型的属性和方法，大幅提升开发效率。

内容的提问来源于stack exchange，提问作者Joscha Götzer