import random
import time
from functools import wraps
# 定义一个装饰器函数
def record_time(func):
# 里面需要定义一个装饰函数
@wraps(func) # 它本身也是一个装饰器,放在wrapper函数上,用于保留被装饰前的函数,需要取消装饰器时使用__wrapped__属性进行取消
def wrapper(*args, **kwargs): # 定义的函数不会立即执行,只有被调用的时候才会执行
start = time.time()
result = func(*args, **kwargs)
# print(args)
end = time.time()
print(f'{func.__name__}执行时间:{end - start:.3f}秒')
return result
return wrapper
# 语法糖,通过装饰器语法糖为download和upload函数添加装饰器,被装饰后的函数其实是record_time中返回的wrapper函数,调用它就相当于在调用wrapper函数
@record_time
def download(filename):
print(f"开始下载{filename}")
time.sleep(random.randint(2, 6))
print(f"关闭文件{filename}")
@record_time
def upload(filename):
print(f"上传文件{filename}")
time.sleep(random.randint(2, 6))
print(f"{filename}上传完成")
download('你是谁')
download.__wrapped__('我是cs') #取消装饰器
经过语法糖装饰后,调用被装饰的函数就相当于调用wrapper函数,使用wrpas装饰器可以保留被装饰前的函数。