fix: Reload and Auto-reloading doesn't work when using Python3.x

sharow · sharow · commit b71f41138732 · 2015-07-06T06:37:28.000+09:00
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -450,7 +450,7 @@ def smarter_request_reload(files_modified=()):
 
         self.incremental_search_target = ''
 
-        self.original_modules = sys.modules.keys()
+        self.original_modules = set(sys.modules.keys())
 
         self.width = None
         self.height = None
@@ -846,9 +846,8 @@ def clear_modules_and_reevaluate(self):
         if self.watcher:
             self.watcher.reset()
         cursor, line = self.cursor_offset, self.current_line
-        for modname in sys.modules.keys():
-            if modname not in self.original_modules:
-                del sys.modules[modname]
+        for modname in (set(sys.modules.keys()) - self.original_modules):
+            del sys.modules[modname]
         self.reevaluate(insert_into_history=True)
         self.cursor_offset, self.current_line = cursor, line
         self.status_bar.message(_('Reloaded at %s by user.') %
