diff --git a/setra_client/client.py b/setra_client/client.py index 8b024c8eb4d8eb776601cc9530363603648a604c..088300cceb92c7538b03c73483a2d17cabf4715f 100644 --- a/setra_client/client.py +++ b/setra_client/client.py @@ -288,8 +288,10 @@ class SetraClient(object): data=batchdata.json(), headers=headers, return_response=True) - if response.status_code in (409, 202): - return 'OK' + if response.status_code == 202: + return response.content, 'Accepted' + elif response.status_code == 409: + return response.content, "Conflict" else: response.raise_for_status() return response @@ -298,19 +300,19 @@ class SetraClient(object): """ PUT updates an existing batch with vouchers and transactions, if the batch exists in setra, and has status=created, or validation failed. - Returns 404 if no batch was found, 202 for successful update, or 409 if batch was found, but - did not meet the status criteria mentioned above. + Returns 404 if no batch was found, 204 for successful update, or 409 if batch + was found, but did not meet the status criteria mentioned above. """ url = self.urls.put_update_batch() headers = {'Content-Type': 'application/json'} response = self.put(url, - data=batchdata.json(), - headers=headers, - return_response=True) - if response.status_code in (204, 202): - return 'OK' + data=batchdata.json(), + headers=headers, + return_response=True) + if response.status_code == 204: + return response.content, 'No Content' elif response.status_code == 409: - return 'batch_found_but_cannot_be_updated' + return response.content, 'batch_found_but_cannot_be_updated' else: response.raise_for_status() return response