如何解析其類存在於不同命名空間（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)