如何pickle和unpickle继承自defaultdict的类的实例？

Question

我有一个继承自 defaultdict 的类，如下所示：

class listdict(defaultdict):
    def __init__(self):
        defaultdict.__init__(self, list)

我可以pickle它，但是当我取消pickle它时，会发生这种情况：

('__init__() takes exactly 1 argument (2 given)', <class 'listdict'>, (<type 'list'>,))

该类没有定义pickle协议的任何特殊方法。普通 defaultdict(list) 的酸洗和取消酸洗按预期工作。谁能启发我吗？

Answer 1

类型通过定义一组（相当大的）方法中的一个或多个来定义它的实例如何被腌制。每个人都有自己微妙的行为。请参阅有关 pickle 协议的文档。对于collections.defaultdict，它使用__reduce__方法：

>>> l = collections.defaultdict(list)
>>> l.__reduce__()
(<type 'collections.defaultdict'>, (<type 'list'>,), None, None, <dictionary-itemiterator object at 0x7f031fb3c470>)

元组中的第一项是类型，第二项是要传递的参数的元组实例化时的类型。如果您不覆盖 __reduce__，第一项将正确更改为您的类型，但第二项不会。这会导致您看到的错误。一个如何修复它的粗略示例：

>>> import collections
>>> import pickle
>>> class C(collections.defaultdict):
...     def __init__(self):
...         collections.defaultdict.__init__(self, list)
...     def __reduce__(self):
...         t = collections.defaultdict.__reduce__(self)
...         return (t[0], ()) + t[2:]
...
>>> c = C()
>>> c[1].append(2)
>>> c[2].append(3)
>>> c2 = pickle.loads(pickle.dumps(c))
>>> c2 == c
True

这只是一个粗略示例，因为需要进行酸洗（例如 __reduce_ex__），而且都相当复杂。在这种情况下，使用__getinitargs__可能会更方便。

或者，您可以使类的 __init__ 方法采用可选可调用，默认为 list，或者您可以只使用函数而不是类:

def listdict():
    return collections.defaultdict(list)

Answer 2

该错误表明您的“listdict”类应该采用一个参数（隐式自我），但得到了两个参数。

您的类继承自defaultdict，并定义一个初始值设定项。该初始化程序调用 defaultdict 的初始化程序并将“list”传递给它，在本例中它可以是函数或类。 （我懒得去检查）。

您的意思可能是这样做：

class listdict(defaultdict):
    def __init__(self, list):
        defaultdict.__init__(self, list)

现在，当使用给定列表初始化 listdict 时，它将该列表传递给 defaultdict 的构造函数，而不是传递对全局列表的引用。

（也就是说，使用与常见全局方法和类相同的名称（例如“str”、“list”等）被认为是不好的风格，这与您感到困惑的原因相同）。