使用子进程时如何在 Python 中复制 tee 行为？

我正在寻找一个 Python 解决方案，它允许我将命令的输出保存在文件中，而不将其从控制台隐藏。

仅供参考：我问的是tee（作为 Unix 命令行实用程序）而不是 Python intertools 模块中的同名函数。

Details

• Python解决方案（不调用tee，Windows下不可用）
• 我不需要为被调用进程提供任何输入到标准输入
• 我无法控制被调用的程序。我所知道的是它会向 stdout 和 stderr 输出一些内容并返回退出代码。
• 在调用外部程序（子进程）时工作
• 为双方工作stderr and stdout
• 能够区分 stdout 和 stderr，因为我可能只想将其中之一显示到控制台，或者我可以尝试使用不同的颜色输出 stderr - 这意味着stderr = subprocess.STDOUT不管用。
• 实时输出（渐进式）- 该过程可以运行很长时间，而且我无法等待它完成。
• Python 3 兼容代码（重要）

参考

以下是我迄今为止发现的一些不完整的解决方案：

• http://devlishgenius.blogspot.com/2008/10/logging-in-real-time-in-python.html（mkfifo 仅适用于 Unix）
• http://blog.kagesenshi.org/2008/02/teeing-python-subprocesspopen-output.html（根本不起作用）

图表http://blog.i18n.ro/wp-content/uploads/2010/06/Drawing_tee_py.png

当前代码（第二次尝试）

#!/usr/bin/python
from __future__ import print_function

import sys, os, time, subprocess, io, threading
cmd = "python -E test_output.py"

from threading import Thread
class StreamThread ( Thread ):
    def __init__(self, buffer):
        Thread.__init__(self)
        self.buffer = buffer
    def run ( self ):
        while 1:
            line = self.buffer.readline()
            print(line,end="")
            sys.stdout.flush()
            if line == '':
                break

proc = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
stdoutThread = StreamThread(io.TextIOWrapper(proc.stdout))
stderrThread = StreamThread(io.TextIOWrapper(proc.stderr))
stdoutThread.start()
stderrThread.start()
proc.communicate()
stdoutThread.join()
stderrThread.join()

print("--done--")

#### test_output.py ####

#!/usr/bin/python
from __future__ import print_function
import sys, os, time

for i in range(0, 10):
    if i%2:
        print("stderr %s" % i, file=sys.stderr)
    else:
        print("stdout %s" % i, file=sys.stdout)
    time.sleep(0.1)

stderr 1
stdout 0
stderr 3
stdout 2
stderr 5
stdout 4
stderr 7
stdout 6
stderr 9
stdout 8
--done--

预期输出是对线路进行排序。请注意，修改 Popen 以仅使用一个 PIPE 是不允许的，因为在现实生活中我会想要使用 stderr 和 stdout 做不同的事情。

此外，即使在第二种情况下，我也无法获得实时的结果，事实上，当过程完成时，所有结果都已收到。默认情况下，Popen 不应使用任何缓冲区 (bufsize=0)。

我发现这是一篇相当旧的帖子，但以防万一有人仍在寻找一种方法来做到这一点：

proc = subprocess.Popen(["ping", "localhost"], 
                        stdout=subprocess.PIPE, 
                        stderr=subprocess.PIPE)

with open("logfile.txt", "w") as log_file:
  while proc.poll() is None:
     line = proc.stderr.readline()
     if line:
        print "err: " + line.strip()
        log_file.write(line)
     line = proc.stdout.readline()
     if line:
        print "out: " + line.strip()
        log_file.write(line)