[feat] Bare import with path.to.module

Author: p1-dta

examples/aliases/package/__init__.py

@@ -3,6 +3,6 @@
 am: lazimp.ModuleType
 
 __getattr__ = lazimp.lazy_import(
-    sub_import={'alias_module': 'package'},
-    aliases={'am': 'alias_module'}
+    aliases={'am': 'alias_module'},
+    alias_module='package',
 )

examples/bare_sub_import/main.py

@@ -0,0 +1,14 @@
+import package
+
+print('Before access to package_2')
+print('load of bare_sub_module and bare_sub_module_2: ', package.package_2)
+print('After access to package_2')
+print('package_2.bare_sub_module: ', package.package_2.bare_sub_module)
+print('package_2.bare_sub_module_2: ',
+      package.package_2.bare_sub_module_2)
+print(
+    'package_2.bare_sub_module_3 access success because package_2 expose it: ',
+    package.package_2.bare_sub_module_3)
+print(
+    'package_2.bare_sub_module_4 access failed because package_2 does not '
+    'expose it: ', package.package_2.bare_sub_module_4)

examples/bare_sub_import/package/__init__.py

@@ -0,0 +1,6 @@
+import lazimp
+
+sub_package: lazimp.ModuleType
+
+__getattr__ = lazimp.lazy_import('package_2.bare_sub_module',
+                                 'package_2.bare_sub_module_2')

examples/bare_sub_import/package_2/__init__.py

@@ -0,0 +1 @@
+print('bare_sub_module loaded')

examples/bare_sub_import/package_2/bare_sub_module.py

@@ -0,0 +1 @@
+print('bare_sub_module_2 loaded')

examples/bare_sub_import/package_2/bare_sub_module_2.py

@@ -0,0 +1 @@
+print('bare_sub_module_3 loaded')

examples/bare_sub_import/package_2/bare_sub_module_3.py

@@ -3,7 +3,6 @@
 math: lazimp.ModuleType
 heavy_module: lazimp.ModuleType
 
-# __getattr__ = lazimp.lazy_import('math', 'package.heavy_module')
 __getattr__ = lazimp.lazy_import(
     'math',
     heavy_module='package',

examples/basic/package/__init__.py

@@ -4,7 +4,7 @@
 
 
 def lazy_import(
-        *imports: str,
+        *bare_imports: str,
         aliases: collections.abc.Mapping[str, str] | None = None,
         sub_import: collections.abc.Mapping[str, str] | None = None,
         **kw_sub_import: str,
@@ -30,18 +30,25 @@ def lazy_import(
             sub_import={'submodule2':'package'}
         )
 
-    :param imports: Any number of module names to import.
+    :param bare_imports: Any number of module names to import.
     :param aliases: A mapping of alias names to module names.
     :param sub_import: A mapping of module names to package names.
     :param kw_sub_import: Keyword arguments of module names to package names.
     :return: A function that returns the imported modules.
     """
-    imports = set(imports)
     if aliases is None:
         aliases = {}
     if sub_import is None:
         sub_import = {}
-    sub_import = sub_import | kw_sub_import
+
+    imports = {name for name in bare_imports if len(name.split('.')) == 1}
+    top_level_import = collections.defaultdict(list)
+    for import_path in bare_imports:
+        root, *path = (import_path.split('.'))
+        if path:
+            top_level_import[root].append(import_path)
+
+    sub_import.update(kw_sub_import)
 
     import functools
     import importlib
@@ -50,11 +57,6 @@ def lazy_import(
     def getattr_(name: str) -> ModuleType:
         name = aliases.get(name, name)
 
-        if name not in imports and name not in sub_import:
-            raise AttributeError(
-                f'Module has no attribute {name!r}'
-            ) from None
-
         if name in imports:
             try:
                 return importlib.import_module(name)
@@ -63,10 +65,23 @@ def getattr_(name: str) -> ModuleType:
                     f'Module has no attribute {name!r}'
                 ) from e
 
-        try:
-            return importlib.import_module(f'.{name}',
-                                           sub_import[name])
-        except ModuleNotFoundError as e:
-            raise AttributeError(f'Module has no attribute {name!r}') from e
+        if name in sub_import:
+            try:
+                return importlib.import_module(f'.{name}', sub_import[name])
+            except ModuleNotFoundError as e:
+                raise AttributeError(
+                    f'Module has no attribute {name!r}') from e
+
+        if name in top_level_import:
+            try:
+                module = importlib.__import__(top_level_import[name].pop())
+                for import_path_ in top_level_import[name]:
+                    module = importlib.__import__(import_path_)
+                return module
+            except ModuleNotFoundError as e:
+                raise AttributeError(
+                    f'Module has no attribute {name!r}') from e
+
+        raise AttributeError(f'Module has no attribute {name!r}')
 
     return getattr_