[Python] suppress stdout and stderr messages

suppress_output.py

"""See also https://www.codeforests.com/2020/11/05/python-suppress-stdout-and-stderr/"""

import os
import sys
from contextlib import contextmanager

@contextmanager
def suppress_output(suppress_stdout=True, suppress_stderr=True):
    orig_streams = (sys.stdout, sys.stderr)
    null_stream = open(os.devnull, "w")

    if suppress_stdout:
        sys.stdout = null_stream
    if suppress_stderr:
        sys.stderr = null_stream

    try:
        yield
    finally:
        sys.stdout, sys.stderr = orig_streams


if __name__ == '__main__':
    with suppress_output():
        print("stdout INSIDE context")
        print("stderr INSIDE context", file=sys.stderr)
    print("stdout OUTSIDE context")
    print("stderr OUTSIDE context", file=sys.stderr)