將subprocess.check_call的stdout重定向到函數？

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) 

Answer 2

Python是鴨類型的，所以你可以隨時將其傳遞給check_call之前包裝你的文件對象。

這個答案有一個doing it for write()的例子，但要徹底，你可能還想包裝writelines()。