Add resilience to API calls

glassfrog/client.py

@@ -2,10 +2,15 @@
 import os
 
 import requests
+from retrying import retry
 
 from glassfrog import exceptions
 
 
+def retry_if_conn_error(exception):
+    return isinstance(exception, ConnectionError) or isinstance(exception, requests.HTTPError)
+
+
 class GlassFrogClient:
     _URL = 'https://api.glassfrog.com/api/v3'
     _TOKEN = os.environ.get('GLASSFROG_API_TOKEN')
@@ -31,9 +36,10 @@ def get(cls, resource, id=None, from_resource=None):
         else:
             url = f'{cls._URL}/{resource}'
 
-        response = requests.get(
-            url=url,
-            headers=cls._get_headers(),
-        )
+        with retry(stop_max_attempt_number=3, retry_on_exception=retry_if_conn_error):
+            response = requests.get(
+                url=url,
+                headers=cls._get_headers(),
+            )
         response.raise_for_status()
         return response.json()

pyproject.toml

@@ -11,6 +11,7 @@ packages = [
 [tool.poetry.dependencies]
 python = ">=3.7,<3.10"
 requests = "*"
+retrying = "*"
 
 
 [tool.poetry.dev-dependencies]