29 lines
612 B
Python
29 lines
612 B
Python
"""Provide capture_stdout as context manager."""
|
|
|
|
import contextlib
|
|
import io
|
|
import sys
|
|
|
|
# From
|
|
# http://stackoverflow.com/questions/2654834/capturing-stdout-within-the-same-process-in-python
|
|
|
|
|
|
class Data(object):
|
|
"""Object for captured data."""
|
|
|
|
def __init__(self):
|
|
"""Container for result."""
|
|
self.result = ''
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def capture_stdout():
|
|
"""Capture and return stdout as context manager."""
|
|
old = sys.stdout
|
|
capturer = io.StringIO()
|
|
sys.stdout = capturer
|
|
data = Data()
|
|
yield data
|
|
sys.stdout = old
|
|
data.result = capturer.getvalue()
|
