装饰器:本质就是函数
原则:
1.不修改被装饰函数的源代码
2.不修改被装饰函数的调用方式
装饰器的储备知识:
装饰器=高阶函数+函数嵌套+闭包
装饰器的架子:
import time
#装饰器的架子
def timmer(func):
def wrapper():
start_time=time.time()#新加的功能
func()#就是在运行test()
stop_time=time.time()#新加的功能
print('运行时间是%s'%(stop_time-start_time))
return wrapper
#被装饰的函数
def test():
time.sleep(3)
print("test函数运行完毕")
#被装饰的函数的调用方式被修改了,修改为res()
#res=timmer(test)#返回的是wrapper的地址
#res()#执行的是wrapper()
#重新赋值,被调用函数的调用方式重新修改为test()
test=timmer(test)
test()
但是这样需要每次都做赋值操作test=timmer(test),用语法糖来解决这个问题
@timmer:
@timmer 就相当于test=timmer(test)
import time
#装饰器的架子
def timmer(func):
def wrapper():
start_time=time.time()#新加的功能
func()#就是在运行test()
stop_time=time.time()#新加的功能
print('运行时间是%s'%(stop_time-start_time))
return wrapper
#被装饰的函数
@timmer #test=timmer(test)
def test():
time.sleep(3)
print("test函数运行完毕")
test()
带有返回值的装饰器(函数闭包加上返回值):
import time
#装饰器的架子
def timmer(func):
def wrapper():
start_time=time.time()#新加的功能
res=func()#就是在运行test()
stop_time=time.time()#新加的功能
print('运行时间是%s'%(stop_time-start_time))
return res
return wrapper
#被装饰的函数
@timmer #test=timmer(test)
def test():
time.sleep(3)
print("test函数运行完毕")
return '这是test的返回值'
res=test()
print(res)
带有参数的装饰器:
import time
#装饰器的架子
def timmer(func):
def wrapper(a,b):
start_time=time.time()#新加的功能
func(a,b)#就是在运行test()
stop_time=time.time()#新加的功能
print('运行时间是%s'%(stop_time-start_time))
return wrapper
#被装饰的函数
@timmer #test=timmer(test)
def test(a,b):
print("result is %d" % (a + b))
time.sleep(3)
print("test函数运行完毕")
test(3,4)
带有不定参数的装饰器:
import time
#装饰器的架子
def timmer(func):
def wrapper(*args, **kwargs):
start_time=time.time()#新加的功能
func(*args, **kwargs)#就是在运行test()
stop_time=time.time()#新加的功能
print('运行时间是%s'%(stop_time-start_time))
return wrapper
#被装饰的函数
@timmer #test=timmer(test)
def test1(a,b):
print("result is %d" % (a + b))
time.sleep(3)
print("test函数运行完毕")
@timmer
def test2(a,b,c):
print("result is %d" % (a + b + c))
time.sleep(3)
print("test函数运行完毕")
test1(3,4)
test2(3,4,5)
多个装饰器:
def dec1(func):
print("1111")
def one():
print("2222")
func()
print("3333")
return one
def dec2(func):
print("aaaa")
def two():
print("bbbb")
func()
print("cccc")
return two
@dec1 #test=dec1(@dec2 test())
@dec2 #test=dec1(dec2(test))
def test():
print("test test")
test()
@dec1到定义test函数是装载装饰器的过程,相当于执行了test=dect1(dect2(test)),此时先执行dect2(test),结果是输出aaaa、将func指向函数test、并返回函数two,然后执行dect1(two),结果是输出1111、将func指向函数two、并返回函数one,然后进行赋值