Function capture_stdout

marimo/_runtime/capture.py:22–49

Capture standard output. Use this context manager to capture print statements and other output sent to standard output. Examples: ```python with mo.capture_stdout() as buffer: print("Hello!") output = buffer.getvalue() ```

()

@contextlib.contextmanager
def capture_stdout() -> Iterator[io.StringIO]:
    """Capture standard output.

    Use this context manager to capture print statements and
    other output sent to standard output.

    Examples:
        ```python
        with mo.capture_stdout() as buffer:
            print("Hello!")
        output = buffer.getvalue()
        ```
    """
    proxy = sys.stdout
    if _is_proxy(proxy):
        # Temporarily swap the thread-local stream to a StringIO so that
        # writes from this thread are captured while other threads are
        # unaffected.
        buffer = io.StringIO()
        old = proxy._get_stream()  # type: ignore[union-attr]
        proxy._set_stream(buffer)  # type: ignore[union-attr]
        try:
            yield buffer
        finally:
            proxy._set_stream(old)  # type: ignore[union-attr]
    else:
        with contextlib.redirect_stdout(io.StringIO()) as buffer:
            yield buffer


@contextlib.contextmanager