python——装饰器
一、简单的装饰器
1.为什么要使用装饰器
装饰器的功能:在不修改原函数及其调用方式的情况下对原函数功能进行扩展
装饰器的本质:就是一个闭包函数
来看一个简单的装饰器:实现计算每个函数的执行时间的功能
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
import time def wrapper(func): def inner(): start=time.time() func() end=time.time() print(end-start) return inner def hahaha(): time.sleep(1) print('aaaaa') hahaha=wrapper(hahaha) hahaha()
2、语法糖
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
import time def wrapper(func): def inner(): start=time.time() func() end=time.time() print(end-start) return inner @wrapper def kkk():#相当于kkk=wrapper(kkk) print('aaaaa') kkk()
带参数的装饰器
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
import time def timer(func): def inner(*args,**kwargs): start = time.time() re = func(*args,**kwargs) end=time.time() print(end- start) return re return inner @timer #==> func1 = timer(func1) def func1(a,b): print('in func1') print(a,b) @timer #==> func1 = timer(func1) def func2(a): print('in func2 and get a:%s'%(a)) return 'fun2 over' func1(1,2) print(func2('aaaaaa'))
带返回值的装饰器
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
import time def timer(func): def inner(*args,**kwargs): start = time.time() re = func(*args,**kwargs) end=time.time() print(end - start) return re return inner @timer #==> func1 = timer(func1) def jjj(a): print('in jjj and get a:%s'%(a)) return 'fun2 over' jjj('aaaaaa') print(jjj('aaaaaa'))
二、开放封闭原则
1.对扩展是开放的
2.对修改是封闭的
三、装饰器的固定结构
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
import time def wrapper(func): # 装饰器 def inner(*args, **kwargs): '''函数执行之前的内容扩展''' ret = func(*args, **kwargs) '''函数执行之前的内容扩展''' return ret return inner @wrapper # =====>aaa=timmer(aaa) def aaa(): time.sleep(1) print('fdfgdg') aaa()
四、带参数的装饰器
带参数的装饰器:就是给装饰器传参
作用:就是当加了很多装饰器的时候,现在忽然又不想加装饰器了,想把装饰器给去掉了,但是那么多的代码,一个一个的去闲的麻烦,那么,我们可以利用带参数的装饰器去装饰它,这就他就像一个开关一样,要的时候就调用了,不用的时候就去掉了。给装饰器里面传个参数,那么那个语法糖也要带个括号。在语法糖的括号内传参。在这里,我们可以用三层嵌套,弄一个标识为去标识。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
# 带参数的装饰器:(相当于开关)为了给装饰器传参 # F=True#为True时就把装饰器给加上了 F=False#为False时就把装饰器给去掉了 def outer(flag): def wrapper(func): def inner(*args,**kwargs): if flag: print('before') ret=func(*args,**kwargs) print('after') else: ret = func(*args, **kwargs) return ret return inner return wrapper @outer(F)#@wrapper def hahaha(): print('hahaha') @outer(F) def shuangwaiwai(): print('shuangwaiwai') hahaha() shuangwaiwai()
五、多个装饰器装饰一个函数
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
def qqqxing(fun): def inner(*args,**kwargs): print('in qqxing: before') ret = fun(*args,**kwargs) print('in qqxing: after') return ret return inner def pipixia(fun): def inner(*args,**kwargs): print('in qqxing: before') ret = fun(*args,**kwargs) print('in qqxing: after') return ret return inner @qqqxing @pipixia def dapangxie(): print('饿了吗') dapangxie() ''' @qqqxing和@pipixia的执行顺序:先执行qqqxing里面的 print('in qqxing: before'),然后跳到了pipixia里面的 print('in qqxing: before') ret = fun(*args,**kwargs) print('in qqxing: after'),完了又回到了qqqxing里面的 print('in qqxing: after')。所以就如下面的运行结果截图一样 '''
六、统计多少个函数被装饰了的小应用
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
统计多少个函数被我装饰了 l=[] def wrapper(fun): l.append(fun)#统计当前程序中有多少个函数被装饰了 def inner(*args,**kwargs): # l.append(fun)#统计本次程序执行有多少个带装饰器的函数被调用了 ret = fun(*args,**kwargs) return ret return inner @wrapper def f1(): print('in f1') @wrapper def f2(): print('in f2') @wrapper def f3(): print('in f3') print(l)