Python中装饰器兼容加括号和不加括号的写法详解
使用Django的时候，我发现一个很神奇的装饰器：@login_required，这是控制一个view的权限的，比如一个视图必须登录才可以访问，可以这样用：
@login_required defmy_view(request): ... returnrender(...)
同时，如果要达到这样一种效果：如果用户没有登录，那么就把用户重定向到登录界面，可以这样用：
@login_required(login_url='/accounts/login/') defmy_view(request): ... returnrender(...)
所以这个装饰器可以带括号写，又可以不带括号写。很神奇有没有。正常的接收参数的装饰器，就算没参数也应该写成@login_required的
好奇去查了一下，在stackoverflow找到一种实现，挺有意思的。先晒出答案：
defdoublewrap(f): ''' adecoratordecorator,allowingthedecoratortobeusedas: @decorator(with,arguments,and=kwargs) or @decorator ''' @wraps(f) defnew_dec(*args,**kwargs): iflen(args)==1andlen(kwargs)==0andcallable(args[0]): #actualdecoratedfunction returnf(args[0]) else: #decoratorarguments returnlambdarealf:f(realf,*args,**kwargs) returnnew_dec
使用起来很简单，只要给装饰器用@doublewrap装饰一下，这个装饰器就支持写括号和不写括号两种写法了。
deftest_doublewrap(): fromutilimportdoublewrap fromfunctoolsimportwraps @doublewrap defmult(f,factor=2): '''multiplyafunction'sreturnvalue''' @wraps(f) defwrap(*args,**kwargs): returnfactor*f(*args,**kwargs) returnwrap #trynormal @mult deff(x,y): returnx+y #tryargs @mult(3) deff2(x,y): returnx*y #trykwargs @mult(factor=5) deff3(x,y): returnx-y assertf(2,3)==10 assertf2(2,5)==30 assertf3(8,1)==5*7
原理也不难，只有短短不到10行代码。
装饰器我们都知道，是用来处理一个函数，返回一个新的函数的（如果你不理解装饰器，可以看一下这个经典的解释）。
new_func=decorator(func)
我们使用的，就是被装饰器装饰的新函数了。装饰器只是一个语法糖，其实它也是一个函数，给它传入一个函数作为参数，就返回一个新的函数。那么既然装饰器也是一个函数，我们就可以用装饰器装饰这个函数。也就是，“装饰器的装饰器”。
装饰器第一个参数肯定是原函数，如果装饰器可以接收参数的话，要么第一个参数是原函数，后面跟别的参数；要么就只有原函数一个参数。所以，我们这个“装饰器的装饰器”做的事情就是，判断装饰器接收的参数，如果只有一个并且第一个参数是可调用的（callable），那么这就是一个无参数的装饰器（不需要加括号）。如果还有别的参数，就返回一个生成装饰器的函数(decorator_maker)。
装饰器是一个函数。装饰器被装饰过之后，这个装饰器运行之前就会先运行装饰器的装饰器的代码，也就是我们的doublewrapp。然后返回值可能是一个装饰器，也可能是一个装饰器的maker（有参数的装饰器），然后装饰器再执行，装饰原函数。
这里有点绕，因为本来装饰器里面一般就会有三四层函数了，（maker,decorator,wrapper,realfunc），再加上一个装饰器的装饰器，会有点理解困难。如果理解不了，最好不要对着网上的博文（包括本文）企图格物致知了，多去看看代码，多写一写。
总结
以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作能带来一定的帮助，如果有疑问大家可以留言交流，谢谢大家对毛票票的支持。
参考资料
HowtocreateaPythondecoratorthatcanbeusedeitherwithorwithoutparameters?