Pickle 没有 __reduce__ 方法的 dict 子类不会加载成员属性

我需要确保dict只能接受某种类型的对象作为值。它也必须是可挑选的。 这是我的第一次尝试：

import pickle

class TypedDict(dict):
    _dict_type = None

    def __init__(self, dict_type, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._dict_type = dict_type

    def __setitem__(self, key, value):
        if not isinstance(value, self._dict_type):
            raise TypeError('Wrong type')
        super().__setitem__(key, value)

如果我用下面的代码测试它（python 3.5）

my_dict = TypedDict(int)
my_dict['foo'] = 98

with open('out.pkl', 'wb') as fin:
    pickle.dump(my_dict, fin)

with open('out.pkl', 'rb') as fin:
    out = pickle.load(fin)

我收到错误：TypeError: isinstance() arg 2 must be a type or tuple of types.
似乎没有加载正确的值_dict_type它改为使用默认值None.
另外，它似乎依赖于协议，就好像它可以正常工作一样protocol=0

但是，如果我覆盖__reduce__方法并调用 super 一切都会神奇地起作用。

def __reduce__(self):
    return super().__reduce__()

怎么可能呢？不应该是两个类（w/o__reduce__） 相等的？我缺少什么？

怎么可能呢？不应该是两个类（w/o__reduce__） 相等的？我缺少什么？

你错过了一个关键的步骤：如果没有__reduce__方法（或者如果失败！）它将使用其他方法来pickle你的类。所以一堂课__reduce__如果没有的话，就不会表现得像一个班级__reduce__（有几种特殊的方法具有类似的行为）！

在第一种情况下，它将默认为基本dict转储和加载，然后处理子类逻辑。所以它将使用几个创建字典__setitem__调用，然后设置实例属性。但你的__setitem__需要实例属性 _dict_type。如果没有则默认为类属性 None，这失败了

TypeError: isinstance() arg 2 must be a type or tuple of types

这就是为什么如果你想腌制你的东西它会起作用TypedDict没有__reduce__如果它不包含任何键值对。因为它不会调用__setitem__然后设置实例属性：

my_dict = TypedDict(int)

with open('out.pkl', 'wb') as fin:
    pickle.dump(my_dict, fin)

with open('out.pkl', 'rb') as fin:
    out = pickle.load(fin)

print(out._dict_type)   # int

另一方面，如果你实现你的__reduce__方法，因为与失败的普通字典不同__reduce__- 它确实适用于子类（但如果您不实现，则不会尝试__reduce__):

>>> d = {1: 1}
>>> dict.__reduce__(d)
TypeError: "can't pickle dict objects"

>>> d = TypedDict(int)
>>> dict.__reduce__(d)
(<function copyreg._reconstructor>,
 (__main__.TypedDict, dict, {}),
 {'_dict_type': int})