Python合并多个装饰器小技巧
程序员文章站
2023-11-23 17:32:46
django程序，需要写很多api，每个函数都需要几个装饰器，例如
复制代码 代码如下:
@csrf_exempt
@require_POST ...
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
参考：