如何解决可变字典类型注解引发的Pylance类型不兼容警告？

问题场景 #

定义了如下类结构：A是父类，B、C继承自A；MyClass中的dic属性声明为dict[str, A]，通过两个方法分别向字典中添加B、C的实例。当尝试访问字典里B实例的method_B方法时，Pylance会提示无法访问A类型的method_B成员。尝试用typing.cast将整个dic强转为dict[str,B]后，虽然能访问方法，但触发新警告：dict[str, B]与dict[str, A]不兼容，因为字典的值类型是不变的，无法直接赋值修改dic的类型。

相关代码：

class A:
    ...

class B(A):
    def method_B(self): ...

class C(A):
    def method_C(self): ...

class MyClass:
    dic: dict[str, A] = {}
    
    def add_B_obj(self, name: str):
        self.dic[name] = B()

    def add_C_obj(self, name: str):
        self.dic[name] = C()

尝试cast的代码及警告：

from typing import cast

m = MyClass()
m.add_B_obj('B_obj')

m.dic = cast(dict[str,B],m.dic) # 触发警告

m.dic['B_obj'].method_B # 可访问方法但存在警告

警告内容：

Cannot assign member "dic" for type "MyClass"
  "dict[str, B]" is incompatible with "dict[str, A]"
    TypeVar "_VT@dict" is invariant
      "B" is incompatible with "A"

解决方案 #

方案1：单独转换字典中的目标值（推荐） #

不要修改整个字典的声明类型，仅在访问具体子类实例时做类型转换，既保留dic的原有类型声明，又能让类型检查器识别子类方法。

代码示例：

from typing import cast

m = MyClass()
m.add_B_obj('B_obj')

# 仅对要访问的具体值做类型转换
b_instance = cast(B, m.dic['B_obj'])
b_instance.method_B()

方案2：使用协变容器类型（只读场景适用） #

如果dic仅用于读取操作、不需要修改，可以将其类型声明为Mapping[str, A]（Mapping的值类型是协变的）。但注意该类型是只读的，无法再通过add_B_obj等方法修改字典内容。

修改后的MyClass：

from typing import Mapping

class MyClass:
    dic: Mapping[str, A] = {}
    
    # 注意：该方案下以下方法会报错，因为Mapping不支持修改
    # def add_B_obj(self, name: str):
    #     self.dic[name] = B()

方案3：用联合类型声明字典值 #

如果能明确dic中只会存在B和C的实例，可以直接将dic的类型声明为dict[str, B | C]，类型检查器会自动识别两个子类的方法，无需额外转换。

修改后的MyClass：

class MyClass:
    dic: dict[str, B | C] = {}
    
    def add_B_obj(self, name: str):
        self.dic[name] = B()

    def add_C_obj(self, name: str):
        self.dic[name] = C()

使用时无需cast：

m = MyClass()
m.add_B_obj('B_obj')
m.dic['B_obj'].method_B()  # 无类型警告

方案4：自定义类型守卫做类型判断 #

如果不确定字典中的值具体是哪个子类，可以通过自定义类型守卫函数做类型检查，类型检查器会自动推断后续代码中的变量类型。

代码示例：

from typing import TypeGuard

def is_B(obj: A) -> TypeGuard[B]:
    return isinstance(obj, B)

def is_C(obj: A) -> TypeGuard[C]:
    return isinstance(obj, C)

# 使用场景
m = MyClass()
m.add_B_obj('B_obj')

target_obj = m.dic['B_obj']
if is_B(target_obj):
    target_obj.method_B()  # 类型检查器识别为B类型
elif is_C(target_obj):
    target_obj.method_C()

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