怎么自己腌制？

Question

我希望我的类实现 Save 和 Load 函数，这些函数只需对类进行 pickle 即可。但显然你不能以下面的方式使用“self”。你怎么能这样做呢？

self = cPickle.load(f)

cPickle.dump(self,f,2)

Answer 1

这就是我最终所做的。更新 __dict__ 意味着我们保留添加到类中的所有新成员变量，并且只更新对象上次 pickle 时存在的成员变量。这似乎是最简单的，同时在类本身内部维护保存和加载代码，因此调用代码只执行 object.save() 。

def load(self):
    f = open(self.filename, 'rb')
    tmp_dict = cPickle.load(f)
    f.close()          

    self.__dict__.update(tmp_dict) 


def save(self):
    f = open(self.filename, 'wb')
    cPickle.dump(self.__dict__, f, 2)
    f.close()

Answer 2

转储部分应该按照您的建议工作。对于加载部分，您可以定义一个 @classmethod 从给定文件并返回它。

@classmethod
def loader(cls,f):
    return cPickle.load(f)

然后调用者会做类似的事情：

class_instance = ClassName.loader(f)

Answer 3

如果你想让你的类从保存的pickle中更新自己......你几乎必须使用__dict__.update，就像你在自己的答案中一样。然而，这有点像一只猫在追它的尾巴……因为您要求实例本质上“重置”自身为之前的状态。

您的答案略有调整。您实际上可以pickle self。

>>> import dill
>>> class Thing(object):
...   def save(self):
...     return dill.dumps(self)
...   def load(self, obj):
...     self.__dict__.update(dill.loads(obj).__dict__)
... 
>>> t = Thing()
>>> t.x = 1
>>> _t = t.save()
>>> t.x = 2
>>> t.x
2
>>> t.load(_t)
>>> t.x
1

我使用 loads 和 dumps 而不是 load 和 dump 因为我希望将 pickle 保存到字符串中。使用 load 和 dump 到文件也可以。
而且，实际上，我可以使用 dill 将类实例 pickle 到文件中，以供以后使用……即使该类是交互式定义的。从上面继续...

>>> with open('self.pik', 'w') as f:
...   dill.dump(t, f)
... 
>>> 

然后停止并重新启动...

Python 2.7.10 (default, May 25 2015, 13:16:30) 
[GCC 4.2.1 Compatible Apple LLVM 5.1 (clang-503.0.40)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import dill
>>> with open('self.pik', 'r') as f:
...   t = dill.load(f)
... 
>>> t.x
1
>>> print dill.source.getsource(t.__class__)
class Thing(object):
  def save(self):
    return dill.dumps(self)
  def load(self, obj):
    self.__dict__.update(dill.loads(obj).__dict__)

>>> 

我正在使用 dill，可在此处使用：https://github.com/uqfoundation

Answer 4

我就是这样做的。优点是不需要创建新对象。您可以直接加载它。

def save(self):
    with open(self.filename, 'wb') as f:
        pickle.dump(self, f)

@classmethod
def load(cls, filename):
    with open(filename, 'rb') as f:
        return pickle.load(f)

如何使用它：

class_object.save()
filename = class_object.filename
del class_object

class_object = ClassName.load(filename)

下面，我用一个完全有效的最小示例更新了答案。这可以根据您自己的需要进行调整。

import pickle

class Test:

    def __init__(self, something, filename) -> None:
        self.something = something
        self.filename = filename


    def do_something(self) -> None:
        print(id(self))
        print(self.something)

    def save(self):
        with open(self.filename, 'wb') as f:
            pickle.dump(self, f)


    @classmethod
    def load(cls, filename):
        with open(filename, 'rb') as f:
            return pickle.load(f)


test_object = Test(44, "test.pkl")
test_object.do_something()
test_object.save()
filename = test_object.filename
del test_object

recovered_object = Test.load(filename)
recovered_object.do_something()

Answer 5

这里的文档中有一个如何pickle实例的示例。 （向下搜索“TextReader”示例）。这个想法是定义 __getstate__ 和 __setstate__ 方法，它们允许您定义需要腌制的数据，以及如何使用该数据重新实例化对象。