T9365: add wrappers for reference tree path from config/partial path

2 files changed, 20 insertions, 0 deletions

diff --git a/python/vyos/configtree.py b/python/vyos/configtree.py
index dceb205ab..b8de379eb 100644
--- a/python/vyos/configtree.py
+++ b/python/vyos/configtree.py
@@ -97,6 +97,8 @@ _PROTOTYPES = {
     'get_nodes_of_kind': ([c_void_p, c_char_p, c_char_p], c_char_p),
     'get_rdeps_of_kind': ([c_void_p, c_char_p, c_char_p], c_char_p),
     'get_rdeps_of_kind_data': ([c_void_p, c_char_p, c_char_p], c_char_p),
+    'reference_path_from_config_path': ([c_void_p, c_char_p], c_char_p),
+    'reference_path_from_partial_path': ([c_void_p, c_char_p], c_char_p),
 }
 
 
diff --git a/python/vyos/referencetree.py b/python/vyos/referencetree.py
index 2f50b19d4..0cd969762 100644
--- a/python/vyos/referencetree.py
+++ b/python/vyos/referencetree.py
@@ -115,3 +115,21 @@ class ReferenceTree:
         ).decode()
         sort = sorted(json.loads(res))
         return list(map(tuple, sort)) if as_tuple else sort
+
+    def reference_path_from_config_path(self, path):
+        import json
+
+        check_path(path)
+        path_str = ' '.join(map(str, path)).encode()
+
+        path = self.__lib.reference_path_from_config_path(self.__pointer, path_str)
+        return json.loads(path)
+
+    def reference_path_from_partial_path(self, path):
+        import json
+
+        check_path(path)
+        path_str = ' '.join(map(str, path)).encode()
+
+        path = self.__lib.reference_path_from_partial_path(self.__pointer, path_str)
+        return json.loads(path)