如何将基类dataclass字段赋值给派生类且满足MyPy类型校验要求

解决方案 #

有两种成熟的实现方案，都可以满足类型安全、不需要手动枚举父类字段的需求：

方案一：Python 3.11+ 原生解包方案 #

Python 3.11 引入了dataclass实例直接**解包的语法支持，搭配mypy 0.981及以上版本可以获得完整的类型检查能力，你最初写的语法本身就是合法的：

from dataclasses import dataclass

@dataclass
class TypeA:
  name: str
  age: int

@dataclass
class TypeB(TypeA):
  more: bool


def upgrade(a: TypeA) -> TypeB:
  return TypeB(
     more=False,
     **a,  # 3.11+原生支持，类型检查完全正常
  )

• 这种写法下mypy会完整校验所有必填参数：如果漏传more参数会直接抛出类型错误，不会出现__dict__/asdict的类型丢失问题
• 后续给TypeA新增/删除字段时，不需要修改upgrade函数的代码，mypy会自动同步校验
• 没有任何冗余代码，性能也最优

方案二：兼容低版本Python的类型安全实现 #

如果需要兼容Python 3.10及以下版本，可以通过「dataclass + 对应TypedDict」的方式实现类型安全的字段复制：

from dataclasses import dataclass, fields
from typing import TypedDict, TypeVar, Type

# 定义和TypeA字段完全对齐的TypedDict，mypy会自动校验字段一致性
class TypeATd(TypedDict):
    name: str
    age: int

T = TypeVar("T", bound="TypeA")
U = TypeVar("U", bound="TypeB")

@dataclass
class TypeA:
    name: str
    age: int

    def to_typed_dict(self) -> TypeATd:
        # 动态读取所有dataclass字段，不需要手动枚举
        return {f.name: getattr(self, f.name) for f in fields(self)}  # type: ignore[return-value]

@dataclass
class TypeB(TypeA):
    more: bool

    @classmethod
    def upgrade_from(cls: Type[U], a: TypeA, more: bool) -> U:
        return cls(**a.to_typed_dict(), more=more)

def upgrade(a: TypeA) -> TypeB:
    return TypeB.upgrade_from(a, more=False)

• 这里的TypedDict和TypeA的字段必须完全对齐，如果你后续修改TypeA的字段，mypy会报错提示TypedDict不匹配，避免遗漏
• 不需要手动枚举父类字段，所有字段通过fields函数自动读取
• 类型校验完整，漏传more等必填参数时mypy会正常报错
• 兼容Python 3.7+所有支持dataclass的版本

你之前用__dict__和asdict丢失类型的原因是：这两个方法的返回值类型都是dict[str, Any]，mypy无法感知字典内的具体键值类型，因此不会做必填参数校验，自然不会提示缺参数的问题。

内容的提问来源于stack exchange，提问作者edA-qa mort-ora-y