如何在mypy中处理继承场景下的类型注解兼容问题？

问题场景 #

你定义了如下继承结构：

class BaseClass:
    base_attribute: str


class A(BaseClass):
    attribute_for_class_A: str


class B(BaseClass):
    attribute_for_class_B: str

通过ClassUsingTheOthers类管理这些实例：

from dataclasses import dataclass
from typing import Dict

@dataclass
class ClassUsingTheOthers:
    fields: Dict[str, BaseClass]

    def get_field(self, field_name: str) -> BaseClass:
        field = self.fields.get(field_name)
        if not field:
            raise ValueError('Not found')
        return field

当你调用需要B实例的函数时：

def function_that_needs_an_instance_of_b(instance: B):
    print(instance.attribute_for_class_B)


if __name__ == "__main__":
    class_using_the_others = ClassUsingTheOthers(
        fields={
            'name_1': A(),
            'name_2': B()
        }
    )
    function_that_needs_an_instance_of_b(class_using_the_others.get_field('name_2'))

mypy会抛出错误：

error: Argument 1 to "function_that_needs_an_instance_of_b" has incompatible type "BaseClass"; expected "B" [arg-type]

在无法修改fields类型注解的前提下，有以下几种解决方案：

方案1：使用类型断言（cast） #

直接告诉mypy返回值的实际类型，适合你确定字段类型的场景：

from typing import cast

# 调用处修改
function_that_needs_an_instance_of_b(cast(B, class_using_the_others.get_field('name_2')))

注意：这种方式没有运行时类型检查，如果实际字段不是B类型，运行时会报错。

方案2：添加类型检查+assert #

在调用处先做运行时类型检查，用assert让mypy识别具体类型：

field = class_using_the_others.get_field('name_2')
assert isinstance(field, B), f"字段name_2的类型应为B，实际是{type(field)}"
function_that_needs_an_instance_of_b(field)

这种方式既有mypy类型校验，又有运行时安全检查，推荐用于需要确保类型正确性的场景。

方案3：扩展ClassUsingTheOthers的方法 #

给ClassUsingTheOthers添加一个支持泛型的类型安全方法，统一处理类型检查和返回：

from typing import TypeVar, Type

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

@dataclass
class ClassUsingTheOthers:
    fields: Dict[str, BaseClass]

    def get_field(self, field_name: str) -> BaseClass:
        field = self.fields.get(field_name)
        if not field:
            raise ValueError('Not found')
        return field

    def get_field_of_type(self, field_name: str, field_type: Type[T]) -> T:
        field = self.get_field(field_name)
        if not isinstance(field, field_type):
            raise TypeError(f"字段{field_name}的类型应为{field_type.__name__}，实际是{type(field).__name__}")
        return field

调用时传入目标类型：

function_that_needs_an_instance_of_b(class_using_the_others.get_field_of_type('name_2', B))

这种方式复用性强，适合多次按类型获取字段的场景，同时兼顾编译期和运行时的类型安全。

内容的提问来源于stack exchange，提问作者Antonio Gamiz Delgado