Support output=void and create a new AsyncResponse object

pmcq · pmcq · commit 657f19ced11f · 2016-02-19T15:29:33.000-08:00
diff --git a/Algorithmia/algorithm.py b/Algorithmia/algorithm.py
@@ -2,6 +2,7 @@
 
 import base64
 import re
+from async_response import AsyncResponse
 
 class algorithm(object):
     def __init__(self, client, algoRef, **query_parameters):
@@ -23,6 +24,8 @@ def __init__(self, client, algoRef, **query_parameters):
     def pipe(self, input1):
         if self.output == 'raw':
             return self.__postRawOutput(input1)
+        elif self.output == 'void':
+            return self.__postVoidOutput(input1)
         else:
             responseJson = self.client.postJsonHelper(self.url, input1, **self.query_parameters)
 
@@ -50,4 +53,11 @@ def __postRawOutput(self, input1):
             elif response.status_code == 500:
                 raise Exception(response.text)
             else:
-                return response.text
+                return response.text
+
+    def __postVoidOutput(self, input1):
+            responseJson = self.client.postJsonHelper(self.url, input1, **self.query_parameters)
+            if 'error' in responseJson:
+                raise Exception(responseJson['error']['message'])
+            else:
+                return AsyncResponse(responseJson)
diff --git a/Algorithmia/async_response.py b/Algorithmia/async_response.py
@@ -0,0 +1,10 @@
+class AsyncResponse(object):
+	'''
+	Response from the API for an asynchronous request (output=void)
+	'''
+	def __init__(self, server_response):
+		self.async_protocol = server_response['async']
+		self.request_id = server_response['request_id']
+
+	def __repr__(self):
+		return 'AsyncResponse(async_protocol=%s, request_id=%s)' % (self.async_protocol, self.request_id) 
