装饰器:
1、装饰器的作用:
对函数进行装饰,可以在不改变函数调用的情况下,添加一些额外的功能。
2、原则:
目前已经学了两个原则,一是依赖倒置原则,二是开放封闭原则,对扩展开放,对修改封闭。
3、装饰器的固定模式:
1>没有返回值的情形:
1 # 如何在正常调用函数的情况下,添加额外功能呢? 2 import time 3 4 5 def timer(f): 6 def inner(): 7 f() 8 return inner 9 10 11 @timer 12 def func(): 13 time.sleep(0.01) 14 print("hello world") 15 16 17 # 加上@timer后,相当于这句话:func = timer(func) 18 func() # 此时已经实现了正常调用函数
2>函数有返回值的情形:
1 import time 2 3 4 def timer(f): 5 def inner(): 6 r = f() 7 return r 8 return inner 9 10 11 @timer 12 def func(): 13 time.sleep(0.01) 14 print("hello world") 15 return "love" 16 17 18 ret = func() # func = timer(func) 19 print(ret)
3>函数带参数的情形:
1 import time 2 3 4 def timer(f): 5 def inner(*args, **kwargs): 6 ret = f(*args, **kwargs) 7 return ret 8 return inner 9 10 11 @timer 12 def func(a, b): 13 time.sleep(0.01) 14 print("hello", a, b) 15 return "love" 16 17 18 res = func(1, 2) 19 print(res)
4>固定模式:
1 def wrapper(func): # func为被装饰的函数,wrapper为装饰器函数 2 def inner(*args, **kwargs): 3 """函数装饰之前做的事情""" 4 ret = func(*args, **kwargs) 5 """函数装饰之后做的事情""" 6 return ret 7 return inner 8 9 10 @wrapper 11 def func(*args, **kwargs): 12 pass 13 14 15 res = func() 16 print(res)
4、补充两个小知识点:
1>拆包和组包:*的应用
1 def func(*args): 2 # for i in args: 3 # print(i) 4 print(*args) # [1,2,3] 5 6 7 l = [1,2,3] 8 func(l)
2>三元运算符:
1 def my_max(a, b): 2 return a if a > b else b 3 4 print(my_max(2, 5))