Add method to update a disk object size

[ganeti-local] / lib / luxi.py

diff --git a/lib/luxi.py b/lib/luxi.py
index 7855434..df9ed89 100644 (file)
--- a/lib/luxi.py
+++ b/lib/luxi.py
@@ -68,6 +68,19 @@ class DecodingError(ProtocolError):
   """Decoding failure on the receiving side"""
 
 
+class RequestError(ProtocolError):
+  """Error on request
+
+  This signifies an error in the request format or request handling,
+  but not (e.g.) an error in starting up an instance.
+
+  Some common conditions that can trigger this exception:
+    - job submission failed because the job data was wrong
+    - query failed because required fields were missing
+
+  """
+
+
 def SerializeJob(job):
   """Convert a job description to a string format.
 
@@ -244,12 +257,29 @@ class Client(object):
       data = simplejson.loads(result)
     except Exception, err:
       raise ProtocolError("Error while deserializing response: %s" % str(err))
+    if (not isinstance(data, dict) or
+        'success' not in data or
+        'result' not in data):
+      raise DecodingError("Invalid response from server: %s" % str(data))
     return data
 
   def SubmitJob(self, job):
     """Submit a job"""
-    return self.SendRequest(REQ_SUBMIT, SerializeJob(job))
+    result = self.SendRequest(REQ_SUBMIT, SerializeJob(job))
+    if not result['success']:
+      raise RequestError(result['result'])
+    return result['result']
 
   def Query(self, data):
     """Make a query"""
-    return self.SendRequest(REQ_QUERY, data)
+    result = self.SendRequest(REQ_QUERY, data)
+    if not result['success']:
+      raise RequestError(result[result])
+    result = result['result']
+    if data["object"] == "jobs":
+      # custom job processing of query values
+      for row in result:
+        for idx, field in enumerate(data["fields"]):
+          if field == "op_list":
+            row[idx] = [opcodes.OpCode.LoadOpCode(i) for i in row[idx]]
+    return result