Question

我有A类。它有一个带参数的构造函数。它使用__getstate__和__setstate__实现自定义酸洗。这很好。

我有另一个类B，它包含一个A类实例作为属性。 B类也实现__getstate__和__setstate__。如何让他们俩合作保存和恢复他们的状态。

让B级__getstate__打电话给A级课程很容易，但pickle对此并不了解，并且当我尝试从B＆＃39; __setstate__恢复时，它无法恢复子对象，因为它还没有实例化。

示例：

import pickle

class A:
    def __init__(self, x, y, z):
        self.x = x
        self.y = y
        self.z = z

    def __eq__(self, other):
        return self.x == other.x and self.y == other.y and self.z == other.z

    def __getstate__(self):
        return self.x, self.y, self.z

    def __setstate__(self, state):
        self.x, self.y, self.z = state

a = A(1, 2, 3)
a_save = pickle.dumps(a)
assert a == pickle.loads(a_save)
# WORKS FINE TO HERE

class B:
    def __init__(self):
        self.a = A(1, 2, 3)
        self.b = 4

    def __getstate__(self):
        # Is this correct?
        return self.a.__getstate__(), self.b

    def __setstate__(self, state):
        return self.a.__setstate__(), self.b

    def __eq__(self, other):
        return self.a == other.a and self.b == other.b

b = B()
b_save = pickle.dumps(b)
# Fails in pickle.loads HERE
assert b == pickle.loads(b_save)

这会产生：

Traceback (most recent call last):
  File "p.py", line 41, in <module>
    assert b == pickle.loads(b_save)
  File "p.py", line 33, in __setstate__
    return self.a.__setstate__(), self.b
AttributeError: 'B' object has no attribute 'a'

如何将a重建为A的实例并从保存的状态恢复其原始属性值？

（请注意，这是一个用来演示问题的玩具示例。真实世界的代码要复杂得多。）

Answer 1

B不应手动调用A __getstate__和__setstate__。 B甚至不需要知道A有这些方法。它应该处理嵌套的A实例，就像A没有这些方法一样：

def __getstate__(self):
    return (self.a, self.b)
def __setstate__(self, state):
    self.a, self.b = state

pickle：如何在包含的对象

1 个答案: