4 案例4:测试程序运行效率
4.1 问题
创建deco.py脚本,要求如下:
有个程序包含多个函数
程序运行耗时较长
为了确定哪个函数是瓶颈,需要计算出每个函数运行时间
要求使用装饰器实现
1234
4.2 方案
如果一个程序有多个函数,查看每个函数运行耗时时长,这时如果修改每个函数,为它加上计时的功能,我们需要耗时去了解每个函数的功能,思考如何修改去增加计时功能,这样会相当繁琐,为避免这种现象出现,我们利用装饰器函数在不变动其他函数基础上,新增计时功能,在每个函数前添加语法糖(调用装饰器函数)即可,实施方法如下:
1.首先,定义一个函数loop(),该函数可以拥有任何功能,这里将for循环输出的数字添加到空列表中,并让进程挂起0.3s的时间
2.定义装饰器函数计算loop()函数运行耗时时长,装饰器函数deco,该函数的传入参数是loop (即被装饰函数),返回参数是内层函数。这里的内层函数-timeit,其实就相当于闭包函数,它起到装饰给定函数的作用
3.@ deco这个语法相当于执行loop = deco(loop),为loop函数装饰并返回
4.装饰器函数在调用loop函数前后各返回一个当前时间,返回两个时间差即计算出运行耗时。
5.装饰器函数参数是你要装饰的函数名,装饰器函数返回是装饰完的函数名
需要要注意的是:为了不破坏原函数的逻辑,我们要保证内层函数timeit和被装饰函数loop的传入参数和返回值类型必须保持一致。
4.3 步骤
实现此案例需要按照如下步骤进行。
步骤一:编写脚本
[root@localhost day05]# vim deco.py
#!/usr/bin/env python3
import time
def deco(func):
def timeit():
start = time.time()
result = func()
end = time.time()
return end - start, result # 将会返回元组
return timeit
@deco
def loop():
result = []
for i in range(1, 6):
result.append(i)
time.sleep(0.3)
return result
if __name__ == '__main__':
# loop = deco(loop)
print(loop())
1234567891011121314151617181920
步骤二:测试脚本执行
[root@localhost day06]# python3 deco.py
(1.50368070602417, [1, 2, 3, 4, 5])
12