装饰器为其他函数增加功能,就是写一个函数,这个函数供其他开发人员调用的的。
装饰器本质上是一个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)