Python - クラス属性をテストするクラスメソッドをデコレートする

Question

私はクラス内の特定の関数を実行する前にユーザーがログインしているかどうかをテストするインターフェイスを設計しようとしています。むしろより：私が持っていることができれば

class UserDoesStuff(object): 
    def doIfLoggedIn(self): 
     if self.checkLogin(): 
      [...do the stuff...] 

私は思っていた。このような何か：

def protected(self): 
    if not self.checkLogin(): 
     raise UserLoginError() 

@protected 
def doIfLoggedIn(self): 
    [...do the stuff...] 

このコースのは動作しませんが、この使用してデコレータを行う方法はありますか？ （余分な引数を持つ最も単純なもの、）

Answer 1

デコレータは、入力としての機能を期待する：

import functools 

def protected(fun): 
    @functools.wraps(fun) 
    def wrapper(self, *args, **kwargs): 
     if not self.checkLogin(): 
      raise UserLoginError() 
     return fun(self, *args, **kwargs) 

    return wrapper # this is what replaces the original method 

@protected 
def doIfLoggedIn(self): 
    ... 

Answer 2

はい、あなたはDjango docs on @login_requiredとhow they do itを見てみましょうことがあります。