python - 将 stdout 从 subprocess.check_call 重定向到函数？

Question

调用 subprocess.check_call() 允许为 stdout 指定文件对象，但在将数据写入文件之前，我想逐行修改它们。

我目前将输出重定向到一个临时文件(由 tempfile.TemporaryFile() 创建)。check_call 完成后，我逐行读取该临时文件，进行修改并编写最终输出文件。

由于输出很大，纯内存解决方案不可行，我想即时修改数据并直接写入最终输出文件。

有人知道如何实现吗？

Answer 1

def check_call_modify(command, modifier_function, output_file)
    p = subprocess.Popen(command, stdout=subprocess.PIPE)
    for line in p.stdout:
        line = modifier_function(line)
        output_file.write(line)    
    p.wait()
    if p.returncode:
        raise subprocess.CalledProcessError(p.returncode, command)
    return p.returncode

使用它传递一个函数来修改每一行和文件。下面的愚蠢示例将以大写形式将 ls -l 的结果保存到 listupper.txt:

with open('listupper.txt', 'w') as f:
    check_call_modify(['ls', '-l'], operator.methodcaller('upper'), f)