Immediately close file object

sebastinas · sebastinas · commit e9651aefe988 · 2020-01-29T14:08:43.000+01:00
diff --git a/bpython/importcompletion.py b/bpython/importcompletion.py
@@ -43,6 +43,8 @@
 
     SUFFIXES = importlib.machinery.all_suffixes()
 else:
+    import imp
+
     SUFFIXES = [suffix for suffix, mode, type in imp.get_suffixes()]
 
 # The cached list of all known modules
@@ -158,20 +160,23 @@ def find_modules(path):
             # Workaround for issue #166
             continue
         try:
+            is_package = False
             with warnings.catch_warnings():
                 warnings.simplefilter("ignore", ImportWarning)
                 fo, pathname, _ = imp.find_module(name, [path])
+                if fo is not None:
+                    fo.close()
+                else:
+                    # Yay, package
+                    is_package = True
         except (ImportError, IOError, SyntaxError):
             continue
         except UnicodeEncodeError:
             # Happens with Python 3 when there is a filename in some
             # invalid encoding
             continue
         else:
-            if fo is not None:
-                fo.close()
-            else:
-                # Yay, package
+            if is_package:
                 for subname in find_modules(pathname):
                     if subname != "__init__":
                         yield "%s.%s" % (name, subname)
