使用pickle为类加载状态

时间:2016-11-23 15:45:29

标签: python tkinter pickle

我正试图用泡菜弄湿我的脚,所以我写了一些像这样的示例代码:

class start(tk.Frame):
    def __init__(self,*args,**kwargs):
        tk.Frame.__init__(self,*args,**kwargs)
        frame = tk.Frame(self,width=600,height=600)
        self.val = 0
        self.plusButton = tk.Button(self,text="plus",command=self.plus)
        self.plusButton.pack()
        self.valLabel = tk.Label(self)
        self.valLabel.pack()
        self.saveButton = tk.Button(self,text="save",command=self.save)
        self.saveButton.pack()
        self.loadButton = tk.Button(self,text="load",command=self.load)
        self.loadButton.pack()
    def load(self):
        self.__dict__ = pickle.load(open( "testtesttest.p", "rb" ))
    def plus(self):
        self.val += 1 
        self.valLabel.config(text="%d"%(self.val))
    def save(self):
        pickle.dump(self.__getstate__, open( "testtesttest.p", "wb" ))

    def __getstate__(self):
        return self.__getstate__


if __name__=='__main__':
   root = tk.Tk()

   start(root).pack()
   root.mainloop()

所以这个应用程序的目标是一旦我点击加号按钮,屏幕上会有越来越多的数字。如果我保存它,关闭窗口,重新打开它,然后按下加载按钮,我会看到最后一次增加到的数字。我很擅长泡菜,现在的代码还给我了:

    Exception in Tkinter callback
Traceback (most recent call last):
File "/Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/tkinter/__init__.py", line 1550, in __call__return self.func(*args)
File "/Users/caoanjie/pickleDemotry.py", line 18, in load 
self.__dict__ = pickle.load(open( "testtesttest.p", "rb" ))pickle.
UnpicklingError: state is not a dictionary

我想知道这里的问题是什么。另外,我在网上看到很多教程或示例代码,例如:

with open('save_game.dat', 'wb') as f:
    player= pickle.load

with是什么意思?

1 个答案:

答案 0 :(得分:5)

你的问题可以简化为一个根本不使用tkinter的小班:

>>> class Foo:
...     def __getstate__(self):
...         print('getstate')
...         return self.__getstate__
... 
>>> obj = pickle.loads(pickle.dumps(Foo().__getstate__))
getstate
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
_pickle.UnpicklingError: state is not a dictionary

您正在挑选__getstate__实例方法,而不是start类的完整状态。 Python允许您这样做,假设您还实现了一个知道如何从该信息重建对象的__setstate__方法。来自docs

  

在unpickling时,如果类定义了__setstate __(),则会以unpickled状态调用它。在这种情况下,不要求状态对象是字典。否则,pickle状态必须是字典,其项目将分配给新实例的字典。

当您进行unpickle时,pickle会创建state的新实例,但由于该类没有__setstate__方法,因此pickle会尝试恢复该对象的__dict__。这失败是因为unpickled对象是实例方法,而不是dict。这表明你的方法存在更大的问题。

pickle重新创建整个对象,它不会还原到现有对象中。在您的情况下,如果您腌制整个start对象,除了您自己创建的对象之外,还会恢复第二个start对象。您可以将该对象的__dict__分配给__dict__,但这是一个非常冒险的主张。您将松开Frame对象的整个状态,以支持您所腌制的对象中发生的情况。无论如何,它可能无法腌制整个对象,因为tkinter是一个C扩展模块。

相反,您应该从要用于与用户交互的tkinter对象中分离要保存和还原的数据。这是一种常见的编程规则:将数据与表示分开。在这里,我有一个包含数据的类,我可以将它与tkinter类分开保存和恢复。

import tkinter as tk
import pickle

class State:
    def __init__(self):
        self.val = 0


class start(tk.Frame):
    def __init__(self,*args,**kwargs):
        tk.Frame.__init__(self,*args,**kwargs)
        frame = tk.Frame(self,width=600,height=600)
        self.state = State()
        self.plusButton = tk.Button(self,text="plus",command=self.plus)
        self.plusButton.pack()
        self.valLabel = tk.Label(self)
        self.valLabel.pack()
        self.saveButton = tk.Button(self,text="save",command=self.save)
        self.saveButton.pack()
        self.loadButton = tk.Button(self,text="load",command=self.load)
        self.loadButton.pack()
    def load(self):
        self.state = pickle.load(open( "testtesttest.p", "rb" ))
        self.valLabel.config(text="%d"%(self.state.val))
    def plus(self):
        self.state.val += 1 
        self.valLabel.config(text="%d"%(self.state.val))
    def save(self):
        pickle.dump(self.state, open( "testtesttest.p", "wb" ), 4)

if __name__=='__main__':
   root = tk.Tk()

   start(root).pack()
   root.mainloop()