1行Python代码，计算程序的运行时间，也可以用在算法和接口的调优上

有时候我们写了一段代码，想知道它运行时间，你会怎么做？

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即可。

4、参考资料

• 利用世界杯，读懂 Python 装饰器
• Python爬虫如何加速？
• 1022页PDF：《流畅的Python》，Python程序员进阶必备图书~