Python：使用“copyreg”为已有减速器的类型定义减速器

Question

（请记住，我正在使用 Python 3 工作，因此解决方案需要在 Python 3 中工作。）

我想使用 copyreg 模块来教 Python 如何 pickle 函数。当我尝试这样做时，_Pickler 对象仍会尝试使用 save_global 函数来 pickle 函数。 （这不适用于未绑定的方法，这就是这样做的动机。）

似乎 _Pickler 首先尝试在自己的 dispatch 中查找在查看 copyreg.dispatch_table 之前要腌制的对象。我不确定这是否是故意的。

我有什么方法可以告诉Python使用我提供的reducer来pickle函数吗？

Answer 1

以下 hack 似乎可以在 Python 3.1 中工作...：

import copyreg
def functionpickler(f):
  print('pickling', f.__name__)
  return f.__name__

ft = type(functionpickler)
copyreg.pickle(ft, functionpickler)

import pickle
pickle.Pickler = pickle._Pickler
del pickle.Pickler.dispatch[ft]

s = pickle.dumps(functionpickler)
print('Result is', s)

其中，两行 hack 是：

pickle.Pickler = pickle._Pickler
del pickle.Pickler.dispatch[ft]

您需要删除函数类型的 dispatch 条目，因为否则它会抢占 copyreg 注册；我认为您无法在 C 编码的 Pickler 上执行此操作，因此您需要将其设置为 Python 编码的 Pickler。

使用您自己的类对 _Pickler 进行子类化，从而生成自己的 dispatch （复制父级并删除函数类型的条目），这样就不会那么麻烦了。 ，然后专门使用您的子类（及其转储方法）而不是 pickle.dump；然而，与泡菜本身的猴子修补相比，它也不太方便。