一、 装饰器
装饰器就是闭包函数的一种应用场景
什么是闭包函数?我们再来回忆一下:
闭包函数:
定义在函数内部的函数,并且该函数包含对外部函数作用域(强调:对全局作用域名字的引用不算闭包)名字的引用,该函数称为闭包函数
说到作用域我们再回忆下:
作用域:
全局范围:内置+全局
全局有效,全局存活
局部范围:局部
局部有效,局部存活
作用域关系是在函数定义阶段就规定死,与调用位置无关,
也就是说,无论函数在哪儿调用,都必须回到当初定义函数时的位置找作用域关系
一 、为何要用装饰器
#开放封闭原则:
#软件一旦上线后,就应该遵循开放封闭原则,对修改源代码是封闭的,对功能的扩展是开放的,也就是我们必须找到一种解决方法:
#能够在不修改一个功能源代码以及调用方式的前提下,为其加上新功能.
二 、什么是装饰器
#装饰器他人的器具,本身可以是任意可调用对象,被装饰者也可以是任意可调用对象。
#强调装饰器的原则:
#1 、不修改被装饰对象的源代码
#2 、不修改被装饰对象的调用方式
#装饰器的目标:
#在遵循1和2的前提下,为被装饰对象添加上新功能
三 、装饰器的使用
#我们现在给下面函数增加一个运行时间的功能
import time
def index():
time.sleep(3)
print('welcome to index page')
#修改一
def index():
start_time=time.time()
time.sleep(3)
print('welcome to index page')
stop_time=time.time()
print('run time is %s' %(stop_time-start_time))
index() #功能实现
#评语:直接改源代码,这么搞被开了。。。
#再来一位童靴,来实现功能
#修改二
import time
def index():
time.sleep(1)
print('welcome to index page')
start_time=time.time()
index()
stop_time = time.time()
print('run time is %s' % (stop_time - start_time))
#评语:有好多函数要实现这个功能,写N遍变这个代码,
#后期维护一脸懵逼,还是被开。。。
#再来一位童靴,功能重复实现看我用强大的函数
#修订三:
import time
def index():
time.sleep(3)
print('welcome to index page')
def wrapper(func): #func=index
start_time=time.time()
func() #index()
stop_time = time.time()
print('run time is %s' % (stop_time - start_time))
wrapper(index)
#评语:修改了原函数的调用方式,依然被开。。。
#终于来了位小牛的童靴,函数的值可以返回,然后我再把重新定义index
#修订四:
import time
def index():
time.sleep(3)
print('welcome to index page')
def outter(func): #func=最原始的index
# func=最原始的index
def wrapper():
start_time=time.time()
func()
stop_time=time.time()
print(stop_time-start_time)
return wrapper
index=outter(index) # 新的index=wrapper
index() #wrapper() 功能基本实现
#评语:在原值没有返回值是没问题,但是有返回值的情况下,这么搞就会发现返回的是None
#这次小牛牛童靴路过看到这情况,say这么搞!
#修订五
import time
def index():
time.sleep(1)
print('welcome to index page')
return 123 #假使这里返回123 返回值可以是任意类型
#==============下面就是装饰器
def timmer(func):
#func=最原始的index
def wrapper(*args,**kwargs): #可变长参数
start_time=time.time()
res=func(*args,**kwargs) #调用最原始的index
stop_time=time.time()
print(stop_time-start_time)
return res #index()运行的返回值
return wrapper
index=timmer(index) # 新的index=wrapper
print(index()) #功能已经实现,返回值123
#评语:这里的装饰的功能已经实现,返回的值也得到,小牛牛不是白叫的
#天上五彩红光,大牛童靴出现!各位童靴火速围观!
import time
def timmer(func):
def wrapper(*args,**kwargs):
start_time=time.time()
res=func(*args,**kwargs)
stop_time=time.time()
print(stop_time-start_time)
return res
return wrapper
@timmer #index=timmer(index) 装饰器的标准格式!
def index():
time.sleep(1)
print('welcome to index page')
return 123
@timmer # home=timmer(home)
def home(name):
time.sleep(2)
print('welcome %s to home page' %name)
# index()
#home('egon')
#评语:大牛就是大牛!大牛say:教你们绝技,不会的童靴可以照下面的模板实现:
#无参装饰器模板
def outer(func): #outer,inner名字功能随意
def inner(*args,**kwargs):
res=func(*args,**kwargs)
return res
return inner
@outer #装饰器要在装饰函数的上方
def duoduo():
pass
四 、装饰器语法
#被装饰函数的正上方,单独一行
@deco1
@deco2
@deco3
def foo():
pass
#foo=deco1(deco2(deco3(foo)))
#这里的思想就是最上面装饰器,装饰的下面所有的函数(deco2,deco3,foo)
#然后deco2装饰(deco3,foo),最后deco3装饰foo
#功能的不同,放的顺序也要注意,不然装饰的效果可能实现的就不对了!
五、多个装饰器的使用:
import time
current_user={
'username':None,
# 'login_time':None
}
def auth(func):
# func=index
def wrapper(*args,**kwargs):
if current_user['username']: #这里是认证过的,下次就不用认证
print('已经登陆过了')
res=func(*args,**kwargs)
return res
uname=input('用户名>>: ').strip()
pwd=input('密码>>: ').strip()
if uname == 'egon' and pwd == '123':
print('登陆成功')
current_user['username']=uname
res=func(*args,**kwargs)
return res
else:
print('用户名或密码错误')
return wrapper
def timmer(func):
def wrapper(*args,**kwargs):
start_time=time.time()
res=func(*args,**kwargs)
stop_time=time.time()
print(stop_time-start_time)
return res
return wrapper
@timmer # timmer 统计的是auth+index的执行时间
@auth #我们只装饰index的话就要把@timmer紧跟index
def index():
time.sleep(1)
print('welcome to index page')
return 123
@auth
@timmer #这里统计的就是home运行的时间
def home(name):
time.sleep(2)
print('welcome %s to home page' %name)
#index()
#home("duoduo")
六、有参数的装饰器的使用:
import time
current_user={
'username':None,
# 'login_time':None
}
def auth(engine): #道理还是那个道理,在外面包了一层engine的值
# engine='file' #这个值外面穿什么进来就是什么
def auth2(func):
# func=index
def wrapper(*args,**kwargs):
if engine == 'file':
if current_user['username']:
print('已经登陆过了')
res=func(*args,**kwargs)
return res
uname=input('用户名>>: ').strip()
pwd=input('密码>>: ').strip()
if uname == 'egon' and pwd == '123':
print('登陆成功')
current_user['username']=uname
res=func(*args,**kwargs)
return res
else:
print('用户名或密码错误')
elif engine == 'mysql': #engine 值得判断情况
print('基于MyQL的认证')
elif engine == 'ldap':
print('基于LDAP的认证')
return wrapper
return auth2 #这里也要返回auth2的内存地址
@auth('ldap') #@auth2 #index=auth2(index) #index=wrapper
def index():
time.sleep(1)
print('welcome to index page')
return 123
index() # wrapper()