mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-07-07 19:35:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Close #19403: make contextlib.redirect_stdout reentrant

Browse source
This commit is contained in:
Nick Coghlan 2013-11-03 17:00:51 +10:00
parent 4e641df09b
commit 8e113b418d
4 changed files with 100 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

12
Lib/contextlib.py
View file

@ -166,20 +166,16 @@ class redirect_stdout:
def __init__(self, new_target):
self._new_target = new_target
self._old_target = self._sentinel = object()
# We use a list of old targets to make this CM re-entrant
self._old_targets = []
def __enter__(self):
if self._old_target is not self._sentinel:
raise RuntimeError("Cannot reenter {!r}".format(self))
self._old_target = sys.stdout
self._old_targets.append(sys.stdout)
sys.stdout = self._new_target
return self._new_target
def __exit__(self, exctype, excinst, exctb):
restore_stdout = self._old_target
self._old_target = self._sentinel
sys.stdout = restore_stdout
sys.stdout = self._old_targets.pop()
class suppress: