装饰器
1.什么是装饰器:
器指的是工具,可以定义成函数
装饰指的是为其他事物添加额外的东西点缀
合到一起解释:
装饰器指的定义一个函数,该函数是用来为其他函数添加额外的功能
2.为何要用装饰器:
开放封闭原则
开放:指的是对拓展功能是开放的
封闭:指的是对修改源代码是封闭的
装饰器就是在不修改被装饰器对象源代码以及调用方式的前提下为被装饰对象添加新功能
3.怎么用:
需求:在不修改index函数的源代码以及调用方式的前提下为其添加统计运行时间的功能
def index(x,y):
time.sleep(3)
print('index %s %s'%(x,y))
index(111,222)
index(y=111,x=222)
index(111,y=222)
解决方案一:失败
问题:没有修改被装饰对象的调用方式,但是修改了其源代码
def index(x,y):
start=time.time()
time.sleep(3)
print('index %s %s' %(x,y))
stop = time.time()
print(stop-start)
index(111,222)
解决方案二:失败
问题:没有修改被装饰对象的调用方式,也没有修改了其源代码,并且加上了新功能,但是代码冗余
import time
def index(x,y):
time.sleep(3)
print('index %s %s' %(x,y))
start=time.time()
index(111,222)
stop=time.time()
print(stop-start)
start=time.time()
index(111,222)
stop=time.time()
print(stop-start)
start=time.time()
index(111,222)
stop=time.time()
print(stop-start)
start=time.time()
index(111,222)
stop=time.time()
print(stop-start)
解决方案三:失败
问题:解决了方案二代码冗余问题,但带来一个新问题即函数的调用方式改变了
import time
def index(x,y):
time.sleep(3)
print('index %s %s' % (x,y))
def wrapper():
start=time.time()
index(111,222)
stop=time.time()
print(stop-start)
wrapper()
方案三的优化一:将index的参数写活了
import time
def index(x,y,z):
time.sleep(3)
print('index %s %s %s' %(x,y,z))
def wrapper(*args,**kwargs):
start=time.time()
index(*args,**kwargs)
stop=time.time()
print(stop-start)
wrapper(333,444,555)
wrapper(333,z=555,y=44444)
方案三的优化二:在优化一的基础上把被装饰对象写活了,原来只能装饰index
import time
def index(x,y,z):
time.sleep(3)
print('index %s %s %s' %(x,y,z))
def home(name):
time.sleep(2)
print('welcome %s to home page' %name)
def outter(func):
#func=index的内存地址
def wrapper(*args,**kwargs):
start=time.time()
func(*args,**kwargs)# index的内存地址()
stop=time.time()
print(stop-start)
return wrapper
index = outter(index) # index=wrapper的内存地址
home=outter(home) # home=wrapper的内存地址
home('egon')
home(name='egon')
方案三的优化三:将wrapper做的跟被装饰对象一摸一样,以假乱真
import time
def index(x,y,z):
time.sleep(3)
print('index %s %s %s' %(x,y,z))
def home(name):
time.sleep(2)
print('welcome %s to home page' %name)
def outter(func):
def wrapper(*args,**kwargs):
start=time,time()
res=func(*args,**kwargs)
stop=time.time()
print(stop - start)
return res
return wrappper
偷梁换柱:home这个名字指向的wrapper函数的内存地址
home=outter(home)
res=home('egon')# res = wrapper('egon')
print('返回值--》',res)
大方向:如何在方案三的基础上不改变函数的调用方式
语法糖:让你开心的语法
import time
装饰器
def timmer(func):
def wrapper(*args,**kwargs):
start=time.time()
res=func(*args,**kwargs)
stop=time.time()
print(stop-start)
return res
return wrapper
#在被装饰对象正上方的单独一行写@装饰器名字
#@timmer # index=timmer(index)
def index(x,y,z):
time.sleep(3)
print('index %s %s %s' %(x,y,z))
# @timmer # home=timmer(ome)
def home(name):
time.sleep(2)
print('welcome %s to home page' % name)
index(x=1,y=2,z=3)
home(''egon)
总结无参装饰器模板
def outter(func):
def wrapper(*args,**kwargs):
#1.调用原函数
#2.为其增加新功能
res=func(*args,**kwargs)
return res
return wrapper
def auth(func):
def wrapper(*args,**kwargs):
#1.调用原函数
#2.为其增加新功能
name=input('your name>>: ').strip()
pwd=input('you password>>: ').strip()
if name == 'egon'and pwd == '123':
res=func(*args,**kwargs)
return res
else:
print('账号密码错误')
return wrapper
@auth
def index():
print('from index')
index()