如何让自定义类在 dataclasses.asdict() 中正确序列化

`dataclasses.asdict()` 在处理继承自 `list` 的自定义类时会因迭代器提前耗尽导致空列表，根本原因是 `__init__` 中直接消费了可迭代参数；修复方式是避免在调用 `super().__init__()` 前遍历或修改原始迭代器。

dataclasses.asdict() 是将数据类实例递归转换为嵌套字典的官方推荐工具，但它对容器类型的处理有明确假设：当遇到 list 或 tuple 实例时，它会通过 type(obj)(...) 构造新实例，并传入一个生成器表达式（_asdict_inner(v) for v in obj）来逐项序列化元素。这一机制依赖于对象能被多次迭代——而问题中的 CustomFloatList 在 __init__ 中直接遍历了传入的 args（可能为生成器、map 对象等一次性迭代器），导致后续 asdict 内部的 for v in obj 遍历时已无元素可取，最终构造出空列表。

? 问题复现与根源分析

以下代码清晰展示了问题本质：

from dataclasses import dataclass, asdict

class CustomFloatList(list):
    def __init__(self, args):
        # ❌ 危险：此处遍历 args 会耗尽迭代器（如 map、range、生成器）
        for i, arg in enumerate(args):
            assert isinstance(arg, float), f"Index {i} must be float, got {type(arg).__name__}"
        super().__init__(args)  # 此时 args 已空！

@dataclass
class Poc:
    x: CustomFloatList

p = Poc(x=CustomFloatList([1.0, 2.0]))  # 注意：直接传 list 也能触发问题（因 list(iter) 不耗尽，但其他类型会）
print(asdict(p))  # {'x': []} ← 错误结果！

⚠️ 关键点：asdict 内部调用 type(obj)(v for v in obj) 时，obj 若已被遍历过一次（如 for ... in args），则第二次遍历（for v in obj）返回空。

✅ 正确实现方案

方案一：预缓存为列表（推荐，语义清晰）

class CustomFloatList(list):
    def __init__(self, args):
        # ✅ 安全：先转为 list，再校验和初始化
        args_list = list(args)  # 缓存所有值，支持多次遍历
        for i, arg in enumerate(args_list):
            if not isinstance(arg, float):
                raise TypeError(f"Index {i} must be float, got {type(arg).__name__}")
        super().__init__(args_list)

方案二：初始化后再校验（更高效，适合大数据）

class CustomFloatList(list):
    def __init__(self, args):
        # ✅ 先委托父类构造，再校验内容
        super().__init__(args)
        # 此时 self 已包含全部元素，可安全遍历
        for i, arg in enumerate(self):
            if not isinstance(arg, float):
                raise TypeError(f"Index {i} must be float, got {type(arg).__name__}")

✅ 两种方案均能确保 asdict(p) 正确输出 {'x': [1.0, 2.0]}。

? 验证示例

@dataclass
class Poc:
    x: CustomFloatList

p = Poc(x=CustomFloatList([3.14, 2.71]))
print(p)           # Poc(x=[3.14, 2.71])
print(asdict(p))   # {'x': [3.14, 2.71]} ← 正确！

? 补充建议

• 避免在 __init__ 中消耗外部迭代器：这是 Python 容器子类的通用原则，不仅影响 asdict，还可能破坏 copy.copy()、json.dumps()（配合自定义 encoder）等场景。
• 考虑使用 __post_init__（仅限 dataclass）：若该类本身也是 @dataclass，可在 __post_init__ 中做类型校验，但本例中 CustomFloatList 是独立容器类，不适用。
• 替代方案：使用 typing.Sequence + @dataclass(frozen=True) + 自定义 __post_init__：若约束逻辑复杂，可放弃继承 list，改用组合模式（如 dataclass 包含 list[float] 字段 + 校验逻辑），提升可测试性与可维护性。

总之，让自定义容器兼容 asdict 的核心是保证其可重复迭代性。优先采用“先构造、后校验”或“预缓存再校验”的策略，即可兼顾类型安全与序列化健壮性。