クラス内のメソッドを飾る方法は？

Question

私はクラス内のメソッドをデコレートしようとしていますが、Pythonはエラーを投げています。私のクラスには、次のようになります。

from pageutils import formatHeader 

myPage(object): 
    def __init__(self): 
     self.PageName = '' 

    def createPage(self): 
     pageHeader = self.createHeader() 

    @formatHeader #<----- decorator 
    def createHeader(self): 
     return "Page Header ",self.PageName 

if __name__=="__main__": 
    page = myPage() 
    page.PageName = 'My Page' 
    page.createPage() 

#------- pageutils.py -------------------- 

def formatHeader(fn): 
    def wrapped(): 
     return '<div class="page_header">'+fn()+'</div>' 
    return wrapped 

Pythonは、次のエラー

 
self.createHeader() 
TypeError: wrapped() takes no arguments (1 given) 

私は暇つぶしのですがスローされますか？

Answer 1

Pythonは自動的にクラスインスタンスを参照として渡します。 （すべてのクラスメソッドに見られるself引数）。

あなたが行うことができます：あなたは装飾のない機能（お使いの場合はcreateHeader）に存在する自己パラメータを省略している

def formatHeader(fn): 
    def wrapped(self=None): 
     return '<div class="page_header">'+fn(self)+'</div>' 
    return wrapped 

Answer 2

。

def formatHeader(fn): 
    from functools import wraps 
    @wraps(fn) 
    def wrapper(self): 
     return '<div class="page_header">'+fn(self)+'</div>' 
    return wrapper 

あなたが飾るしたい機能の署名について不明な点がある場合は、次のように、あなたはそれではなく、一般的にすることができます。

def formatHeader(fn): 
    from functools import wraps 
    @wraps(fn) 
    def wrapper(*args, **kw): 
     return '<div class="page_header">'+fn(*args, **kw)+'</div>' 
    return wrapper