feat: ✨ load_modules_with_keywords function

Author: remimd

injection/integrations/__init__.py

@@ -1,11 +0,0 @@
-from importlib.util import find_spec
-from typing import Literal
-
-__all__ = ("_is_installed",)
-
-
-def _is_installed(package: str, needed_for: object, /) -> Literal[True]:
-    if find_spec(package) is None:
-        raise RuntimeError(f"To use `{needed_for}`, {package} must be installed.")
-
-    return True

injection/integrations/fastapi.py

@@ -2,15 +2,13 @@
 from types import GenericAlias
 from typing import Any, TypeAliasType
 
+from fastapi import Depends
+
 from injection import Module, mod
 from injection.exceptions import InjectionError
-from injection.integrations import _is_installed
 
 __all__ = ("Inject",)
 
-if _is_installed("fastapi", __name__):
-    from fastapi import Depends
-
 
 def Inject[T](  # noqa: N802
     cls: type[T] | TypeAliasType | GenericAlias,

injection/utils.py

@@ -1,13 +1,15 @@
-from collections.abc import Callable, Iterator
+from collections.abc import Callable, Iterable, Iterator
 from contextlib import contextmanager
 from importlib import import_module
+from importlib.util import find_spec
 from pkgutil import walk_packages
 from types import ModuleType as PythonModule
 from typing import ContextManager
 
+from injection import __name__ as injection_package_name
 from injection import mod
 
-__all__ = ("load_packages", "load_profile")
+__all__ = ("load_modules_with_keywords", "load_packages", "load_profile")
 
 
 def load_profile(*names: str) -> ContextManager[None]:
@@ -33,6 +35,36 @@ def cleaner() -> Iterator[None]:
     return cleaner()
 
 
+def load_modules_with_keywords(
+    *packages: PythonModule | str,
+    keywords: Iterable[str] | None = None,
+) -> dict[str, PythonModule]:
+    """
+    Function to import modules from a Python package if one of the keywords is contained in the Python script.
+    The default keywords are:
+        - `from injection`
+        - `import injection`
+    """
+
+    if keywords is None:
+        keywords = f"from {injection_package_name}", f"import {injection_package_name}"
+
+    b_keywords = tuple(keyword.encode() for keyword in keywords)
+
+    def predicate(module_name: str) -> bool:
+        if (spec := find_spec(module_name)) and (module_path := spec.origin):
+            with open(module_path, "rb") as script:
+                for line in script:
+                    line = b" ".join(line.split(b" ")).strip()
+
+                    if line and any(keyword in line for keyword in b_keywords):
+                        return True
+
+        return False
+
+    return load_packages(*packages, predicate=predicate)
+
+
 def load_packages(
     *packages: PythonModule | str,
     predicate: Callable[[str], bool] = lambda module_name: True,