GoodManWEN/Python-context-manager-example.md

Forked from cleiveliu/Python-context-manager-example.md

Created August 13, 2021 03:19

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/GoodManWEN/e51ec7e765e194856e20cf9d74758e49.js"></script>
Save GoodManWEN/e51ec7e765e194856e20cf9d74758e49 to your computer and use it in GitHub Desktop.

Download ZIP

Python context manager example

Raw

Python-context-manager-example.md

source

import contextlib
import sys


def your_task():
    print("The test !")


@contextlib.contextmanager
def close_stdout(redirect_to="/var/log/your.log"):
    raw_stdout = sys.stdout
    file = open(redirect_to, "a+")
    sys.stdout = file

    yield

    sys.stdout = raw_stdout
    file.close()


with close_stdout(redirect_to="./test.log"):
    your_task()