Python 함수 호출에서 표준 출력을 캡처하는 방법은 무엇입니까?

Python 함수 호출에서 표준 출력 캡처

객체를 수정하고 통계를 stdout으로 인쇄하는 Python 라이브러리를 사용할 때 다음이 필요할 수 있습니다. 추가 분석을 위해 이 출력을 캡처합니다. 그러나 이러한 정보를 반환하도록 함수를 직접 수정하는 것이 항상 가능한 것은 아닙니다.

이 문제를 해결하려면 캡처 컨텍스트 관리자를 활용할 수 있습니다.

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

class Capturing(list):
    def __enter__(self):
        self._stdout = sys.stdout
        sys.stdout = self._stringio = StringIO()
        return self
    def __exit__(self, *args):
        self.extend(self._stringio.getvalue().splitlines())
        del self._stringio  # free up some memory
        sys.stdout = self._stdout</code>

캡처 컨텍스트 관리자를 사용할 수 있습니다. 다음과 같습니다:

<code class="python">with Capturing() as output:
    do_something(my_object)</code>

함수 호출 후 출력 목록에는 함수에 의해 인쇄된 줄이 포함됩니다.

이 기술은 여러 번 적용할 수 있으며 결과는 연결될 수 있습니다.

<code class="python">with Capturing() as output:
    print('hello world')

print('displays on screen')

with Capturing(output) as output:  # note the constructor argument
    print('hello world2')

print('done')
print('output:', output)</code>

출력:

displays on screen
done
output: ['hello world', 'hello world2']

이 접근 방식은 do_something() 함수를 직접 수정할 수 없는 경우 stdout 출력을 캡처하기 위한 효과적인 해결 방법을 제공합니다.

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