我正在寻找一种 Python 解决方案,它允许我将命令的输出保存在文件中,而不会将其隐藏在控制台中。
仅供参考:我问的是tee(作为 Unix 命令行实用程序),而不是 Python intertools 模块中同名的函数。
细节
- Python解决方案(不调用
tee
,Windows下不可用) - 我不需要为被调用进程向标准输入提供任何输入
- 我无法控制被调用的程序。我所知道的是它会输出一些东西到 stdout 和 stderr 并返回一个退出代码。
- 调用外部程序(子进程)时工作
- 为双方
stderr
工作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)。