summaryrefslogtreecommitdiff
path: root/Lib/contextlib.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/contextlib.py')
-rw-r--r--Lib/contextlib.py52
1 files changed, 51 insertions, 1 deletions
diff --git a/Lib/contextlib.py b/Lib/contextlib.py
index f8e026b7e2..41ff9cc448 100644
--- a/Lib/contextlib.py
+++ b/Lib/contextlib.py
@@ -4,7 +4,8 @@ import sys
from collections import deque
from functools import wraps
-__all__ = ["contextmanager", "closing", "ContextDecorator", "ExitStack"]
+__all__ = ["contextmanager", "closing", "ContextDecorator", "ExitStack",
+ "ignore", "redirect_stdout"]
class ContextDecorator(object):
@@ -140,6 +141,55 @@ class closing(object):
def __exit__(self, *exc_info):
self.thing.close()
+class redirect_stdout:
+ """Context manager for temporarily redirecting stdout to another file
+
+ # How to send help() to stderr
+
+ with redirect_stdout(sys.stderr):
+ help(dir)
+
+ # How to write help() to a file
+
+ with open('help.txt', 'w') as f:
+ with redirect_stdout(f):
+ help(pow)
+
+ # How to capture disassembly to a string
+
+ import dis
+ import io
+
+ f = io.StringIO()
+ with redirect_stdout(f):
+ dis.dis('x**2 - y**2')
+ s = f.getvalue()
+
+ """
+
+ def __init__(self, new_target):
+ self.new_target = new_target
+
+ def __enter__(self):
+ self.old_target = sys.stdout
+ sys.stdout = self.new_target
+ return self.new_target
+
+ def __exit__(self, exctype, excinst, exctb):
+ sys.stdout = self.old_target
+
+@contextmanager
+def ignore(*exceptions):
+ """Context manager to ignore specified exceptions
+
+ with ignore(OSError):
+ os.remove(somefile)
+
+ """
+ try:
+ yield
+ except exceptions:
+ pass
# Inspired by discussions on http://bugs.python.org/issue13585
class ExitStack(object):