设置一个基类来处理跟踪类实例,同时仍然允许垃圾收集。这篇文章的第二个答案很接近,但在从保存的实例中解压时,我还需要添加到列表中。instances
这是我的代码:
import weakref
import pickle
from pprint import pprint
import copy
class BaseClass(object):
def __new__(cls, *args, **kwargs):
instance = object.__new__(cls, *args, **kwargs)
if 'instances' not in cls.__dict__:
cls.instances = []
cls.instances.append(weakref.ref(instance))
return instance
def __getstate__(self):
print "pickling"
d = self.__dict__
return d
def __setstate__(self, state):
print "unpickling"
self.__dict__ = state
if 'instances' not in self.__dict__:
self.instances = []
self.__class__.instances.append(weakref.ref(self))
return self
BaseClass.instances = []
if __name__ == '__main__':
a = BaseClass()
b = BaseClass()
c = BaseClass()
pprint(BaseClass.instances)
f = file("test.pkl","wb")
pickle.dump(c,f)
f.close()
f = file("test.pkl","rb")
d = pickle.load(f)
f.close()
pprint(BaseClass.instances)
所以,奇怪的是:“unpickling”永远不会被打印出来,并且d
实例没有进入instances
列表,表明__setstate__
永远不会被调用。
如果我替换d = pickle.load(f)
为d = copy.copy(pickle.load(f))
,则将d
其添加到instances
列表中,并且“酸洗”打印两次而不是一次。
如果我替换class BaseClass(object):
为class BaseClass:
,则“酸洗”和“解酸”每个打印一次,如预期的那样,并d
进入列表但a
,b
和c
不。
有人知道发生了什么吗?似乎显式继承object
破坏了默认值__getstate__
和__setstate__
行为。