schema: add get_module_from_prefix to module class

This new method from the Module class gives a module given the prefix
defined in the import section.

Signed-off-by: Samuel Gauthier <samuel.gauthier@6wind.com>
Acked-by: Robin Jarry <robin@jarry.cc>

Author: samuel-gauthier

libyang/schema.py

@@ -116,6 +116,13 @@ def imports(self) -> Iterator["Import"]:
         for i in ly_array_iter(self.cdata.parsed.imports):
             yield Import(self.context, i, self)
 
+    def get_module_from_prefix(self, prefix: str) -> Optional["Module"]:
+        for i in self.imports():
+            if i.prefix() != prefix:
+                continue
+            return self.context.get_module(i.name())
+        return None
+
     def __iter__(self) -> Iterator["SNode"]:
         return self.children()
 

tests/test_schema.py

@@ -379,6 +379,7 @@ def test_leaf_type_derived(self):
         mod = t.module()
         self.assertIsNot(mod, None)
         self.assertEqual(mod.name(), "yolo-system")
+        self.assertEqual(mod.get_module_from_prefix("types").name(), "wtf-types")
 
     def test_leaf_type_status(self):
         leaf = next(self.ctx.find_path("/yolo-system:conf/yolo-system:hostname"))