Python dataclass用 slots 控制实例字段开销的实现方法

我在一次需要创建大量记录对象的脚本里，先把注意力放在了字段类型和序列化，后来才发现对象本身的动态属性空间也会累积开销。对只保存固定字段的模型，@dataclass(slots=True) 是更直接的收敛方式：它会为字段生成槽位，通常不再给每个实例准备普通的 __dict__，同时保留 dataclass 的构造、比较和 repr 能力。官方文档地址：https://docs.python.org/3/library/dataclasses.html。

先把类定义改成 @dataclass(slots=True)，再检查动态赋值、继承、弱引用和序列化调用方；slots 适合字段集合稳定的对象，不是所有 Python 类的默认优化开关。

先判断对象是否真的适合固定字段

slots 的收益建立在“字段集合稳定”上。订单快照、批量解析结果、事件载荷这类对象通常在构造后只更新既有字段；插件上下文、临时缓存和依赖注入对象则可能依赖运行时追加属性。后者直接开启 slots，最先遇到的往往不是性能问题，而是 AttributeError。

用 slots=True 把固定字段变成受约束的对象布局

最小改法只是在装饰器上增加参数。它会返回一个带生成槽位的新类，因此不要把装饰前的类对象缓存到其他注册表里；业务代码仍然可以用属性语法读写已声明字段。

from dataclasses import dataclass

@dataclass(slots=True)
class LineItem:
    # 固定字段对应槽位，避免把临时键任意挂到实例上。
    sku: str
    quantity: int
    unit_price: int

    def total(self) -> int:
        # 价格使用整数分，避免示例把金额精度问题混进布局讨论。
        return self.quantity * self.unit_price

item = LineItem("A-17", 2, 1299)
print(item.total())
# 未声明的 item.note = "gift" 会触发 AttributeError。

这里的重点不是“所有对象都要更小”，而是把模型的边界写进类定义。若旧代码依赖 obj.__dict__ 做日志或序列化，需要改成字段级读取；对 dataclass 来说，fields(LineItem) 比直接查看 __slots__ 更可靠。

继承场景要先检查字段重复和弱引用需求

父类和子类都使用 slots 时，子类只需要声明自己的新增字段。较新的 Python 文档说明，如果字段名已存在于基类槽位，生成的子类不会再次覆盖它，所以不要通过遍历 __slots__ 推断完整字段集合。

from dataclasses import dataclass, fields
import weakref

@dataclass(slots=True)
class BaseRecord:
    # 基类负责所有记录共有的身份字段。
    record_id: str

@dataclass(slots=True, weakref_slot=True)
class OrderRecord(BaseRecord):
    # 子类只增加订单字段；record_id 继续使用父类槽位。
    amount: int

order = OrderRecord("r-100", 8800)
names = [item.name for item in fields(order)]
# weakref_slot=True 让缓存可以保存对实例的弱引用。
ref = weakref.ref(order)
print(names, ref() is order)

weakref_slot=True 必须和 slots=True 一起使用；如果子类或某个父类仍保留普通实例字典，实际布局也会受到继承链影响。这个边界需要在迁移前查清，而不是只看装饰器这一行。

迁移后按四项清单验收

1. 搜索 __dict__、动态 setattr 和未声明字段赋值，确认它们不是隐藏依赖。
2. 用 fields()、显式字段映射或现有序列化适配器替代遍历 __slots__。
3. 如果对象进入弱引用缓存，给出 weakref_slot=True 并覆盖父子类组合测试。
4. 用代表性实例数量做一次内存和构造时间对比；只有结果与维护成本都可接受时才保留迁移。

还要留意无参数 super() 与某些 __init_subclass__() 参数组合的兼容性。slots 改变的是类创建和属性布局，不能替代缓存策略、对象生命周期设计或真正的性能测量。

相关问题

slots=True 会让字段值自动变成只读吗？

不会。已声明字段仍可赋值；它限制的是未声明的实例属性，而不是冻结对象。需要不可变语义时，另行评估 frozen=True。

为什么不直接遍历 __slots__ 得到字段名？

继承时槽位可能来自多个父类，重复字段也可能被跳过；使用 fields() 才能按 dataclass 字段语义读取。

所有 dataclass 都应该开启 slots 吗？

不应该。字段固定、实例数量大且调用方不依赖动态属性时更合适；插件对象、扩展元数据和依赖动态挂载的类应保留原布局。