Restore sys.std* in utils.fake_io on exceptions.

2015-02-26 17:47:07 +01:00 · 2015-02-26 17:47:07 +01:00 · 8e0dddf86a
commit 8e0dddf86a
parent fcbd69e209
1 changed files with 9 additions and 7 deletions
--- a/qutebrowser/utils/utils.py
+++ b/qutebrowser/utils/utils.py
@ -388,7 +388,9 @@ def fake_io(write_func):
    fake_stdout = FakeIOStream(write_func)
    sys.stderr = fake_stderr
    sys.stdout = fake_stdout
+    try:
        yield
+    finally:
        # If the code we did run did change sys.stdout/sys.stderr, we leave it
        # unchanged. Otherwise, we reset it.
        if sys.stdout is fake_stdout: