Python redirect stdout to console and file

tee.py

"""https://stackoverflow.com/questions/14906764/how-to-redirect-stdout-to-both-file-and-console-with-scripting"""
import sys
import traceback


class Tee(object):
    """Context manager that copies stdout and any exceptions to a log file.
    
    Example usage:
    
    print("Print")
    with Tee('test.txt'):
        print("Print+Write")
        raise Exception("Test")
    print("Print")
    """
    
    def __init__(self, filename):
        self.file = open(filename, 'w')
        self.stdout = sys.stdout

    def __enter__(self):
        sys.stdout = self

    def __exit__(self, exc_type, exc_value, tb):
        sys.stdout = self.stdout
        if exc_type is not None:
            self.file.write(traceback.format_exc())
        self.file.close()

    def write(self, data):
        self.file.write(data)
        self.stdout.write(data)

    def flush(self):
        self.file.flush()
        self.stdout.flush()