如何创建针对带约束TypeVar及其Iterator的泛型类？解决mypy报错

问题原因分析 #

你碰到的"Type variable Model is unbound"错误，本质是对TypeVar的用法理解有误：当你用TypeVar的位置参数（第一个参数是变量名，后续参数）时，这些参数必须是具体的Python类型（比如int、自定义类），用来限定类型变量的可选范围。但你这里传入了另一个TypeVar（Model）作为备选类型，这是不合法的——mypy会认为Model是一个未绑定的未确定类型，因此抛出错误。

你的核心需求是让泛型类RecordReader的返回类型可以是：

1. 受MODELS_BASE_CLASS约束的某个子类实例
2. 该子类的迭代器类型

下面是两种符合mypy规范的实现方案：

方案一：双TypeVar约束（精确匹配原始意图） #

这种方案严格遵循你最初的设计，用两个TypeVar分别定义基类约束和返回类型范围：

from typing import TypeVar, Iterator, Generic, Union
from lib import BaseClass

# 项目中已有的基类定义
MODELS_BASE_CLASS = BaseClass

# 第一步：定义带基类约束的Model类型变量
Model = TypeVar('Model', bound=MODELS_BASE_CLASS)

# 第二步：定义RecordReaderType，限定为Model或Model的迭代器
RecordReaderType = TypeVar(
    'RecordReaderType',
    bound=Union[Model, Iterator[Model]]
)

class RecordReader(Generic[RecordReaderType]):
    def func(self) -> RecordReaderType:
        # 替换为你的实际业务逻辑
        if some_condition:
            return FirstSubClass()  # 返回单个Model实例，符合约束
        else:
            return iter([FirstSubClass(), SecondSubClass()])  # 返回迭代器，符合约束

方案优势：

• 可以明确指定实例的返回类型，比如：

  # 单个实例类型的Reader
  single_reader: RecordReader[FirstSubClass] = RecordReader()
  # 迭代器类型的Reader
  iter_reader: RecordReader[Iterator[SecondSubClass]] = RecordReader()
  

• mypy会严格检查func()的返回值是否匹配你指定的类型。

方案二：简化版（更简洁，满足多数场景） #

如果你不需要精确指定返回类型是单个实例还是迭代器，只想确保返回值是某个Model子类或其迭代器，可以简化为单个TypeVar：

from typing import TypeVar, Iterator, Generic, Union
from lib import BaseClass

MODELS_BASE_CLASS = BaseClass
Model = TypeVar('Model', bound=MODELS_BASE_CLASS)

class RecordReader(Generic[Model]):
    def func(self) -> Union[Model, Iterator[Model]]:
        # 业务逻辑实现
        ...

这种方式下，当你创建RecordReader[FirstSubClass]实例时，mypy会自动推断func()的返回类型为Union[FirstSubClass, Iterator[FirstSubClass]]，同样能通过类型检查，代码更简洁。

内容的提问来源于stack exchange，提问作者Anton Vahmin