乘法操作中绑定TypeVar的参数解析问题求助

问题背景 #

现有基类Base及其子类X/Y/Z，绑定Base的TypeVar，以及泛型容器Product[T]用于存储有序对象集合。当前存在以下类型推断问题：

• 直接实例化Product时类型推断正常，但Base的__mul__方法在混合子类相乘（如X()*Y()）时，MyPy无法自动将TypeVar解析为共同父类Base，报操作数类型不支持错误。
• 独立乘法函数能处理X()*Y()场景，但X()*Product[Y()]仍无法正确推断类型。
• 需求：无需为所有子类组合编写重载，让任意乘法序列都能被MyPy正确推断类型。

现有问题代码 #

基础定义 #

from __future__ import annotations
from typing import TypeVar, Generic

class Base: ...
class X(Base): ...
class Y(Base): ...
class Z(Base): ...

T = TypeVar("T", bound=Base)

class Product(Generic[T]):
    objs: list[T]
    def __init__(self, objs: list[T]):
        self.objs = list(objs)

有问题的乘法实现 #

class Base:
    def __mul__(self: T, other: T | Product[T]) -> Product[T]:
        if isinstance(other, Base):
            return Product([self, other])  # type: ignore[list-item]
        if isinstance(other, Product):
            return Product(other.objs + [self])
        return NotImplemented
    __rmul__ = __mul__

解决方案 #

核心思路是允许乘法操作的左右操作数为不同的Base子类，通过两个独立TypeVar分别约束操作数类型，利用MyPy自动将Union类型提升到共同父类的特性实现正确推断。同时可选将Product设置为协变类型，增强兼容性。

修改后的完整代码 #

from __future__ import annotations
from typing import TypeVar, Generic, Union

# 协变TypeVar，使Product[子类]可兼容Product[Base]
T_co = TypeVar("T_co", bound=Base, covariant=True)
# 约束Base实例的两个独立TypeVar
T = TypeVar("T", bound=Base)
U = TypeVar("U", bound=Base)

class Base:
    def __mul__(self: T, other: U | Product[U]) -> Product[Union[T, U]]:
        if isinstance(other, Base):
            return Product([self, other])
        if isinstance(other, Product):
            return Product(other.objs + [self])
        return NotImplemented

    def __rmul__(self: T, other: U | Product[U]) -> Product[Union[T, U]]:
        # 右乘时调整元素顺序
        if isinstance(other, Base):
            return Product([other, self])
        if isinstance(other, Product):
            return Product(other.objs + [self])
        return NotImplemented

class Product(Generic[T_co]):
    objs: list[T_co]
    def __init__(self, objs: list[T_co]):
        self.objs = list(objs)

class X(Base): ...
class Y(Base): ...
class Z(Base): ...

关键改进说明 #

1. 双TypeVar约束：用T和U分别约束乘法左右操作数，打破原代码中操作数必须为同一类型的限制。
2. Union返回类型：返回Product[Union[T, U]]，MyPy会自动将Union[X, Y]这类子类组合提升到共同父类Base，生成Product[Base]类型。
3. 协变Product：将Product的TypeVar设为协变（covariant=True），使Product[X]可以安全赋值给Product[Base]，增强类型兼容性。
4. 独立实现__rmul__：调整右乘逻辑，确保元素顺序正确，同时保持类型推断一致性。

验证场景 #

以下代码在MyPy中均可正确推断类型，无报错：

# 同类型相乘 → Product[X]
p1: Product[X] = X() * X()
# 混合子类相乘 → Product[Base]
p2: Product[Base] = X() * Y()
# 实例×Product[Y] → Product[Base]
p3: Product[Base] = X() * Product([Y()])
# Product[X]×实例 → Product[Base]
p4: Product[Base] = Product([X()]) * Y()
# 链式相乘 → Product[Base]
p5: Product[Base] = X() * Y() * Z()

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