捕獲程式輸出：超越樸素的解決方案

捕獲程式輸出：超越樸素的解決方案

# writer.py
import sys

def write():
    sys.stdout.write("foobar")

# writer.py
import sys

def write():
    sys.stdout.write("foobar")

# writer.py
import sys

def write():
    sys.stdout.write("foobar")

import sys 從 cStringIO 導入 StringIO # 將標準輸出重定向到 StringIO 物件 備份= sys.stdout sys.stdout = StringIO() # 執行寫入操作 寫（） # 檢索並恢復標準輸出 輸出 = sys.stdout.getvalue() sys.stdout.close() sys.stdout = 備份 # 處理捕獲的輸出 print(out.upper())

Python 3.4 的上下文管理器：

# writer.py
import sys

def write():
    sys.stdout.write("foobar")

Python 3.4 的上下文管理器：

對於Python 3.4 及更高版本，可以使用contextlib 獲得更簡單、更簡潔的解決方案。 redirect_stdout 上下文管理器：

from contextlib import redirect_stdout 導入io f = io.StringIO() # 使用上下文管理器將 stdout 重新導向到 f 使用redirect_stdout(f)： 幫助（戰俘） # 檢索 f 捕獲的輸出 s = f.getvalue()這種優雅的方法簡化了輸出捕獲過程，使其更容易在 Python 腳本中處理。