将 subprocess.Popen 的输出通过管道传输到文件

Question

我需要使用 subprocess.Popen 启动多个长时间运行的进程，并希望每个进程自动通过管道传输 stdout 和 stderr分隔日志文件。每个进程将同时运行几分钟，我希望每个进程写入两个日志文件（stdout 和 stderr）作为进程跑步。

我是否需要在循环中的每个进程上不断调用 p.communicate() 才能更新每个日志文件，或者是否有某种方法来调用原始 Popen 命令这样 stdout 和 stderr 自动流式传输以打开文件句柄？

Answer 1

您可以将 stdout 和 stderr 作为参数传递给 Popen()

subprocess.Popen(self, args, bufsize=0, executable=None, stdin=None, stdout=None,
                 stderr=None, preexec_fn=None, close_fds=False, shell=False,
                 cwd=None, env=None, universal_newlines=False, startupinfo=None, 
                 creationflags=0)

例如

>>> import subprocess
>>> with open("stdout.txt","wb") as out, open("stderr.txt","wb") as err:
...    subprocess.Popen("ls",stdout=out,stderr=err)
... 
<subprocess.Popen object at 0xa3519ec>
>>> 

Answer 2

根据文档，

stdin、stdout 和 stderr 指定
执行程序的标准输入，
标准输出和标准错误
分别是文件句柄。有效的
值为 PIPE，一个现有文件
描述符（正整数），
现有文件对象，并且无。

因此，只需将打开写入的文件对象作为命名参数 stdout= 和 stderr= 传递，就可以了！

Answer 3

我同时运行两个子进程，并将两个子进程的输出保存到一个日志文件中。我还内置了一个超时来处理挂起的子进程。当输出太大时，总是会触发超时，并且来自任一子进程的标准输出都不会保存到日志文件中。 Alex 上面提出的答案并没有解决这个问题。

# Currently open log file.
log = None

# If we send stdout to subprocess.PIPE, the tests with lots of output fill up the pipe and
# make the script hang. So, write the subprocess's stdout directly to the log file.
def run(cmd, logfile):
   #print os.getcwd()
   #print ("Running test: %s" % cmd)
   global log
   p = subprocess.Popen(cmd, shell=True, universal_newlines = True, stderr=subprocess.STDOUT, stdout=logfile)
   log = logfile
   return p


# To make a subprocess capable of timing out
class Alarm(Exception):
   pass

def alarm_handler(signum, frame):
   log.flush()
   raise Alarm


####
## This function runs a given command with the given flags, and records the
## results in a log file. 
####
def runTest(cmd_path, flags, name):

  log = open(name, 'w')

  print >> log, "header"
  log.flush()

  cmd1_ret = run(cmd_path + "command1 " + flags, log)
  log.flush()
  cmd2_ret = run(cmd_path + "command2", log)
  #log.flush()
  sys.stdout.flush()

  start_timer = time.time()  # time how long this took to finish

  signal.signal(signal.SIGALRM, alarm_handler)
  signal.alarm(5)  #seconds

  try:
    cmd1_ret.communicate()

  except Alarm:
    print "myScript.py: Oops, taking too long!"
    kill_string = ("kill -9 %d" % cmd1_ret.pid)
    os.system(kill_string)
    kill_string = ("kill -9 %d" % cmd2_ret.pid)
    os.system(kill_string)
    #sys.exit()

  end_timer = time.time()
  print >> log, "closing message"

  log.close()

Answer 4

根据 Alex Martelli 的回答，我创建了一个对我有用的小例子。

runit.sh

执行的程序：

#!/usr/bin/env bash

sleep 5
ls -la
sleep 5
ls -la /Users
sleep 5
echo "Hello World!!"

在Python中分离进程

import os
from subprocess import Popen
import shlex


if __name__ == "__main__":

    fhout = open(f'{os.getcwd()}/stdout.log', 'w')
    fherr = open(f'{os.getcwd()}/stderr.log', 'w')
    command = shlex.split(f"{os.getcwd()}/runit.sh")
    p = Popen(command, shell=True, stdout=fhout, stderr=fherr, text=True, close_fds=True)

    pid = p.pid

    print(f'PID of the process is: {pid}')