如何在不修改原始類的情況下刪除instancemethod對象，但不修改原始類

Question

我想持久保留reverend.thomas.Bayes中的對象。當然，如果我嘗試直接醃製這些類中的一個，我得到：

TypeError: can't pickle instancemethod objects 

要解決這個問題，我已經試過聲明兩個功能：

import types 
from itertools import chain 
from copy import copy 
from reverend.thomas import Bayes 

def prepare_bayes_for_pickle(bayes_obj): 
    dic = copy(bayes_obj.__dict__) #I also tried using deepcopy instead of copy 
    for k in dic: 
     if type(k) == types.MethodType: 
      dic.pop(k) 
    return dic 

def reconstruct_bayes_from_pickle(bayes_dic): 
    b = Bayes() 
    # Merge b with bayes_dic, with bayes_dic taking precedence 
    dic = dict(chain(bayes_dic, b)) 
    b.__dict__ = dic 
    return b 

基本上，我嘗試複製對象的__dict__，並嘗試通過測試types.MethodType的類型來刪除instancemethod。

然後我會通過創建一個新Bayes對象，然後bayes_dic合併回來在一起（在它是拆封。）

但是，我還沒有起牀第二種方法又重建對象，因爲我仍然無法醃製從prepare_bayes_for_pickle返回的對象，但未收到原始錯誤。

Answer 1

k是一個關鍵，即屬性/方法名稱。你需要測試屬性本身：

if type(dic[k]) == types.MethodType: 
      ^~~~~~ here 

我更喜歡使用理解;你也應該使用isinstance：

dic = dict((k, v) for k, v in bayes_obj.__dict__ 
      if not isinstance(v, types.MethodType)) 

Answer 2

這聽起來像是得到一個方釘，以適應一個圓孔。如何使用pickle來清理參數，並取消重建reverand.Thomas.Bayes對象？

>>> from collections import namedtuple 
>>> ArgList = namedtuple('your', 'arguments', 'for', 'the', 'reverand') 
>>> def pickle_rtb(n): 
...  return pickle.dumps(ArgList(*n.args)) 
... 
>>> def unpickle_rtb(s): 
...  return reverand.Thomas.Bayes(*pickle.loads(s)) 
... 
>>> s = pickle_rtb(reverand.Thomas.Bayes(1, 2, 3, 4, 5)) # note arguments are a guess 
>>> rtb = unpickle_norm(s) 

受this的啓發SO問題。

Answer 3

更好的解決方案將是您添加__getstate__方法到Bayes類（具有附帶__setstate__）：

import types 
from reverend.thomas import Bayes 

def Bayes__getstate__(self): 
    state = {} 
    for attr, value in self.__dict__.iteritems(): 
     if not isinstance(value, types.MethodType): 
      state[attr] = value 
     elif attr == 'combiner' and value.__name__ == 'robinson': 
      # by default, self.combiner is set to self.robinson 
      state['combiner'] = None 
    return state 

def Bayes__setstate__(self, state): 
    self.__dict__.update(state) 
    # support the default combiner (an instance method): 
    if 'combiner' in state and state['combiner'] is None: 
     self.combiner = self.robinson 

Bayes.__getstate__ = Bayes__getstate__ 
Bayes.__setstate__ = Bayes__setstate__ 

現在Bayes類總是可以進行酸洗並沒有額外的處理拆封。

我確實看到這個類有一個self.cache = {}映射;也許這應該在酸洗時排除？忽略它在__getstate__和self.buildCache()__setstate__如果是這種情況。