lin将stdout从subprocess.Popen保存到文件行

Question

我的python脚本使用子进程调用另一个脚本，这会产生非常慢的输出（逐行）。我不想在整个过程结束时将输出逐行写入文件，而是将整个输出作为字符串写入。下面的代码在“脚本”结束时将输出写入“文件”。

args = ("script")
file = open('output.txt', 'w')
subprocess.Popen(args,stdout=file)

有可能吗？Thanx，克里斯

Answer 1

我想我可以共享一个不使用.poll（）、.wait（）或.communicate（）的解决方案。几点：

• 我使用import codecs，因为我的输出包含东亚UTF-8文本
• 我用try:捕获每一行，以筛选出损坏/无效的UTF-8文本
• 我使用'\x0a'强制Linux换行，而不考虑平台。
• 如果需要捕获stderr，请使用for line in iter(subproc.stderr.readline, ''):
• 这种方法只在子程序创建输出时生成输出
• 在本例中，使用kw字典是过分的，但是演示了如何将**kwargs与子流程一起使用

代码：

import subprocess
import codecs
import os

kw = {
    'bufsize': 0,
    'executable': None,
    'stdin': subprocess.PIPE,
    'stdout': subprocess.PIPE,
    'stderr': subprocess.PIPE,
    'preexec_fn': None,
    'close_fds': False,
    'shell': False,
    'cwd': None,
    'env': None,
    'universal_newlines': False,
    'startupinfo': None,
    'creationflags': 0,
    }

args = ['ls', '-lart']
kw['cwd'] = os.path.expanduser('~')
logfile = os.path.expanduser('~/stdout.txt')
stdlog = []

try:
    subproc = subprocess.Popen(args,**kw)
except:
    print 'Error loading subprocess. Check arguments and kwargs'
    exit()

log = codecs.open(logfile,'w','utf-8')
log.write(': Starting log for: \"%s\"\x0a'%(' '.join(args)))
for line in iter(subproc.stdout.readline, ''):
    try:
        stdlog.append(line.rstrip().decode('utf-8'))
        log.write(stdout[-1]+'\x0a')
        print stdout[-1]
    except:
        pass

log.flush()
log.close()