首页 > 代码库 > @修饰符--python中的装饰器

@修饰符--python中的装饰器

装饰器模式可以在不影响其他对象的情况下,以动态、透明的方式给单个对象添加职责,也能够处理那些可以撤销的职责。经常用于日志记录、性能测试等场合。

想象一下这个很常见的场景,你写了一个方法只提供给以登陆的用户访问(事实上我也是通过django的@login_required才了解到@修饰符的),你可以写以下代码:

 

[python] view plaincopy
  1. def A():  
  2.     if user.is_login():  
  3.         do something  
  4.     else:  
  5.         pass  


这当然没什么问题,但是你又写了一个方法B,也要求只有登录用户可以访问,于是有写了以下代码:

 

 

[python] view plaincopy
  1. def B():  
  2.     if user.is_login():  
  3.         do something  
  4.     else:  
  5.         pass  


问题出来了,你在复制粘贴,如果不是两个方法而是一堆方法,你可能就有点受不了啦。当然聪明的你可以想出这个方法:

 

 

[python] view plaincopy
  1. def A():  
  2.     pass  
  3. def B():  
  4.     pass  
  5. def login_required(fn):  
  6.     def ff():  
  7.         if user.is_login():  
  8.             fn()  
  9.         else:  
  10.             pass  
  11.     return ff  
  12.   
  13. A = login_required(A)  
  14. B = login_required(B)  

 

 

你可能没有想到,对于这么好用的东西,python优雅的支持,这就是@修饰符 

 

[python] view plaincopy
  1. def login_required(fn):  
  2.     def ff():  
  3.         if user.is_login():  
  4.             fn()  
  5.         else:  
  6.             pass  
  7.     return ff  
  8.  
  9. @login_required  
  10. def A():  
  11.     pass  
  12. @login_required  
  13. def B():  
  14.     pass  


在方法A上边写一个@修饰符,调用方法A的时候会调用修饰符后边的方法B,方法B以A方法为参数,而且需要返回一个可调用的对象,这个可调用的对象会使用A方法提供的参数执行。看这个例子:

 

 

[python] view plaincopy
  1. #!/usr/bin/env python  
  2.   
  3. def a(fn):  
  4.     print ‘a‘  
  5.     def d(st):  
  6.         print st+‘d‘  
  7.     return d  
  8.   
  9. def b(fn):  
  10.     print ‘b‘  
  11.     return fn  
  12.  
  13. @a  
  14. @b  
  15. def c(st):  
  16.     print st  
  17.       
  18. c(‘c‘)  


输出结果:bacd

 

我们调用c(‘c‘)的时候会先调用b(c),b(c)打印字符"b"然后返回c,然后再调用a(c),a(c)打印字符"a",然后返回方法d,然后再执行d(‘c‘),打印cd。