Python 装饰器 多装饰器同时装饰一个函数 多参数函数
装饰器是在不修改源代码的情况下,使用装饰器增加原函数的功能。
在软件开发中有一个原则——"开放-封闭",简单地说就是已经实现的功能不允许被修改,但可以被扩展。
- 封闭:已经实现的功能不可以被修改
- 开放:对现有的功能可以扩展
这个小实验写了三个装饰器和两个普通函数,对比可以得出装饰器运行的过程
import time
def dec1(func):#装饰器1 参数为被装饰的函数func
print('dec1', func.__name__)#打印装饰器的名字及被装饰的函数名
def wrapper1(*args, **kwargs):
start_time = time.time()
func(*args, **kwargs)#可同时传入多个参数,是具体函数而定
end_time = time.time()
print('dec1', end_time - start_time)
return wrapper1
#可以根据参数func的参数,定制、添加原函数的功能。
def dec2(func):#装饰器2
print('dec2', func.__name__)
def wrapper2(*args, **kwargs):
start_time = time.time()
func(*args, **kwargs)#可同时传入多个参数,是具体函数而定
end_time = time.time()
print('dec2', end_time - start_time)
return wrapper2
def dec3(func):#装饰器3
print('dec3', func.__name__)
def wrapper3(*args, **kwargs):
start_time = time.time()
func(*args, **kwargs)#可同时传入多个参数,是具体函数而定
end_time = time.time()
print('dec3', end_time - start_time)
return wrapper3
'''
当多个装饰器装饰同一个函数时,装饰器的运行顺序是从下到上的,比如下面这个func函数,装饰器运行的顺序应当是dec3->dec2->dec1
'''
@dec1
@dec2
@dec3
def func(a, b):#函数func,同时被3个装饰器装饰
print(a, b)
@dec2
def func1(a, b, c):#函数func1 与函数func 对比 二者参数的个数不同
print(a, b, c)
if __name__ == '__main__':
func('1', '2')
func1(1, 2, 3)
运行的结果
dec3 func
dec2 wrapper3
dec1 wrapper2
dec2 func1
1 2
dec3 0.0
dec2 0.0
dec1 0.0
1 2 3
dec2 0.0
Process finished with exit code 0
分析函数func的运行过程
- 传入dec3的方法是func
- 传入dec2的方法是运行dec3返回的wrapper3
- 传入dec1的方法是运行dec2返回的wrapper2
- 执行函数func
- 执行dec3后续代码
- 执行dec2后续代码
- 执行dec1后续代码
对于带返回值的函数,在写装饰器时要返回函数的返回值。如下面这段代码
def dec1(func):#装饰器1 参数为被装饰的函数func
print('dec1', func.__name__)#打印装饰器的名字及被装饰的函数名
def wrapper1(*args, **kwargs):
start_time = time.time()
res = func(*args, **kwargs)#可同时传入多个参数,是具体函数而定
end_time = time.time()
print('dec1', end_time - start_time)
return res#对于有返回值的函数要加返回值
return wrapper1
有参装饰器模板:
def decorator(parameter):# 有参装饰器
def dec1(func):# 装饰器1 参数为被装饰的函数func
print('dec1', func.__name__)# 打印装饰器的名字及被装饰的函数名
def wrapper1(*args, **kwargs): # 可以调用decorator传入的参数parameter
start_time = time.time()
res = func(*args, **kwargs)# 可同时传入多个参数,是具体函数而定
end_time = time.time()
print('dec1', end_time - start_time)
return res # 对于有返回值的函数要加返回值
return wrapper1
return dec1