def warning(msg): if sys.stdout != sys.__stdout__: tmp = sys.stdout sys.stdout = sys.__stdout__ print('WARNING: ' + msg) sys.stdout = tmp else: print('WARNING: ' + msg)