_check_module.find(): support namespace package traversal

Author: kislyuk

argcomplete/_check_module.py

@@ -10,30 +10,7 @@
 import os
 import sys
 import tokenize
-
-try:
-    from importlib.util import find_spec  # type:ignore
-except ImportError:
-    import typing as t
-    from collections import namedtuple
-    from imp import find_module  # type:ignore
-
-    ModuleSpec = namedtuple("ModuleSpec", ["origin", "has_location", "submodule_search_locations"])
-
-    def find_spec(  # type:ignore
-        name: str,
-        package: t.Optional[str] = None,
-    ) -> t.Optional[ModuleSpec]:
-        """Minimal implementation as required by `find`."""
-        try:
-            f, path, _ = find_module(name)
-        except ImportError:
-            return None
-        has_location = path is not None
-        if f is None:
-            return ModuleSpec(None, has_location, [path])
-        f.close()
-        return ModuleSpec(path, has_location, None)
+from importlib.util import find_spec
 
 
 class ArgcompleteMarkerNotFound(RuntimeError):
@@ -42,7 +19,11 @@ class ArgcompleteMarkerNotFound(RuntimeError):
 
 def find(name, return_package=False):
     names = name.split(".")
-    spec = find_spec(names[0])
+    for package_name_boundary in range(len(names)):
+        spec = find_spec(".".join(names[:package_name_boundary+1]))
+        if spec is not None and spec.origin is not None:
+            break
+
     if spec is None:
         raise ArgcompleteMarkerNotFound('no module named "{}"'.format(names[0]))
     if not spec.has_location:
@@ -53,7 +34,7 @@ def find(name, return_package=False):
         return spec.origin
     if len(spec.submodule_search_locations) != 1:
         raise ArgcompleteMarkerNotFound("expecting one search location")
-    path = os.path.join(spec.submodule_search_locations[0], *names[1:])
+    path = os.path.join(spec.submodule_search_locations[0], *names[package_name_boundary+1:])
     if os.path.isdir(path):
         filename = "__main__.py"
         if return_package: