有时候我们写了一段代码,想知道它运行时间,你会怎么做?
Python有一个第三方库叫:potime
,提供了一个简便的方法,我们一起来看看吧。
1、下载
1行命令,下载第三方库
pip install potime
2、使用方法
例如上次我们的优化了根据内容查找文件的功能,使查找时间加快了N倍。
如果我们想测试一段代码:看看用Python在电脑里找出一个不记得名字,只记得内容的文件,需要多长时间,就可以用到potime
这个库了。
代码如下
import office # 导入自动化办公专用库
from potime import RunTime # 导入计算代码运行时间的库
@RunTime # 计算代码运行时间
def search_file():
office.file.search_by_content(search_path='d://', content='程序员晚枫') # 根据内容,查找文件
if __name__ == '__main__':
search_file() # 运行程序
运行结果如下
使用说明
导入potime
后,不用改变原有代码,直接在你想计算时间的方法上面,添加装饰器:@RunTime
,就可以在程序运行后,看到运行结果了。
3、拓展应用
在工作中,有时我们需要做一些算法或者接口的调优,加快运行时间。这个potime
也可以直接用在算法或者接口的调优上。
例如对flask接口的测试,代码如下:
# coding=utf-8
from flask import Flask
from potime import RunTime # 导入时间计算模块
# 创建flask应用程序
app = Flask(__name__)
# 写一个函数来处理浏览器发送过来的请求
@app.route("/") # 当访问网址时,默认执行下面函数
@RunTime # 计算当前接口的运行时间
def index():
a = 0
for i in range(1000000):
a = a + i
return 'weclome to python-office'
if __name__ == "__main__":
app.run(debug=True) # 启动应用程序
如上图index
方法所示,直接在接口对应的方法上,添加@RunTime
即可。