[break] change api, bare imports are now varargs, sub imports can be kwvarargs

Author: p1-dta

README.md

@@ -30,7 +30,10 @@ import lazimp
 math: lazimp.ModuleType
 heavy_module: lazimp.ModuleType
 
-__getattr__ = lazimp.lazy_import({'math'}, {'heavy_module': 'package'})
+__getattr__ = lazimp.lazy_import(
+    'math',
+    heavy_module='package',
+)
 ```
 
 Now, when you import the package:

examples/basic/package/__init__.py

@@ -3,4 +3,8 @@
 math: lazimp.ModuleType
 heavy_module: lazimp.ModuleType
 
-__getattr__ = lazimp.lazy_import({'math'}, {'heavy_module': 'package'})
+# __getattr__ = lazimp.lazy_import('math', 'package.heavy_module')
+__getattr__ = lazimp.lazy_import(
+    'math',
+    heavy_module='package',
+)

src/lazimp/__init__.py

@@ -1,34 +1,61 @@
 import collections.abc
 
-# noinspection PyUnresolvedReferences
 from types import ModuleType
 
 
 def lazy_import(
-        bare_import: collections.abc.Container | None = None,
-        sub_import: collections.abc.Mapping[str, str | None] | None = None,
+        *imports: str,
         aliases: collections.abc.Mapping[str, str] | None = None,
-):
-    if bare_import is None:
-        bare_import = set()
-    if sub_import is None:
-        sub_import = {}
+        sub_import: collections.abc.Mapping[str, str] | None = None,
+        **kw_sub_import: str,
+) -> collections.abc.Callable[[str], ModuleType]:
+    """
+    Lazily import modules.
+
+    >>> lazy_import('package', 'package_2', 'package_3', ...)
+    >>> lazy_import('package.submodule')  # fail ATM
+    >>> lazy_import(submodule='package')
+    >>> lazy_import(sub_import={'submodule':'package'})
+    >>> lazy_import('module', aliases={'md': 'module'})
+    >>> lazy_import(aliases={'smd': 'submodule'}, submodule='package')
+    >>> lazy_import(
+            'module',
+            aliases={'md': 'module', 'smd': 'submodule'},
+            submodule='package',
+        )
+    >>> lazy_import(
+            'module',
+            aliases={'md': 'module', 'smd': 'submodule', 'smd2': 'submodule2'},
+            submodule='package',
+            sub_import={'submodule2':'package'}
+        )
+
+    :param 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
 
     import functools
     import importlib
 
     @functools.cache
-    def getattr_(name: str):
+    def getattr_(name: str) -> ModuleType:
         name = aliases.get(name, name)
 
-        if name not in bare_import and name not in sub_import:
+        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 bare_import:
+        if name in imports:
             try:
                 return importlib.import_module(name)
             except ModuleNotFoundError as e: