Python에서 스크립트 stdout 출력을 캡처하는 방법은 무엇입니까?

Python의 스크립트에서 stdout 출력 캡처

외부 스크립트로 작업할 때 추가 처리를 위해 stdout 출력을 캡처해야 하는 경우가 많습니다. 일반적인 접근 방식은 write() 함수를 사용하여 출력을 인쇄한 다음 변수에 저장하는 것입니다. 그러나 다음 스크립트의 예시처럼 이 접근 방식이 항상 작동하는 것은 아닙니다.

<code class="python"># module writer.py
import sys

def write():
    sys.stdout.write("foobar")</code>

write() 함수의 출력을 캡처하기 위해 cStringIO 모듈과 관련된 솔루션이 제안되었습니다.

<code class="python">import sys
from cStringIO import StringIO

# setup the environment
backup = sys.stdout

# ####
sys.stdout = StringIO()     # capture output
write()
out = sys.stdout.getvalue() # release output
# ####

sys.stdout.close()  # close the stream 
sys.stdout = backup # restore original stdout

print out.upper()   # post processing</code>

그러나 이 목표를 달성하기 위한 더 효율적이고 Python적인 방법이 있습니다:

contextlib.redirect_stdout 사용(Python 3.4 )

이 방법은 contextlib를 활용합니다. .redirect_stdout 컨텍스트 관리자는 stdout 출력을 일시적으로 StringIO 객체로 리디렉션합니다.

<code class="python">from contextlib import redirect_stdout
import io

f = io.StringIO()
with redirect_stdout(f):
    help(pow)
s = f.getvalue()</code>

io.StringIO 직접 사용

또 다른 접근 방식은 StringIO 객체를 직접 생성하고 stdout 출력을 수동으로 경로 지정:

<code class="python">import io

output = io.StringIO()
sys.stdout = output
write()
sys.stdout.seek(0)
captured_output = output.read()</code>

이러한 방법을 구현하면 개발자는 복잡한 해결 방법을 사용하지 않고도 스크립트 출력을 효과적으로 캡처하고 처리할 수 있습니다.

위 내용은 Python에서 스크립트 stdout 출력을 캡처하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!