Add response format

Author: all-seeing-code

pydgraph/txn.py

@@ -56,14 +56,14 @@ def __init__(self, client, read_only=False, best_effort=False):
         self._read_only = read_only
         self._best_effort = best_effort
 
-    def query(self, query, variables=None, timeout=None, metadata=None, credentials=None, resp_format="json"):
+    def query(self, query, variables=None, timeout=None, metadata=None, credentials=None, resp_format="JSON"):
         """Executes a query operation."""
         req = self.create_request(query=query, variables=variables, resp_format=resp_format)
         return self.do_request(req, timeout=timeout, metadata=metadata, credentials=credentials)
 
-    def async_query(self, query, variables=None, timeout=None, metadata=None, credentials=None):
+    def async_query(self, query, variables=None, timeout=None, metadata=None, credentials=None, resp_format="JSON"):
         """Async version of query."""
-        req = self.create_request(query=query, variables=variables)
+        req = self.create_request(query=query, variables=variables, resp_format=resp_format)
         return self.async_do_request(req, timeout=timeout, metadata=metadata, credentials=credentials)
 
     def mutate(self, mutation=None, set_obj=None, del_obj=None,
@@ -191,10 +191,16 @@ def create_mutation(self, mutation=None, set_obj=None, del_obj=None,
             mutation.cond = cond.encode('utf8')
         return mutation
 
-    def create_request(self, query=None, variables=None, mutations=None, commit_now=None, resp_format="json"):
+    def create_request(self, query=None, variables=None, mutations=None, commit_now=None, resp_format="JSON"):
+        switch(resp_format) {
+        case "RDF":
+            resp_format = api.Request.RespFormat.RDF
+        default:
+            resp_format = api.Request.RespFormat.JSON
+        };
         """Creates a request object"""
         request = api.Request(start_ts = self._ctx.start_ts, commit_now=commit_now,
-                              read_only=self._read_only, best_effort=self._best_effort)
+                              read_only=self._read_only, best_effort=self._best_effort, resp_format = resp_format)
 
         if variables is not None:
             for key, value in variables.items():