荟萃馆

位置:首页 > 技能 > 装饰

Python如何合并多个装饰器?教你几个小技巧

装饰1.72W

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

<a href=Python如何合并多个装饰器?教你几个小技巧" title="Python如何合并多个装饰器?教你几个小技巧">

复制代码 代码如下:

@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

标签:Python 合并