Fix issue dbt-labs#4178

Allow retries when the answer is None
b-per · Nov 8, 2021 · b98fb7f · b98fb7f
1 parent dd84f9a
commit b98fb7f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/core/dbt/clients/registry.py b/core/dbt/clients/registry.py
@@ -30,6 +30,8 @@ def _get(path, registry_base_url=None):
     logger.debug('Response from registry: GET {} {}'.format(url,
                                                             resp.status_code))
     resp.raise_for_status()
+    if resp is None:
+        raise requests.exceptions.InvalidJSONError('Request error: The response is None', response=resp)
     return resp.json()
 
 

diff --git a/core/dbt/utils.py b/core/dbt/utils.py
@@ -615,7 +615,7 @@ def _connection_exception_retry(fn, max_attempts: int, attempt: int = 0):
     """
     try:
         return fn()
-    except (requests.exceptions.ConnectionError, requests.exceptions.Timeout) as exc:
+    except (requests.exceptions.ConnectionError, requests.exceptions.Timeout, requests.exceptions.InvalidJSONError) as exc:
         if attempt <= max_attempts - 1:
             logger.debug('Retrying external call. Attempt: ' +
                          f'{attempt} Max attempts: {max_attempts}')