python中的decorator的作用

2018-07-27 06:28:25来源:博客园 阅读 ()

新老客户大回馈,云服务器低至5折

1、概念

装饰器(decorator)就是:定义了一个函数,想在运行时动态增加功能,又不想改动函数本身的代码。可以起到复用代码的功能,避免每个函数重复性编写代码,简言之就是拓展原来函数功能的一种函数。在python中,装饰器(decorator)分为函数装饰器类装饰器两种。python中内置的@语言就是为了简化装饰器调用。

列出几个装饰器函数:

打印日志:@log

检测性能:@performance

数据库事务:@transaction

URL路由:@post('/register')

2、使用方法

(1)无参数decorator

编写一个@performance,它可以打印出函数调用的时间。

 1 import time
 2 
 3 def performance(f):
 4     def log_time(x):
 5         t1 = time.time()
 6         res = f(x)
 7         t2 = time.time()
 8         print 'call %s() in %fs' %(f.__name__,(t2 - t1))
 9         return res
10     return log_time
11 
12 @performance
13 def factorial(n):
14     return reduce(lambda x,y : x*y,range(1,n+1))
15 
16 print factorial(10)

运行结果:

 1 call factorial() in 0.006009s 2 3628800 

运行原理:

此时,factorial就作为performance的函数对象,传递给f。当调用factorial(10)的时候也就是调用log_time(10)函数,而在log_time函数内部,又调用了f,这就造成了装饰器的效果。说明f是被装饰函数,而x是被装饰函数的参数。

(2)带参数decorator

请给 @performace 增加一个参数,允许传入's'或'ms'。

 1 import time
 2 
 3 def performance(unit):
 4     def perf_decorator(f):
 5         def wrapper(*args, **kw):
 6             t1 = time.time()
 7             r = f(*args, **kw)
 8             t2 = time.time()
 9             t = (t2 - t1)*1000 if unit =='ms' else (t2 - t1)
10             print 'call %s() in %f %s'%(f.__name__, t, unit)
11             return r
12         return wrapper
13     return perf_decorator
14 
15 @performance('ms')  
16 def factorial(n):
17     return reduce(lambda x,y: x*y, range(1, n+1))
18 
19 print factorial(10)

运行结果:

 1 call factorial() in 9.381056 ms 2 3628800 

运行原理:

它的内部逻辑为factorial=performance('ms')(factorial);

这里面performance('ms')返回是perf_decorator函数对象,performance('ms')(factorial)其实就是perf_decorator(factorial),然后其余的就和上面是一样的道理了。

 

标签:

版权申明:本站文章部分自网络,如有侵权,请联系:west999com@outlook.com
特别注意:本站所有转载文章言论不代表本站观点,本站所提供的摄影照片,插画,设计作品,如需使用,请与原作者联系,版权归原作者所有

上一篇:python全栈开发-面向对象-进阶2

下一篇:Python打印简单三角形