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。