python - 如何取消其类存在于不同命名空间(python)中的对象？

Question

如果我有一个定义类的脚本:

script = """

class myClass:
    def __init__(self):
        self.name = 'apple'
        self.color = 'green'

"""

然后在它自己的命名空间字典中执行这个脚本:

NS = {}
exec script in NS

然后创建该类的实例并对其进行 pickle:

a = NS['myClass']()

import pickle

save = pickle.dumps(a)

现在，如果我尝试解开它:

load = pickle.loads(save)

我得到了错误

AttributeError: 'module' object has no attribute 'myClass'

我想这是行不通的，因为 python 不知道在哪里可以找到 myClass 来重建对象。但是 myClass 确实存在于 NS 字典中。有没有办法告诉 pickle 在哪里可以找到它正在加载的对象的类？

Answer 1

我发现了一个解决方案。看来问题是在 dict 中执行代码会阻止 python 确定类的定义位置。解决方案是创建一个空模块，执行模块中的代码，然后将模块添加到 sys.modules 以便 python 知道它。

script = """
class myClass:
    def __init__(self):
        self.name = 'apple'
        self.color = 'green'
"""

import imp, sys

moduleName = 'custom'

module = imp.new_module(moduleName)

exec script in module.__dict__

sys.modules[moduleName] = module

现在可以对类的实例进行 pickle 和 unpickle:

import pickle
a = module.myClass()
s = pickle.dumps(a)
b = pickle.loads(s)