fix reload in Python 3

thomasballinger · thomasballinger · commit 2deadfeec845 · 2015-12-17T11:39:53.000-05:00
diff --git a/bpython/curtsiesfrontend/sitefix.py b/bpython/curtsiesfrontend/sitefix.py
@@ -3,6 +3,8 @@
 
 from six.moves import builtins
 
+from bpython import py3
+
 
 def resetquit(builtins):
     """Redefine builtins 'quit' and 'exit' not so close stdin
@@ -18,8 +20,11 @@ def monkeypatch_quit():
     if 'site' in sys.modules:
         resetquit(builtins)
 
-
-orig_reload = builtins.reload
+if py3:
+    import importlib
+    orig_reload = importlib.reload
+else:
+    orig_reload = builtins.reload
 
 
 def reload(module):
@@ -39,4 +44,7 @@ def reload(module):
 
 
 def monkeypatch_reload():
-    builtins.reload = reload
+    if py3:
+        importlib.reload = reload
+    else:
+        builtins.reload = reload
