python合并多个装饰器小技巧

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

Posted in 未分类

发表评论