Redirect stdout to a file in python using with block

stdio_redirect.py

import sys

class stdio_redirect:
    def __init__(self, stdin=None, stdout=None, stderr=None):
        self.stdin,  self.sys_stdin   = stdin,  sys.stdin
        self.stdout, self.sys_stdout  = stdout, sys.stdout
        self.stderr, self.sys_stderr  = stderr, sys.stderr

    def __enter__(self):
        sys.stdin  = self.stdin
        sys.stdout = self.stdout
        sys.stderr = self.stderr
        return self.sys_stdin, self.sys_stdout, self.stderr

    def __exit__(self, exc_type, exc_value, traceback):
        sys.stdin  = self.sys_stdin
        sys.stdout = self.sys_stdout
        sys.stderr = self.sys_stderr

if __name__ == '__main__':
    with stdio_redirect(stdout=sys.stderr, stderr=sys.stdout) as (f0, f1, f2):
        print("1: stderr")
        print("2: stdout", file=sys.stderr)
        print("3: stdout", file=f1)
    print("4: stdout")
    print("5: stderr", file=sys.stderr)