如何腌制 NotImplementedType

Question

如何在 Python 中添加对酸洗传统上不可选择的类型的支持？

我有一个需要腌制的复杂对象，它包括对 NotImplementedType 类的引用。该类是第三方的，所以我不能覆盖它的 __copy__() 或 __deepcopy__() 或 __getstate__() 方法。

我不完全确定为什么 Pickle 不能序列化 NotImplementedType。我敢肯定有一些愚蠢的教条理由，我不在乎。事实是这个类永远不会改变，也不包含任何状态，所以它应该是可序列化的。果然dill package可以序列化这个类就好了。

所以我尝试为NotImplementedType 实现一个自定义的copy_reg 处理程序，该处理程序使用莳萝，例如：

copy_reg.pickle(NotImplementedType,
   lambda code: (dill.loads, (dill.dumps(code),)),
   dill.loads)

但是，尝试深度复制我的对象会引发异常：

'NoneType' object has no attribute 'update'

来自/usr/lib/python2.7/copy.py 中的第 347 行。深入研究这段代码表明，copy_reg/deepcopy/pickle 模块期望序列化和反序列化实例，而不是对类的引用，并且它抛出这个异常是因为它试图实例化 NotImplementedType 而不是仅仅查找类引用。有什么解决办法吗？

Answer 1

所讨论的类实际上包含NotImplementedType 的可能性很小；相反，它可能正在腌制NotImplemented，而NotImplemented 的归约函数试图根据其类型腌制它，但NotImplementedType 并未直接暴露（您只能通过type(NotImplemented) 获得），因此尝试创建可以找到NotImplementedType 的限定名称失败。这在 Python 3 中已修复（移动的另一个参数），但在 Python 2 中修复看起来也相当简单，因为为 NotImplementedType 定义一个普通的 copy_reg 处理程序将确保 NotImplemented 在不引用其自身类型的情况下被腌制.

执行此操作的简单方法是让您的 reducer 函数返回命名为 NotImplemented 的字符串，然后在 unpickled 时将其作为全局查找。方便的是，这意味着您不需要定义任何辅助方法（在 unpickling 端可能不存在）：

copy_reg.pickle(type(NotImplemented), lambda obj: 'NotImplemented')

您根本没有定义构造函数（第三个copy_reg.pickle 参数）； pickle 将包含重新创建 NotImplemented 所需的所有信息，而不依赖于 unpickling 方面，在不尝试序列化 NotImplementedType 的情况下返回 NotImplemented。

Answer 2

copy_reg.pickle 函数明确允许返回字符串的函数。如果它是一个字符串，它将使用该字符串作为名称来“查找”（另请参阅__reduce__ documentation）：

import copy_reg
copy_reg.pickle(type(NotImplemented), lambda x: 'NotImplemented')

这将允许复制（和腌制）NotImplemented：

import copy

>>> type(copy.deepcopy(NotImplemented))              # still a NotImplementedType object
NotImplementedType

>>> copy.deepcopy(NotImplemented) is NotImplemented  # and still the same object!
True