如何在Python中完整保存/读取类

Question

som = SOM_CLASS() # includes many big difficult data structures
som.hard_work()
som.save_to_disk(filename)
#then later or another program
som = SOM_CLASS()
som.read_from_file(filename)
som.do_anythink_else()

或者

som = SOM_CLASS()
save(som)
#...
load(som)
som.work()

有什么简单的方法可以做到这一点？

Answer 1

看看Python的pickle库吧。

Answer 2

我用的是这段代码：

import cPickle
import traceback

class someClass():
    def __init__(self):
        #set name from variable name. http://stackoverflow.com/questions/1690400/getting-an-instance-name-inside-class-init
        (filename,line_number,function_name,text)=traceback.extract_stack()[-2]
        def_name = text[:text.find('=')].strip()
        self.name = def_name

        try:
            self.load()
        except:
            ##############
            #to demonstrate
            self.someAttribute = 'bla'
            self.someAttribute2 = ['more']
            ##############

            self.save()

    def save(self):
        """save class as self.name.txt"""
        file = open(self.name+'.txt','w')
        file.write(cPickle.dumps(self.__dict__))
        file.close()

    def load(self):
        """try load self.name.txt"""
        file = open(self.name+'.txt','r')
        dataPickle = file.read()
        file.close()

        self.__dict__ = cPickle.loads(dataPickle)

这段代码可以把一个类的实例（也就是对象）保存下来，并且可以根据它的名字再加载回来。这个代码是我博客上的内容，链接是 http://www.schurpf.com/python-save-a-class/。

Answer 3

你可以使用pickle来进行数据的序列化和反序列化。这个工具是向后兼容的，也就是说，未来的版本会支持所有旧版本的协议。

import pickle

som = SOM_CLASS()
fileObject = <any file-like object>
pickle.dump(som, fileObject)
#...
som = pickle.load(fileObject)
som.work()

不过要注意，如果你把序列化的对象传输到另一台电脑上，确保这个连接是安全的，因为pickle可能不太安全（这是每个使用pickle的人都应该了解的文章）。

另外一个选择是较旧的模块marshal。