获取 python 子进程运行时间

Question

我正在使用 python 同时运行多个子进程。

我想获得每个进程的运行时间。

我正在使用subprocess模块。

我做了什么：我创建了两个单独的for循环：第一个用于运行每个进程第二个等待所有进程结束。

for prcs in batch:
    p = subprocess.Popen([prcs])
    ps.append(p)
for p in ps:
    p.wait()

此代码适用于同时运行进程，但我不知道要添加什么才能分别获取每个进程的运行时间。

编辑：有没有办法通过模块子进程获取运行时间？ 例如：runtime = p.runtime()

Answer 1

我同意@quamrana 的观点，最简单的方法是使用线程。

首先，我们需要导入一些标准库模块：

import collections
import subprocess
import threading
import time

我们不是使用列表来存储流程，而是使用有序字典来跟踪流程及其时间。 由于我们不知道每个线程需要多长时间，我们需要一些方法来跟踪我们的 {process: time} 对的原始顺序。 线程本身可以存储在列表中。

ps = collections.OrderedDict()
ts = []

将与每个进程配对的值初始化为当前时间会使整个事情更清晰，尽管通常不建议将相同的变量用于两个不同的事情（在这种情况下，开始时间后跟过程持续时间）。 我们线程的目标只是等待线程完成并将 ps 有序字典从开始时间更新到进程持续时间。

def time_p(p):
    p.wait()
    ps[p] = time.time() - ps[p]

for prcs in batch:
    p = subprocess.Popen([prcs])
    ps[p] = time.time()
    ts.append(threading.Thread(target=time_p, args=(p,)))

现在，我们只需启动每个线程，然后等待它们全部完成。

for t in ts:
    t.start()

for t in ts:
    t.join()

一旦它们都完成了，我们就可以打印出每个的结果：

for prcs, p in zip(batch, ps):
    print('%s took %s seconds' % (prcs, ps[p]))