Python嵌套dataclasses如何处理空字典默认值？

问题背景 #

你定义了嵌套的dataclass，baseClass支持从config.json读取配置，但配置中b字段存在两种情况：完整数据或空字典，之前的两种写法都无法适配。

错误写法原因 #

• 写法一：b: Optional[B] = {}
  类型标注是Optional[B]（允许B实例或None），但默认值是{}（字典类型），类型不匹配会触发检查错误；同时空字典不等于None，加载配置时也无法被识别为可选空值。

• 写法二：b: B = field(default_factory=dict)
default_factory返回字典，但字段类型要求是B实例，类型不兼容，初始化时直接报错。

解决方案 #

根据业务需求，分两种场景处理：

场景1：空字典时将b设为None

将b的类型设为Optional[B]，默认值设为None，同时在baseClass的加载逻辑中，把空字典转为None。

from dataclasses import dataclass, fields
from typing import Optional

@dataclass
class B(baseClass):
    d: float
    e: int

@dataclass
class A(baseClass):
    b: Optional[B] = None
    c: str

# 扩展baseClass的from_dict方法（假设baseClass已有该方法）
class baseClass:
    @classmethod
    def from_dict(cls, data):
        processed_data = data.copy()
        # 处理A类的b字段：空字典转为None
        if cls is A and processed_data.get('b') == {}:
            processed_data['b'] = None
        # 递归处理嵌套dataclass
        for field in fields(cls):
            field_type = field.type
            if field.name in processed_data and processed_data[field.name] is not None:
                if hasattr(field_type, 'from_dict'):
                    processed_data[field.name] = field_type.from_dict(processed_data[field.name])
        return cls(**processed_data)

场景2：空字典时创建B的默认实例

如果希望空字典时生成带默认值的B实例，先给B的字段设置默认值，再在A中配置默认工厂或在加载逻辑中处理。

from dataclasses import dataclass, field, fields

# 给B的字段设置默认值
@dataclass
class B(baseClass):
    d: float = 0.0
    e: int = 0

@dataclass
class A(baseClass):
    # 用default_factory创建B的默认实例
    b: B = field(default_factory=B)
    c: str

# 扩展baseClass的from_dict方法
class baseClass:
    @classmethod
    def from_dict(cls, data):
        processed_data = data.copy()
        # 处理A类的b字段：空字典时使用默认实例
        if cls is A:
            b_data = processed_data.get('b', {})
            if not b_data:
                processed_data['b'] = B()
            else:
                processed_data['b'] = B.from_dict(b_data)
        return cls(**processed_data)

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