如何将并行子进程的输出写入不同的文件

时间:2014-04-25 18:17:09

标签: python subprocess output

我有以下python脚本。如何单独记录每个命令的输出,即每个命令包含该命令的输出一个文件?

#!/usr/bin/env python                                                                                                                                                                                                                                                           

from subprocess import Popen
import sys

commands = [
    'command1',
    'command2',
    'command3'
]

processes = [Popen(cmd, shell=True) for cmd in commands]

for p in processes:
    p.wait()

1 个答案:

答案 0 :(得分:2)

只需将stdout参数设置为相应的文件:

import shlex
from contextlib import ExitStack # $ pip install contextlib2 (on Python 2)
from subprocess import Popen

with ExitStack() as stack:
    for i, cmd in enumerate(commands):
        output_file = stack.enter_context(open('output%d.log' % i, 'w'))
        stack.callback(Popen(shlex.split(cmd), stdout=output_file).wait)

要从子进程重定向stderr输出,可以设置stderr参数。如果stderr=subprocess.STDOUT则合并stdout和stderr。

ExitStack用于关闭文件并等待已启动的子进程退出,即使with语句中发生异常,例如,如果某些命令无法启动。