reset stdin/out/err when reloading sys

thomasballinger · thomasballinger · commit 561954415c58 · 2015-12-17T10:34:54.000-05:00
fixes bpython#567
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -524,6 +524,7 @@ def __enter__(self):
             sys.meta_path = [ImportFinder(self.watcher, self.orig_meta_path)]
 
         sitefix.monkeypatch_quit()
+        sitefix.monkeypatch_reload()
         return self
 
     def __exit__(self, *args):
diff --git a/bpython/curtsiesfrontend/sitefix.py b/bpython/curtsiesfrontend/sitefix.py
@@ -1,4 +1,5 @@
 import sys
+import functools
 
 from six.moves import builtins
 
@@ -16,3 +17,26 @@ def __call__(self, code=None):
 def monkeypatch_quit():
     if 'site' in sys.modules:
         resetquit(builtins)
+
+
+orig_reload = builtins.reload
+
+
+def reload(module):
+    if module is sys:
+        orig_stdout = sys.stdout
+        orig_stderr = sys.stderr
+        orig_stdin = sys.stdin
+        orig_reload(sys)
+        sys.stdout = orig_stdout
+        sys.stderr = orig_stderr
+        sys.stdin = orig_stdin
+    else:
+        builtins.reload(sys)
+
+
+functools.update_wrapper(reload, orig_reload)
+
+
+def monkeypatch_reload():
+    builtins.reload = reload
