Python dataclass slots 与 weakref_slot 怎么一起用

我第一次把缓存对象改成槽位数据类时，真正卡住的不是 slots=True，而是之后还要把对象交给弱引用缓存。组合写法很明确：在 Python 3.11 及以上使用 @dataclass(slots=True, weakref_slot=True)。前者生成字段槽位并通常移除实例字典，后者额外加入 __weakref__ 槽位；只写 weakref_slot=True 会直接报错。

先弄清 slots 和 weakref_slot 各自做什么

slots=True 会让 dataclass 生成 __slots__，实例的字段通过固定槽位保存，常见结果是不能再随意写入未声明属性。weakref_slot=True 则为实例增加 __weakref__，让 weakref.ref() 能够引用它。两者解决的是存储布局和引用能力两个不同问题。

这组参数的版本边界也要记住：slots 是 Python 3.10 加入的，weakref_slot 是 Python 3.11 加入的。部署环境如果仍有 Python 3.10，不能直接照抄第二个参数。

用 dataclass 同时打开 slots 与 weakref_slot

下面的类适合表示一个短生命周期会话：只允许声明好的字段，同时允许缓存层持有弱引用。代码里的注释解释关键边界，示例不依赖第三方包。

from dataclasses import dataclass

@dataclass(slots=True, weakref_slot=True)
class Session:
    # 槽位只保存这两个声明过的业务字段
    session_id: str
    payload: bytes

item = Session("s-001", b"data")
item.payload = b"new-data"  # 槽位类仍然可变

try:
    item.debug = True  # 未声明属性没有可写的实例字典
except AttributeError:
    print("debug 不是已声明字段")

这里的 slots 不会自动把对象变成只读对象，它只是限制属性存储位置。需要值对象语义时再考虑 frozen=True，不要把内存布局和不可变性混成一个开关。

验证弱引用和字段反射

弱引用不会替对象保活。只要其他强引用都释放，weakref.ref 的调用结果就会变成 None；如果注册了回调，回调会在对象即将被回收时收到弱引用对象本身，而不是把对象重新保存成强引用。

import weakref
from dataclasses import fields

events: list[str] = []

def on_collect(ref: weakref.ReferenceType[Session]) -> None:
    # 回调只记录事件，不保存 Session，避免重新延长生命周期
    events.append("session released")

item = Session("s-002", b"payload")
ref = weakref.ref(item, on_collect)
assert ref() is item

field_names = [field.name for field in fields(item)]
print(field_names)  # ['session_id', 'payload']

del item
assert ref() is None
assert events == ["session released"]

字段反射要用 fields()。官方文档特别提醒，继承场景下生成的 __slots__ 可能不会重复列出基类已有字段，因此直接读取 __slots__ 并不能可靠还原完整数据字段。

几个容易踩到的边界

如果类继承了带参数的 __init_subclass__()，slots=True 还可能触发类型错误；这属于槽位类与继承初始化的边界，不是 weakref 失效。另一个实用判断是：弱引用适合观察、缓存和回收通知，不适合承担对象的唯一所有权。

相关问题

slots=True 会让 dataclass 自动不可变吗？

不会。它限制未声明属性的存储方式，声明过的字段仍可以重新赋值；需要只读语义应单独评估 frozen=True。

为什么 weakref_slot=True 不能单独使用？

因为它只负责加入 __weakref__ 槽位，必须建立在槽位类布局上，所以官方要求同时指定 slots=True。

如何取得继承后的完整字段列表？

对数据类或实例调用 dataclasses.fields()，再读取每个 Field.name；不要依赖某一个类的 __slots__。

弱引用回调为什么不要闭包捕获对象？

闭包若保存被观察对象，会形成强引用，回收条件就被推迟。回调只保留必要的字符串、计数或弱引用即可。