添加裝飾功能的一類

Question

我有一個裝飾功能（簡化版）：

class Memoize: 
    def __init__(self, function): 
     self.function = function 
     self.memoized = {} 
    def __call__(self, *args, **kwds): 
     hash = args 
     try: 
      return self.memoized[hash] 
     except KeyError: 
      self.memoized[hash] = self.function(*args) 
      return self.memoized[hash] 


@Memoize 
def _DrawPlot(self, options): 
    do something... 

現在我想將這個方法添加到esisting學前班。

ROOT.TChain.DrawPlot = _DrawPlot 

當我把這個方法：

chain = TChain() 
chain.DrawPlot(opts) 

我：

self.memoized[hash] = self.function(*args) 
TypeError: _DrawPlot() takes exactly 2 arguments (1 given) 

爲什麼沒有傳播的自我？

Answer 1

問題是，您已經定義了您自己的可調用類，然後嘗試將其用作方法。當你使用一個函數作爲屬性時，作爲一個屬性訪問該函數稱其爲__get__方法來返回除函數本身之外的東西 - 綁定方法。當你有自己的班級而沒有定義__get__時，它只是返回你的實例而不隱式傳遞self。

如果您不熟悉描述符，則在http://docs.python.org/reference/datamodel.html#descriptors上解釋一些描述符。 __get__，__set__和__delete__方法改變了作爲屬性起作用時與對象的交互方式。

---

您可以實現memoize的功能，並使用內置的__get__魔法功能已經

import functools 

def memoize(f): 
    @functools.wraps(f) 
    def memoized(*args, _cache={}): 
     # This abuses the normally-unwanted behaviour of mutable default arguments. 
     if args not in _cache: 
      _cache[args] = f(*args) 
     return _cache[args] 
    return memoized 

或沿

import functools 

class Memoize(object): #inherit object 
    def __init__(self, function): 
     self.function = function 
     self.memoized = {} 
    def __call__(self, *args): #don't accept kwargs you don't want. 
     # I removed "hash = args" because it shadowed a builtin function and 
     # because it was untrue--it wasn't a hash, it was something you intended for 
     # Python to hash for you. 
     try: 
      return self.memoized[args] 
     except KeyError: 
      self.memoized[args] = self.function(*args) 
      return self.memoized[args] 
    def __get__(self, obj, type): 
     if obj is None: #We looked up on the class 
      return self 

     return functools.partial(self, obj) 

注行修改你的類如果你傳入的任何參數都是可變的（這在技術上是不可能的），這兩個都會窒息。這可能適合您的情況，但您也可能想要處理args不可用的情況。