python tee example

tee.py

from contextlib import contextmanager
from tee import StdoutTee, StderrTee

# λ  python test.py
# test tee
# λ  cat stdout.log
# test tee

@contextmanager
def Tee():
    # must use buffer io on windows due to python 3.6 bug
    with StdoutTee("stdout.log", buff=1), StderrTee("stderr.log", buff=1):
        yield

def main():
    with Tee():
        print('test tee')

if __name__ == '__main__':
    main()