Python中Set[A]不接受Set[B]？Mypy类型提示疑问

问题原因 #

单个B实例能传给接受A的函数，是因为Python类继承支持协变——子类实例可以安全当作父类实例使用。但set[B]不能直接传给接受set[A]的函数，核心原因是set属于**不变型（invariant）**泛型容器：
如果允许set[B]被当作set[A]传入，函数内部可能会向集合中添加A的其他子类（比如C）的实例，导致原本只能容纳B的集合混入非B元素，破坏类型安全。

举个直观的反例：

class A:
    pass

class B(A):
    pass

class C(A):
    pass

def func_of_set_of_a(s: set[A]) -> None:
    s.add(C())  # 若传入set[B]，这行代码会非法向B集合中加入C实例

# mypy报错：Argument 1 has incompatible type "set[B]"; expected "set[A]"
func_of_set_of_a(set[B]())

解决办法 #

针对你不想用联合类型的需求，有三种实用方案：

1. 使用只读抽象集合类型AbstractSet #

collections.abc.AbstractSet是只读的集合抽象类型，它是协变的——只读操作不会修改集合内容，不存在混入其他类型的风险。修改函数参数类型为AbstractSet[A]即可兼容set[B]：

from collections.abc import AbstractSet

def func_of_set_of_a(s: AbstractSet[A]) -> None:
    for item in s:
        # 仅做遍历、判断成员等只读操作
        print(item)

func_of_set_of_a(set[B]())  # mypy不再报错

2. 用协变类型变量约束 #

通过TypeVar定义协变的类型变量，限定其为A的子类，这样函数可以接受任意子类的集合，但要注意只能用于只读场景：

from typing import TypeVar

A_co = TypeVar("A_co", bound=A, covariant=True)

def func_of_set_of_a(s: set[A_co]) -> None:
    # 禁止执行add、update等修改集合的操作，否则mypy会报错
    pass

func_of_set_of_a(set[B]())  # 正常通过类型检查

3. 定义泛型函数 #

使用泛型函数让参数类型适配任意A的子类集合，既保证类型安全，又支持所有子类：

from typing import TypeVar

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

def func_of_set_of_a(s: set[T]) -> None:
    # 可以针对T类型的元素做操作，比如返回T类型的列表
    result: list[T] = list(s)
    return result

func_of_set_of_a(set[B]())  # 支持B的集合
func_of_set_of_a(set[C]())  # 也支持其他A的子类集合

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