在python中保存和加载类

时间:2014-03-16 11:39:47

标签: python class pickle

我正在尝试使用pickle保存和加载一些类。在我的项目中,我有一些类包含列表和/或字典和/或其他类的元组的变量。当我加载保存的类时,其中的类是不同的。我找到了一个关于静态方法__new__ here的精彩教程。我做了以下代码。当执行函数加载时,我获得了class1的3个实例和class2的2个实例。为什么呢?

import pickle


def saveclass(obj):
    f = file(obj.name, 'wb')
    pickle.dump(obj, f, -1)
    f.close()


def loadclass(name):
    f = file(name, 'rb')
    retclass = pickle.load(f)
    f.close()
    return retclass


class basicclass(object):

    def __new__(cls, *args, **kw):
    instances = cls.instances
    obj = object.__new__(cls, *args, **kw)
    if obj in instances:
        return instances[instances.index(obj)]
    else:
        instances.append(obj)
        return obj


class class1(basicclass):
    instances = []

    def __init__(self, name):
    basicclass.__init__(self)
    self.name = name


class class2(basicclass):
    instances = []

    def __init__(self, name, otherclass):
    basicclass.__init__(self)
    self.name = name
    self.otherclass = otherclass


c1 = class1("class1")
c2 = class2("class2", c1)

print c1.name
print c2.name, 'has', c2.otherclass.name
print c2.name, "'s 'inside' class is c1:", c2.otherclass == c1

print "Class1 instances"
for inst in c1.instances:
    print inst.name, ':', inst

print "Class2 instances"
for inst in c2.instances:
    print inst.name, ':', inst

print 'saving classes'
saveclass(c1)
saveclass(c2)


print 'Resetting classes'
c1 = None
c2 = None

print 'Reloading classes'

c1 = loadclass("class1")
c2 = loadclass("class2")


print c1.name
print c2.name, 'has', c2.otherclass.name
print c2.name, "'s 'inside' class is c1:", c2.otherclass == c1

print "Class1 instances"
for inst in c1.instances:
    print inst.name, ':', inst

print "Class2 instances"
for inst in c2.instances:
    print inst.name, ':', inst

输出:

  

> class1 
> class2 has class1 
> class2 's 'inside' class is c1: True 
> Class1 instances 
> class1 : <__main__.class1 object at 0xb7302a8c> 
> Class2 instances 
> class2 : <__main__.class2 object at 0xb7302a2c> 
> saving classes
> Resetting classes
> Reloading classes
> class1
> class2 has class1
> class2 's 'inside' class is c1: False 
> Class1 instances 
> class1 : <__main__.class1 object at 0xb7302a8c>
> class1 : <__main__.class1 object at 0xb72fe10c> 
> class1 : <__main__.class1 object at 0xb72fe1ac>
> Class2 instances 
> class2 : <__main__.class2 object at 0xb7302a2c>
> class2 : <__main__.class2 object at 0xb72fe18c>

1 个答案:

答案 0 :(得分:3)

我想到了两种可能性:

  1. 您希望在酸洗和去除污渍时保留身份。已存在的类的对象具有特殊标识符。如果已经存在,则在取消对象时重新使用该对象。 Here is an example使用__new__检查对象是否已存在。你也可以和json一起使用它。当对象中存在可变状态时会出现问题。然后加载旧状态更新它们。

  2. 使用相同的序列化程序取消并挑选对象。不要只使用dumpload,而是使用相同的PicklerUnpickler

  3. 我认为这应该已经在某处得到解答了。也许你也尝试使用&#34;对象身份&#34; &#34;序列&#34;等,并附上找到的答案。