Я ищу решение Python, которое позволит мне сохранить вывод команды в файл, не скрывая его с консоли.
К вашему сведению: я спрашиваю о tee (в качестве утилиты командной строки Unix) и а не одноименная функция из модуля Python intertools.
Подробности
- Решение Python (не вызывает
tee
, недоступно в Windows) - Мне не нужно вводить какие-либо данные в stdin для вызываемого процесса
- Я не контролирую вызываемую программу. Все, что я знаю, это то, что он выведет что-то в 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)
Real output
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).