第八章、函数进阶之装饰器02
一、无参装饰器
1. 什么是装饰器？
装饰器指的是为被装饰对象添加功能，因此定义装饰器就是定义一个函数，只不过是该函数是用来为其他函数添加额外的功能
注意：
- 装饰器本身是可以任意调用的对象
- 被装饰的对象也可以是任意可调用了的对象
2. 为什么要用装饰器
装饰器的实现必须遵循两大原则
-
不修改被装饰的源代码
-
不修改被装饰的对象的调用方式
装饰器就是在遵循以上两个原则的前提下被装饰对象添加功能
#真正的装饰器 import time def index(): print('您好啊！') time.sleep(1) def deco(func): # func = 原index '''装饰器''' def f1(): #重新创建后的index start = time.time() func() #原index end = time.time() print(end-start) return f1 index = deco(index) index() #f1
3. 怎么用装饰器
第一种传参方式：
def index():
print('welcome to index')
time.sleep(1)
def time_count(func):
# func = 最原始的index
def wrapper():
start = time.time()
func()
end = time.time()
print(f"func time is {start-end}")
return wrapper
f = time_count(index)
f()
#index = time_count(index) # index为被装饰函数的内存地址，即index = wrapper
#index() # wrapper()
welcome to index
func time is -1.0008289813995361
下面方式是对第一种的再优化
第二种传参函数：包给函数——外包（指的是再定义命名为原函数的名字的变量，被赋予了原函数的内存地址，再调用被装饰后的函数名字（），以后就可以直接用原来函数的名字调用了）
import time
def index():
print('welcome to index')
time.sleep(1)
def time_count(func):
# func = 最原始的index
def wrapper():
start = time.time()
func()
end = time.time()
print(f"{func} time is {start-end}")
return wrapper
# f = time_count(index)
# f()
index = time_count(index) # index为被装饰函数的内存地址，即index = wrapper
index() # wrapper()
welcome to index
func time is -1.0008289813995361
4. 完善装饰器
上述的装饰器，最后调用index()(实际上是被装饰后的index（）)的时候，其实是在调用wrapper()，因此如果原始的index()有返回值的时候，wrapper()函数的返回值应该和index()的返回值相同，也就是说，我们**需要同步原始的index()和wrapper()方法的返回值。 **具体怎么做：
import time
def index():
print('welcome to index')
time.sleep(1)
return 123
def time_count(func):
# func = 最原始的index
def wrapper():
start = time.time()
res = func()
end = time.time()
print(f"{func} time is {start-end}")
return res
return wrapper
index = time_count(index)
res = index()
print(f"res: {res}")
#此处return就已经把原始的index的返回值接收过来了
res = func()
end = time.time()
print(f"{func} time is {start-end}")
return res
welcome to index
<function index at 0x102977620> time is -1.0050289630889893
res: 123
如果原始的home()方法需要传参，那么我们之前的装饰器是无法实现该功能的，由于有timeinner()=home()，所以给timeinner()方法传参即可。
import time
def home(name):
print(f"welcome {name} to home page")
time.sleep(1)
return name
def time_count(func):
# func = 最原始的index
def timeinner(*args, **kwargs):
start = time.time()
res = func(*args, **kwargs)
end = time.time()
print(f"{func} time is {start-end}")
return res
return timeinner
home = time_count(home)#就相当于把time_count()的函数返回值（其实是内部函数的地址）赋值给另一个变量名（其实就是被装饰后的原函数的名字）
res = home('egon')#调用了timeinner("egon")
print(f"res: {res}")
5. 装饰器语法糖
被装饰的函数正上方单独写上@装饰器名
import time
def time_count(func):
# func = 最原始的index
def wrapper():
start = time.time()
func()
end = time.time()
print(f"{func} time is {start-end}")
return wrapper
@time_count
def index():
print('welcome to index')
time.sleep(1)
welcome to index
func time is -1.000335454940796
6. 装饰器模板
#双层装饰器
def deco(func):
def wrapper(*args,**kwargs):
res = func(*args,**kwargs)
return res
return wrapper
#三层装饰器
# 三层装饰器:给双层装饰器加参数
def sanceng (engine):
def outter(func):
def wrapper(*args,**kwargs): # wrapper是未来要运行的函数
#加功能
print(engine)
res = func(*args,**kwargs) #func是被装饰的函数
return res
return wrapper
return outter
@sanceng('db')
def shopping():
print('shopping')
shopping()
二、有参函数
1. 三层闭包
# 三层装饰器:给双层装饰器加参数
def sanceng (engine):
def outter(func):
def wrapper(*args,**kwargs): # wrapper是未来要运行的函数
#加功能
print(engine)
res = func(*args,**kwargs)
print(res)#func是被装饰的函数
return res
return wrapper
return outter
@sanceng('db')
def shopping(x="666"):
print('shopping')
return x
shopping()
db
shopping
666