프로그램 출력 캡처: 단순한 솔루션을 넘어서

Python 스크립팅에서는 추가 처리를 위해 프로그램 출력을 캡처하는 것이 일반적으로 필요합니다. 순진한 솔루션은 간단해 보일 수 있지만 종종 부족합니다. stdout에 쓰는 다음 스크립트를 고려하십시오.

# writer.py
import sys

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

다음 코드를 사용하여 출력을 캡처하려는 시도가 실패합니다.

# mymodule.py
from writer import write

out = write()
print(out.upper())

출력을 효과적으로 캡처하려면 보다 강력한 솔루션이 필요합니다. 한 가지 접근 방식은 시스템의 stdout 스트림을 수정하는 것입니다:

import sys
from cStringIO import StringIO

# Redirect stdout to a StringIO object
backup = sys.stdout
sys.stdout = StringIO()

# Perform the write operation
write()

# Retrieve and restore stdout
out = sys.stdout.getvalue()
sys.stdout.close()
sys.stdout = backup

# Process the captured output
print(out.upper())

Python 3.4용 컨텍스트 관리자 :

Python 3.4 이상에서는 contextlib를 사용하여 더 간단하고 간결한 솔루션을 사용할 수 있습니다. Redirect_stdout 컨텍스트 관리자:

from contextlib import redirect_stdout
import io

f = io.StringIO()
# Redirect stdout to f using the context manager
with redirect_stdout(f):
    help(pow)

# Retrieve captured output from f
s = f.getvalue()

이 우아한 접근 방식은 출력 캡처 프로세스를 단순화하여 Python 스크립트에서 더 쉽게 처리할 수 있게 해줍니다.