Add get_field method 10/6810/1
authorMichael Marchetti <mmarchetti@sandvine.com>
Wed, 31 Oct 2018 16:03:40 +0000 (16:03 +0000)
committerMichael Marchetti <mmarchetti@sandvine.com>
Wed, 31 Oct 2018 16:03:40 +0000 (16:03 +0000)
Signed-off-by: Michael Marchetti <mmarchetti@sandvine.com>
osmclient/sol005/ns.py

index 90bf8db..c640d00 100644 (file)
@@ -174,7 +174,7 @@ class Ns(object):
                 if not resp or 'id' not in resp:
                     raise ClientException('unexpected response from server - {} '.format(
                                       resp))
-                print(resp['id'])
+                return resp['id']
             else:
                 msg = ""
                 if resp:
@@ -374,3 +374,12 @@ class Ns(object):
                     exc.message)
             raise ClientException(message)
 
+    def get_field(self, ns_name, field):
+        nsr = self.get(ns_name)
+        if nsr is None:
+            raise NotFound("failed to retrieve ns {}".format(ns_name))
+
+        if field in nsr:
+            return nsr[field]
+
+        raise NotFound("failed to find {} in ns {}".format(field, ns_name))