Redirect stdout to what ever you want: context manager and decorator implementation

redirect.py

import sys
from contextlib import contextmanager
from functools import wraps


def redirect_stdout_to(out_new):
    """Redirect stdout to out_new  with context manager."""
    out_old = sys.stdout
    sys.stdout = out_new
    yield
    sys.stdout = out_old

def redirect_stdout_to(out_new):
    """Redirect stdout to out_new with decorator."""
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            out_old = sys.stdout
            sys.stdout = out_new
            try:
                return func(*args, **kwargs)
            finally:
                sys.stdout = out_old
        return wrapper
    return decorator

@contextmanager
def redirect_stdout_to(out_new):
    """
    Redirect stdout to out_new. It can be used as both context manager and a decorator.
    """
    stdout_old = sys.stdout
    sys.stdout = out_new
    try:
        yield
    except Exception as e:
        exception_name = e.__class__.__name__
        print(e)
    finally:
        sys.stdout = stdout_old

# Usage
with redirect_stdout_to(sys.stderr):
    print('Hello stderr')

@redirect_stdout_to(sys.stderr)
def whatnot():
    print('whatnot is the worsest name that you can choose')
   
whatnot()

class Person:
    @redirect_stdout_to(sys.stderr)
    def whanot(self):
        print('Hello, world!')