在 Python 2 中取消 Python 3 中的类

Question

如果使用协议 2 对 Python 3 类进行 pickle，则它应该可以在 Python 2 中工作，但不幸的是，这会失败，因为某些类的名称已更改。

假设我们有如下调用的代码。

发送方

pickle.dumps(obj,2)

接收方

pickle.loads(atom)

举个具体的例子，如果obj={}，那么给出的错误是：

导入错误：没有名为内置的模块

这是因为 Python 2 使用 __builtin__ 代替。

问题是解决这个问题的最好方法。

Answer 1

此问题是 Python 问题 3675。这个bug实际上在Python 3.11中得到了修复。

如果我们导入：

from lib2to3.fixes.fix_imports import MAPPING

MAPPING 将 Python 2 名称映射到 Python 3 名称。我们希望这是相反的。

REVERSE_MAPPING={}
for key,val in MAPPING.items():
    REVERSE_MAPPING[val]=key

我们可以重写 Unpickler 并加载，

class Python_3_Unpickler(pickle.Unpickler):
    """Class for pickling objects from Python 3"""
    def find_class(self,module,name):
        if module in REVERSE_MAPPING:
            module=REVERSE_MAPPING[module]
        __import__(module)
        mod = sys.modules[module]
        klass = getattr(mod, name)
        return klass

def loads(str):
    file = pickle.StringIO(str)
    return Python_3_Unpickler(file).load()  

然后我们将其称为“loads”而不是 pickle.loads。

这应该可以解决问题。