带有/usr/bin/time : How can I capture timing information, 的Python 子进程但忽略所有其他输出？

Question

我正在尝试测量通过子进程调用的可执行程序的执行时间(以秒为单位)。我不希望发出可执行文件的输出(stderr 或 stdout)。

我试过timeit和resource库，都没有准确捕捉到进程的时间，貌似只能在Python工作线程中捕捉到时间。

下面的尝试将丢失 stderr 重定向的计时信息 b/c。但是，如果没有 stderr 重定向，将发出命令 'f_cmd' stderr 输出。

def doWithTiming(f_cmd):
    DEVNULL = open(os.devnull, 'w')
    return subprocess.check_output([ "/usr/bin/time", "--format=%e seconds"] + f_cmd.split(), stderr=DEVNULL)

如何忽略 f_cmd 的所有输出但保留/usr/bin/time 的输出？

Answer 1

%e /usr/bin/time format is :

Elapsed real (wall clock) time used by the process, in seconds.

使用抑制的 stdout/stderr 运行子进程并获取耗时:

#!/usr/bin/env python
import os
import time
from subprocess import check_call, STDOUT

DEVNULL = open(os.devnull, 'wb', 0)

start = time.time()
check_call(['sleep', '1'], stdout=DEVNULL, stderr=STDOUT)
print("{:.3f} seconds".format(time.time() - start))

---

timeit.default_timer 在 Python 2 的 POSIX 上是 time.time 因此你应该得到一个有效时间，除非你对 timeit 的使用是不正确。

---

resource 模块返回的信息不包括“真实”时间，但您可以使用它来获取“用户”和“系统”时间，即 < em>“进程在用户模式下花费的 CPU 秒总数。” 和 “进程在内核模式下花费的 CPU 秒总数。” 相应地: p>

#!/usr/bin/env python
import os
import time
from subprocess import Popen, STDOUT

DEVNULL = open(os.devnull, 'wb', 0)

start = time.time()
p = Popen(['sleep', '1'], stdout=DEVNULL, stderr=STDOUT)
ru = os.wait4(p.pid, 0)[2]
elapsed = time.time() - start
print(" {:.3f}real {:.3f}user {:.3f}system".format(
       elapsed, ru.ru_utime, ru.ru_stime))

---

您可以使用 psutil.Popen 启动子进程，并在子进程运行时获取 附加信息(cpu、内存、网络连接、线程、fds、子进程、等)以便携的方式。

另请参阅，How to get the max memory usage of a program using psutil in Python .

---

对于测试(以确保基于 time.time() 的解决方案产生相同的结果)，您可以捕获 /usr/bin/time 输出:

#!/usr/bin/env python
import os
from collections import deque
from subprocess import Popen, PIPE

DEVNULL = open(os.devnull, 'wb', 0)

time_lines_count = 1 # how many lines /usr/bin/time produces
p = Popen(['/usr/bin/time', '--format=%e seconds'] + 
          ['sleep', '1'], stdout=DEVNULL, stderr=PIPE)
with p.stderr:
    q = deque(iter(p.stderr.readline, b''), maxlen=time_lines_count)
rc = p.wait()
print(b''.join(q).decode().strip())

或者使用带有命名管道的-o选项:

#!/usr/bin/env python
import os
from contextlib import contextmanager
from shutil     import rmtree
from subprocess import Popen, STDOUT
from tempfile   import mkdtemp

DEVNULL = open(os.devnull, 'wb', 0)

@contextmanager
def named_pipe():
    dirname = mkdtemp()
    try:
        path = os.path.join(dirname, 'named_pipe')
        os.mkfifo(path)
        yield path
    finally:
        rmtree(dirname)

with named_pipe() as path:
    p = Popen(['/usr/bin/time', '--format=%e seconds', '-o', path] + 
              ['sleep', '1'], stdout=DEVNULL, stderr=STDOUT)
    with open(path) as file:
        time_output = file.read().strip()
    rc = p.wait()
print(time_output)