Plumb in get_modules

seddonym · Oct 4, 2024 · 24918d4 · 24918d4
1 parent c0ff6ea
commit 24918d4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/rust/src/lib.rs b/rust/src/lib.rs
@@ -44,6 +44,12 @@ impl GraphWrapper {
         }
     }
 
+    pub fn get_modules(&self) -> HashSet<String> {
+        self._graph.get_modules().iter().map(
+            |module| module.name.clone()
+        ).collect()
+    }
+
     #[allow(unused_variables)]
     #[pyo3(signature = (module, is_squashed = false))]
     pub fn add_module(&mut self, module: &str, is_squashed: bool) {

diff --git a/src/grimp/adaptors/rustgraph.py b/src/grimp/adaptors/rustgraph.py
@@ -20,6 +20,7 @@ def __init__(self) -> None:
 
     @property
     def modules(self) -> Set[str]:
+        self._rustgraph.get_modules()
         return super().modules
 
     def add_module(self, module: str, is_squashed: bool = False) -> None: