Python抽象基类子类类型错误：能否仅重写函数返回类型？

你遇到的问题是静态类型检查器（如Pylance）无法识别子类对基类属性的类型窄化——即便通过__post_init__做了运行时校验，静态检查阶段仍会遵循基类方法的原始类型签名。不用完全重写__iter__方法，有两种简洁的解决方案：

方案1：重载类型注解并复用基类实现 #

在Python 3.12+（或用typing_extensions.override兼容旧版本）中，你可以用@override装饰器标记方法，重新注解返回类型后直接调用super().__iter__()复用基类逻辑，无需重写迭代逻辑：

from typing import Generator, Any, override
from abc import ABC

class Base(ABC):
    data: list[str | int]
    def __iter__(self) -> Generator[str | int, Any, None]:
        yield from self.data

class Test1(Base):
    data: list[str]
    
    @override
    def __iter__(self) -> Generator[str, Any, None]:
        return super().__iter__()

这样类型检查器会认可Test1.__iter__返回的是仅包含str的生成器，调用capitalize()时就不会再提示元素可能为int的错误。

方案2：重构基类为泛型类（推荐） #

如果允许修改基类，用泛型可以从根源解决类型继承的问题，让子类指定具体的元素类型，自动推导__iter__的返回类型，完全不需要修改或重载__iter__方法：

from typing import Generator, Any, Generic, TypeVar
from abc import ABC

# 定义类型变量，限定为str或int
T = TypeVar('T', str, int)

class Base(ABC, Generic[T]):
    data: list[T]
    def __iter__(self) -> Generator[T, Any, None]:
        yield from self.data

# 子类指定类型参数为str
class Test1(Base[str]):
    data: list[str]

此时Test1的__iter__会自动继承Generator[str, Any, None]的返回类型，静态检查器能正确识别迭代元素为str，运行时和类型检查都能正常工作。

内容的提问来源于stack exchange，提问作者3dSpatialUser