Python合并多个装饰器小技巧

时间:2021-11-30 21:04:39

django程序,需要写很多api,每个函数都需要几个装饰器,例如

复制代码 代码如下:

@csrf_exempt 
@require_POST 
def  foo(request): 
    pass 


既然那么多个方法都需要写2个装饰器,或者多个,有啥办法把多个合并成一行呢?
上面的函数执行过程应该是

复制代码 代码如下:

csrf_exempt(require_POST(foo)) 


修改成

复制代码 代码如下:

def compose(*funs): 
    def deco(f): 
        for fun in reversed(funs): 
            f = fun(f) 
        return f 
    return deco

 
函数改写成

复制代码 代码如下:

@compose(csrf_exempt, require_POST) 
def foo(request): 
    pass 

 

参考:
Can I combine two decorators into a single one in Python