A Tee context manager to redirect `print` calls to a file, as a quick logging method

python-stdout-tee.py

import sys


class Tee(object):
    def __init__(self, name, mode, autoflush=False):
        self.file = open(name, mode)
        self.stdout = sys.stdout
        sys.stdout = self
        self.autoflush = autoflush

    def write(self, data):
        self.file.write(data)
        self.stdout.write(data)
        if self.autoflush:
            self.flush()

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

    def close(self):
        self.file.close()
        sys.stdout = self.stdout

    def __enter__(self):
        return self

    def __exit__(self, _type, _value, _traceback):
        self.close()