Restructure import's module loading.

jedi/evaluate/imports.py

@@ -449,30 +449,29 @@ class Importer(object):
 
 
 def _load_module(evaluator, path=None, source=None, sys_path=None, parent_module=None):
-    def load(source):
-        dotted_path = path and compiled.dotted_from_fs_path(path, sys_path)
-        if path is not None and path.endswith(('.py', '.zip', '.egg')) \
-                and dotted_path not in settings.auto_import_modules:
-            if source is None:
-                with open(path, 'rb') as f:
-                    source = f.read()
-        else:
-            return compiled.load_module(evaluator, path)
-        p = path
-        p = FastParser(evaluator.grammar, source_to_unicode(source), p)
-        save_parser(path, p)
-        return p.get_root_node()
-
     if sys_path is None:
         sys_path = evaluator.sys_path
 
-    cached = load_parser(path)
+    dotted_path = path and compiled.dotted_from_fs_path(path, sys_path)
-    module_node = load(source) if cached is None else cached.get_root_node()
+    if path is not None and path.endswith(('.py', '.zip', '.egg')) \
-    if isinstance(module_node, compiled.CompiledObject):
+            and dotted_path not in settings.auto_import_modules:
-        return module_node
 
-    from jedi.evaluate.representation import ModuleContext
+        cached = load_parser(path)
-    return ModuleContext(evaluator, module_node)
+        if cached is None:
+            if source is None:
+                with open(path, 'rb') as f:
+                    source = f.read()
+
+            p = FastParser(evaluator.grammar, source_to_unicode(source), path)
+            save_parser(path, p)
+            module_node = p.get_root_node()
+        else:
+            module_node = cached.get_root_node()
+
+        from jedi.evaluate.representation import ModuleContext
+        return ModuleContext(evaluator, module_node)
+    else:
+        return compiled.load_module(evaluator, path)
 
 
 def add_module(evaluator, module_name, module):