Python多个Decorator装饰器的使用

在tornado项目中一些RequestHandler中的get()，post()方法需要验证登陆用户和用户权限，并返回template或者json。

class IndexHandler(RequestHandler):    @render_json    @login_required    @permission_required(10001)    get(self):        pass

对于同一个get()方法使用了多个装饰器，运行的顺序一般是自上而下，但太多个装饰器个人认为还是比较难以理解。

---

Demo

def first_decorator(func):    print('--first--')    def first_wrapper(*args, **kwargs):        print('------first---%s---' % func.__name__)        return func(*args, **kwargs)    return first_wrapperdef second_decorator(func):    print('--second--')    def second_wrapper(*args, **kwargs):        print('------second---%s---' % func.__name__)        return func(*args, **kwargs)    return second_wrapper@first_decorator@second_decoratordef test_func_1():    print('excute %s' % inspect.stack()[0][3])

定义了两个装饰器，并装饰了函数test_func_1()，此时应该输出：

--second----first--------first---second_wrapper---------second---test_func_1---excute test_func_1

拆分装饰器@语法糖以后，等价于。

def test_func_1():    print('excute %s' % inspect.stack()[0][3])first_decorator(second_decorator(test_func_1))()

进一步拆分。

second_wrapper = second_decorator(test_func_1)  # 返回实际是second_wrapper# second()first = first_decorator(second_wrapper)  # second_wrapper 传入first_decorator()first()

实际上首先执行了second_decorator(test_func_1)返回second_wrapper函数，而此时--second--已经打印了。

然后first_decorator()接受之前返回的second_wrapper并返回了first_wrapper函数，而此时--first--已经打印了。

最后执行first_wrapper(second_wrapper)()，先打印出------first---second_wrapper---了，因为此时first_decorator中接受的func是second_wrapper，然后执行func(*args, **kwargs)即second_wrapper()，打印出------second---test_func_1---，最后执行test_func_1()，打印excute test_func_1
(感觉仍有问题，需要重新整理一下思路。)

---

模拟Handler

def login_required(func):    def wrapper(self, *args, **kwargs):        if not self.current_user:            # 未登录情况            raise Exception('please login')        else:            return func(self, *args, **kwargs)    return wrapperdef permission_required(permission_code):    def decorator(func):        def wrapper(self, *args, **kwargs):            if not self.current_user.permission_code == permission_code:                raise Exception('not have permission')            else:                return func(self, *args, **kwargs)        return wrapper    return decoratorclass User():    def __init__(self, name, permission_code=0):        self.name = name        self.permission_code = permission_codeclass BaseHandler():    def __init__(self, current_user=None):        self.current_user = current_user    @login_required    @permission_required(1)    def get(self):        print('get success')        return 'hello'    @permission_required(1)    @login_required    def post(self):        print('post success')        return 'hello'

是否能简化成一个装饰器？

# 简化成一个装饰器？def check_login_permission(permission_code=0):    # 0默认不需要权限    def decorator(func):        def wrapper(self, *args, **kwargs):            if not self.current_user:                raise Exception('please login')            if permission_code and self.current_user.permission_code != permission_code:                raise Exception('not have permission')            return func(self, *args, **kwargs)        return wrapper    return decoratorclass NewBaseHandler():    def __init__(self, current_user=None):        self.current_user = current_user    @check_login_permission(0)    def get(self):        print('get success')        return 'hello'try:    # user_1 = User('JiangW_1', 1)    # new_base_handler = NewBaseHandler(user_1)    new_base_handler = NewBaseHandler(None)    new_base_handler.get()except Exception as e:    print(e)