如何解析其类存在于不同命名空间（python）中的对象？

Question

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

script = """

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

"""

然后在自己的命名空间dict中执行此脚本：

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 dict中。 有没有办法告诉pickle在哪里找到它正在加载的对象的类？

Answer 1

您实际上可以更进一步，让对象重建为您想要的任何类型。

import pickle
import copy_reg

class myClass(object):
    def __init__(self):
        self.apple = 'banana'

class otherclass(object):
    def __init__(self):
        self.apple = 'existential woe'

def pickle_an_object(o):
    print "pickling %s" % str(o)
    return otherclass, (o.apple,)

copy_reg.pickle(myClass, pickle_an_object)

foo = myClass()

s = pickle.dumps(foo)

del myClass
del otherclass

class otherclass(object):
    def __init__(self, appletype):
        self.apple = 'not %s' % appletype

o2 = pickle.loads(s)

print o2.apple

基本的想法是，你把你的班级打包成各种各样的“特洛伊木马”，它的重建导致了与原来不同的类的实例化。

不要紧 ，什么otherclass在酸洗面包含的内容。 重要的是它存在于与“目标”类相同的模块路径中 - pickle只是将模块名称的字符串表示放入序列化流中。

那么，要详细分解上面代码中发生的事情：

• 我们为myClass注册了一个自定义pickler。 这可以通过copy_reg或__reduce_ex__函数完成。
• 我们的定制Pickler会说“泡菜将此作为一个实例otherclass ”（这是一个虚拟的。你并不需要的“真正的”内容otherclass在酸洗面，因为所有进入泡菜是模块/类名）。
• 我们挑选对象并“通过网络发送”到其他类的真实版本所在的otherclass 。
• 在远程端，使用自定义otherclass函数返回的元组中的数据实例化otherclass 。

Python可以非常强大！

Answer 2

我发现了一个解决方案。 似乎问题是在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)