Skip to content

Instantly share code, notes, and snippets.

@chengui
Created September 26, 2013 10:25
Show Gist options
  • Save chengui/6712354 to your computer and use it in GitHub Desktop.
Save chengui/6712354 to your computer and use it in GitHub Desktop.
redirect std stream to file stream
import os
import sys
class RedirectStdStreams(object):
def __init__(self, stdout=None, stderr=None):
self._stdout = stdout or sys.stdout
self._stderr = stderr or sys.stderr
def __enter__(self):
self.old_stdout, self.old_stderr = sys.stdout, sys.stderr
self.old_stdout.flush(); self.old_stderr.flush()
sys.stdout, sys.stderr = self._stdout, self._stderr
def __exit__(self, exc_type, exc_value, traceback):
self._stdout.flush(); self._stderr.flush()
sys.stdout = self.old_stdout
sys.stderr = self.old_stderr
if __name__ == '__main__':
devnull = open(os.devnull, 'w')
print('Fubar')
with RedirectStdStreams(stdout=devnull, stderr=devnull):
print("You'll never see me")
print("I'm back!")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment