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。