python - 防止字段被 pickle

Question

我有这样一个类:

class Something(object):

    def __init__(self):
        self._thing_id
        self._cached_thing

    @property
    def thing(self):
        if self._cached_thing:
            return self._cached_thing

        return Thing.objects.get(id=self._thing_id)

像这样对对象进行 pickling 时，我想防止对 _cached_thing 字段进行 pickling，因为它是易变的并且是专门在内存中实现的。

有没有办法向 Pickle 建议我只希望对我的字段的一个子集进行 pickle？

Answer 1

可以通过三种方式自定义 Pickle，如 the docs 中所述。 .

• 提供__getstate__ 和__setstate__ 方法。
• 提供 __getnewargs__/__getnewargs_ex__(以及采用这些参数的构造函数)。
• 提供 __reduce__(以及提供给 __reduce__ 的函数来反转它)。

第一个通常是最简单的:

class Something(object):

    def __init__(self):
        self._thing_id
        self._cached_thing

    def __getstate__(self):
        return self._thing_id
    def __setstate__(self, thing_id):
        self._thing_id = thing_id

    # etc.

如果你想要更通用的东西，它会选择所有值(包括由子类设置的值，或在创建后动态设置的值等)除了你的黑名单，请注意默认值是“实例的__dict__ 是 pickle 的”，所以只需过滤一下:

_blacklist = ['_cached_thing']
def __getstate__(self):
    return {k: v for k, v in self.__dict__.items() if k not in self._blacklist}
def __setstate__(self, state):
    self.__dict__.update(state)

请参阅 gnibbler 的 comment关于这个问题:如果你正在做一些通用的事情，你应该认真考虑提出某种命名约定，而不是在每个类中放置一个黑名单。任何知道或学习该约定的读者都会立即知道哪些属性是“缓存”值而不是“真实”值的一部分，事情的工作原理会更加明显，每个类中你要做的工作更少，并且更少的地方因打字错误而搞砸了……