使用Python Protocol为含多态属性的类做类型提示遇问题

你遇到的问题根源在于Container类的attr被静态类型检查器（pytype）认定为固定的Tuple[Parent, ...]类型——无论初始化时传入的是纯Child1实例还是混合实例，类型检查器只会依据类定义的标注来判定类型。而你定义的ContainerChild1Proto要求attr是Tuple[Child1, ...]，由于Tuple在Python类型系统中是不变类型，Tuple[Parent, ...]与Tuple[Child1, ...]不兼容，因此pytype判定Container实例不符合协议。

要让类型检查器能区分不同元素类型的Container实例，需要将Container改造为泛型类，使其attr类型随传入的元素类型动态变化。

修改后的完整代码 #

from typing import Tuple, Generic, TypeVar

# 定义绑定到Parent的类型变量
T = TypeVar('T', bound='Parent')

class Parent:
    ...

class Child1(Parent):
    def method1(self): ...

class Child2(Parent):
    def method2(self): ...

# 让Container继承Generic，使用类型变量T
class Container(Generic[T]):
    def __init__(self, attr: Tuple[T, ...]):
        self.attr = attr 

# 直接约束参数为Container[Child1]
def func(container_argument: Container[Child1]):
    for e in container_argument.attr:
        e.method1()

# 类型检查器自动推断为Container[Child1]
child1_only_container = Container((Child1(),))
# 自动推断为Container[Parent]
mixed_container = Container((Child1(), Child2()))

func(child1_only_container)  # 正常通过检查
func(mixed_container)  # 触发类型错误，符合预期

方案原理 #

泛型Container[T]会根据初始化时传入的attr元素类型，生成具体的类型实例（比如Container[Child1]或Container[Parent]）。此时func明确要求参数是Container[Child1]，类型检查器就能准确识别：

• child1_only_container的类型是Container[Child1]，符合要求；
• mixed_container的类型是Container[Parent]，与Container[Child1]不兼容，触发错误。

若坚持使用Protocol的替代方案 #

如果你一定要用Protocol实现，可以结合泛型Protocol定义，但泛型Container的方式更直接且符合Python类型系统的最佳实践：

from typing import Protocol, Generic, TypeVar

T = TypeVar('T', bound='Parent')

class ContainerProto(Generic[T], Protocol):
    attr: Tuple[T, ...]

def func(container_argument: ContainerProto[Child1]):
    for e in container_argument.attr:
        e.method1()

# 需先将Container改造为泛型类，后续使用方式与之前一致

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