feat: ✨ Add ProfileLoader.required_module_names method

Author: remimd

injection/loaders.py

@@ -149,6 +149,15 @@ class ProfileLoader:
     def __is_empty(self) -> bool:
         return not self.module_subsets
 
+    def required_module_names(self, name: str | None = None, /) -> frozenset[str]:
+        names = {self.module.name}
+
+        if name is not None:
+            names.add(name)
+
+        subsets = (self.__walk_subsets_for(name) for name in names)
+        return frozenset(itertools.chain.from_iterable(subsets))
+
     def init(self) -> Self:
         self.__init_subsets_for(self.module)
         return self
@@ -179,6 +188,12 @@ def __is_initialized(self, module: Module) -> bool:
     def __mark_initialized(self, module: Module) -> None:
         self.__initialized_modules.add(module.name)
 
+    def __walk_subsets_for(self, module_name: str) -> Iterator[str]:
+        yield module_name
+
+        for name in self.module_subsets.get(module_name, ()):
+            yield from self.__walk_subsets_for(name)
+
 
 @runtime_checkable
 class LoadedProfile(Protocol):

tests/loaders/test_profile_loader.py

@@ -8,6 +8,42 @@
 
 
 class TestProfileLoader:
+    def test_required_module_names_with_success_return_frozenset(self):
+        loader = ProfileLoader(
+            {
+                mod().name: ["a", "b", "c"],
+                "dev": ["m", "n", "o"],
+                "b": ["x", "y", "z"],
+                "c": ["i", "j"],
+            }
+        )
+        assert loader.required_module_names() == {
+            mod().name,
+            "a",
+            "b",
+            "c",
+            "x",
+            "y",
+            "z",
+            "i",
+            "j",
+        }
+
+    def test_required_module_names_with_name_return_frozenset(self):
+        loader = ProfileLoader(
+            {
+                mod().name: ["a"],
+                "dev": ["z"],
+                "test": ["i"],
+            }
+        )
+        assert loader.required_module_names("dev") == {
+            mod().name,
+            "dev",
+            "a",
+            "z",
+        }
+
     def test_load_with_success(self):
         profile_name = "test"
         global_profile_name = uuid4().hex