Python pickling 保持对象身份

Question

有什么方法可以保留 pickle 对象的身份，即打印以下内容True:

import pickle

class Foo:
    pass

x = Foo()
print(x is pickle.loads(pickle.dumps(x)))          #False

我在 Linux 机器上使用 cPickle 和 cpython 3.x，不需要便携的东西。

Answer 1

是的，这是可能的；您需要以某种方式在 pickle 结果中包含“身份”；在这种情况下，最自然的方法是使用 __getnewargs__ 并让 __new__ 方法返回现有的缓存实例。

import uuid
import weakref


class Foo(object):
    ident_cache = weakref.WeakValueDictionary()

    def __new__(cls, identity=None, **kwargs):
        if identity is None:
            identity = uuid.uuid1()
        try:
            self = cls.ident_cache[identity]
        except KeyError:
            self = super(Foo, cls).__new__(cls)
            self.__identity = identity
            self.__init__(**kwargs)
            cls.ident_cache[identity] = self
        return self

    def __getnewargs__(self):
        return (self.__identity,)

    def __init__(self, foo):
        self.foo = foo

>>> import pickle
>>> a = Foo(foo=1)
>>> b = pickle.loads(pickle.dumps(a, pickle.HIGHEST_PROTOCOL))
>>> a is b
True

重要的一点是，您必须使用协议(protocol)版本 2(或假设更高版本)；因为否则，__new__ 永远不会被调用。这只是 pickle.dumps 的问题，loads 不关心。