handle not json responses (#95)

hotgluexyz · Feb 2, 2024 · 63eca0f · 63eca0f
1 parent b36da08
commit 63eca0f
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/tap_quickbooks/quickbooks/rest.py b/tap_quickbooks/quickbooks/rest.py
@@ -63,16 +63,17 @@ def _query_recur(
                     yield record
 
         except HTTPError as ex:
-            response = ex.response.json()
-            if isinstance(response, list) and response[0].get("errorCode") == "QUERY_TIMEOUT":
-                start_date = singer_utils.strptime_with_tz(start_date_str)
-                day_range = (end_date - start_date).days
-                LOGGER.info(
-                    "Quickbooks returned QUERY_TIMEOUT querying %d days of %s",
-                    day_range,
-                    catalog_entry['stream'])
-                retryable = True
-            else:
+            try:
+                response = ex.response.json()
+                if isinstance(response, list) and response[0].get("errorCode") == "QUERY_TIMEOUT":
+                    start_date = singer_utils.strptime_with_tz(start_date_str)
+                    day_range = (end_date - start_date).days
+                    LOGGER.info(
+                        "Quickbooks returned QUERY_TIMEOUT querying %d days of %s",
+                        day_range,
+                        catalog_entry['stream'])
+                    retryable = True
+            except:
                 raise ex
 
         if retryable: