如何在 Python 中捕捉標準輸出？

在Python 中捕捉標準輸出

在Python 中，由於各種原因捕獲到標準輸出(stdout) 的輸出通常很不錯有用，例如保存以供以後處理或測試。一個常見的場景涉及從庫中呼叫函數，例如 do_something(my_object)，它將資訊列印到 stdout。

要捕捉此函數呼叫的輸出，可以使用以下方法：

1. 實作一個串聯上下文管理器（Capturing）：

   • 定義一個繼承自list 的名為Capturing 的類別。
   • 實作其__enter__ 和 __exit__ 方法來擷取 stdout 輸出。
   • 在其 __enter__ 方法中，將目前的 stdout 儲存到 self._stdout 中，並將 stdout 重定向到記憶體中的 StringIO 緩衝區 (self._stringio)。
   • 在其 __exit__ 方法中，將 stdout 恢復到原始狀態後將捕獲的輸出附加到其內部列表（自身）。

2. 使用上下文管理器：

   • 使用 with 語句建立 Capturing 的實例。
   • 將此實例作為參數傳遞給Capturing 的__init__ 方法，以將其捕獲的輸出附加到先前建立的實例

3. 取得捕獲的輸出：

   • 退出with 區塊後，輸出變數現在包含捕獲的stdout 輸出為字串清單。

用法範例：

<code class="python">from contextlib import redirect_stdout  # redirect_stdout is only available in Python 3.4 or later

def do_something(my_object):
    print('stdout info')

with redirect_stdout(StringIO()) as stdout:
    do_something(my_object)
    captured_output = stdout.getvalue()</code>

使用Python 3.4 語法：

<code class="python">with StringIO() as output:
    with redirect_stdout(output):
        do_something(my_object)
    captured_output = output.getvalue()</code>

透過執行下列步驟，您可以輕鬆捕捉以下步驟，您可以輕鬆捕捉Python 函數呼叫的stdout 輸出，使您能夠利用列印的資訊進行進一步處理、偵錯或測試目的。

以上是如何在 Python 中捕捉標準輸出？的詳細內容。更多資訊請關注PHP中文網其他相關文章！