feat: add fallback to support namespace packages

percevalw · web-flow · commit 2dc3d9d3690b · 2025-06-06T09:18:28.000+02:00
diff --git a/src/mkdocstrings_handlers/python/_internal/handler.py b/src/mkdocstrings_handlers/python/_internal/handler.py
@@ -263,6 +263,9 @@ def collect(self, identifier: str, options: PythonOptions) -> CollectorItem:
                 _logger.debug(f"{len(unresolved)} aliases were still unresolved after {iterations} iterations")
                 _logger.debug(f"Unresolved aliases: {', '.join(sorted(unresolved))}")
 
+            if identifier not in self._modules_collection:
+                self.try_to_import(identifier, loader)
+
         try:
             doc_object = self._modules_collection[identifier]
         except KeyError as error:
@@ -278,6 +281,47 @@ def collect(self, identifier: str, options: PythonOptions) -> CollectorItem:
 
         return doc_object
 
+
+    def try_to_import(self, identifier: str, loader):
+        import importlib.util
+        from pathlib import Path
+
+
+        parts = ()
+        doc_object = None
+        for part in identifier.split("."):
+            parts = parts + (part,)
+            sub_identifier = ".".join(parts)
+            spec = importlib.util.find_spec(".".join(parts))
+            if spec is None:
+                if doc_object is not None and part in doc_object.members:
+                    # If we already have a doc_object, we can just use it
+                    doc_object = doc_object.get_member(part)
+                    continue
+                raise CollectionError(f"Module {identifier} could not be found")
+            origin = spec.origin
+            if origin is None:
+                # Pick the first loc that is accessible from here (ie is a children of cwd)
+                origin = next(
+                    (
+                        loc
+                        for loc in spec.submodule_search_locations
+                        if loc.startswith(os.getcwd())
+                    ),
+                    None,
+                )
+
+            doc_object = loader._load_module(
+                part,
+                Path(origin),
+                parent=doc_object,
+            )
+            try:
+                self._modules_collection.set_member(sub_identifier, doc_object)
+            except:
+                pass
+        return doc_object
+
     def render(self, data: CollectorItem, options: PythonOptions) -> str:
         """Render the collected data.
 
