计算函数执行时间的装饰器(简洁版)
import time
import datetime
def timer(func):
def wrapper(*args, **kwargs):
start_time = time.time()
res = func(*args, **kwargs)
end_time = time.time()
execution_time = (end_time - start_time)*1000
run_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
with open("text.txt", "a") as f:
f.write(str(run_time) + ":" + str(func.__name__) + "函数运行时间为" + str(execution_time) + "\n")
return res
return wrapper
@timer
def print_hi_1(name):
time.sleep(3)
print(f'Hi, {name}')
if __name__ == '__main__':
print_hi('PyCharm')
print_hi_1("python3")