T1598: add a vyos-hostsd operation for retrieving name servers by tag.

1 files changed, 7 insertions, 0 deletions

diff --git a/python/vyos/hostsd_client.py b/python/vyos/hostsd_client.py
index e2f05071b..f009aba98 100644
--- a/python/vyos/hostsd_client.py
+++ b/python/vyos/hostsd_client.py
@@ -30,6 +30,8 @@ class Client(object):
             reply = json.loads(reply_msg)
             if 'error' in reply:
                 raise VyOSHostsdError(reply['error'])
+            else:
+                return reply["data"]
         except zmq.error.Again:
             raise VyOSHostsdError("Could not connect to vyos-hostsd")
 
@@ -60,3 +62,8 @@ class Client(object):
     def delete_name_servers(self, tag):
         msg = {'type': 'name_servers', 'op': 'delete', 'tag': tag}
         self._communicate(msg)
+
+    def get_name_servers(self, tag):
+        msg = {'type': 'name_servers', 'op': 'get', 'tag': tag}
+        return self._communicate(msg)
+