From 4897613e9143e8f09b6bef7080e2e7b5f33189b1 Mon Sep 17 00:00:00 2001 From: FabioFleitas Date: Fri, 5 Feb 2016 13:00:56 -0800 Subject: [PATCH] parse_response now handles 504 Gateway Timeout errors --- lob/api_requestor.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/lob/api_requestor.py b/lob/api_requestor.py index ed8c64d..ea5b131 100644 --- a/lob/api_requestor.py +++ b/lob/api_requestor.py @@ -22,6 +22,10 @@ def __init__(self, key=None): self.api_key = key or lob.api_key def parse_response(self, resp): + if resp.status_code == 504: + raise error.APIConnectionError(resp.content or resp.reason, # pragma: no cover + resp.content, resp.status_code, resp) + payload = resp.json() if resp.status_code == 200: return payload