装饰器为其他函数增加功能,就是写一个函数,这个函数供其他开发人员调用的的。
装饰器本质上是一个Python函数,它可以让其他函数在不需要做任何代码变动的前提下增加额外功能,装饰器的返回值也是一个函数对象。
它经常用于有切面需求的场景,比如:插入日志、性能测试、事务处理、缓存、权限校验等场景。
装饰器是解决这类问题的绝佳设计,有了装饰器,我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。
不带参数的装饰器函数
1 import time 2 def wrapper(func): #装饰器函数 3 def inner(): 4 start=time.time() 5 func() 6 end=time.time() 7 print(end-start) 8 return inner 9 @wrapper #通过装饰器调用, 10 def kkk():#相当于kkk=wrapper(kkk),wrapper(kkk)返回inner,最后为kkk=inner。 11 print('aaaaa') 12 kkk() #因为上面返回了inner,kkk=inner,执行kkk(),相当于执行inner(),此时函数又返回执行内部inner函数,func()函数之前已经传进去。
带参数的装饰器函数
1 def logger(flag = ' '): 2 def show_time(f): 3 def inner(x,y): 4 start = time.time() 5 f(x,y) #这个函数f会回到foo那个功能函数 6 end = time.time() 7 print('speed: %s'%(end-start)) 8 if flag == 'true': 9 print('日志记录') 10 11 return inner #返回inner函数这个变量名 12 return show_time 13 14 @logger('true') #logger仅仅是给了show_time一个参数,返回show_time,还是执行@show_time。foo=logger('true')>>foo=show_time(foo)>>foo=inner 15 def foo(a,b): 16 print(a+b) 17 time.sleep(2) #定时,多少秒后执行代码 18 foo(1,2)