如何正确使用带参数化泛型约束的TypeVar？

问题场景 #

现有如下泛型类定义：

from typing import Generic, TypeVar

T = TypeVar('T')

class Widget(Generic[T]):
    content: T

class Jibbit(Generic[T]):
    element: T

class ThingHolder:
    thing: Widget | Jibbit

ThingHolder的thing属性可接收Widget或Jibbit的实例，但两者都是泛型类。现在需要将ThingHolder改为参数化类型，以区分ThingHolder[Widget[int]]和ThingHolder[Jibbit[int]]这类不同的类型。

错误原因分析 #

如果直接定义如下TypeVar：

Thing = TypeVar('Thing', bound=Widget | Jibbit)

会触发类型检查错误——Widget和Jibbit是未指定类型参数的原始泛型类，而类型检查器要求约束的是具体的实例类型（即参数化后的泛型类型，如Widget[int]），而非泛型构造器本身。

正确实现方式 #

我们需要让TypeVar约束为任意参数化后的Widget或Jibbit实例类型，可以通过Any作为占位符表示任意类型参数，再将其作为bound：

from typing import Generic, TypeVar, Any

T = TypeVar('T')

class Widget(Generic[T]):
    content: T

class Jibbit(Generic[T]):
    element: T

# 约束Thing为任意参数化后的Widget或Jibbit类型
Thing = TypeVar('Thing', bound=Widget[Any] | Jibbit[Any])

class ThingHolder(Generic[Thing]):
    thing: Thing

用法示例 #

现在可以正确区分不同参数化的ThingHolder：

# 合法：thing为Widget[int]类型
holder1: ThingHolder[Widget[int]] = ThingHolder()
holder1.thing = Widget[int]()
holder1.thing.content = 123

# 合法：thing为Jibbit[str]类型
holder2: ThingHolder[Jibbit[str]] = ThingHolder()
holder2.thing = Jibbit[str]()
holder2.thing.element = "hello"

# 类型检查报错：无法将Jibbit[int]赋值给ThingHolder[Widget[int]]的thing属性
holder1.thing = Jibbit[int]()

替代方案 #

如果不需要支持Widget/Jibbit的子类，也可以直接指定TypeVar的允许类型：

Thing = TypeVar('Thing', Widget[Any], Jibbit[Any])

这种方式限定Thing只能是参数化后的Widget或Jibbit类型，灵活性稍弱，但能满足基础需求。

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