Add auto retry logic for the sync github client

Author: Eijebong

src/simple_github/client.py

@@ -13,6 +13,8 @@
 from gql.transport.requests import RequestsHTTPTransport
 from requests import Response as RequestsResponse
 from requests import Session
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
 
 if TYPE_CHECKING:
     from simple_github.auth import Auth
@@ -132,6 +134,13 @@ def _get_requests_session(self) -> Session:
         assert session.transport.session
         return session.transport.session
 
+    def _get_retry_session(self) -> Session:
+        session = self._get_requests_session()
+        retry = Retry(total=5, backoff_factor=1, status_forcelist={500, 502, 503, 504})
+        adapter = HTTPAdapter(max_retries=retry)
+        session.mount("https://", adapter)
+        return session
+
     def request(self, method: str, query: str, **kwargs) -> RequestsResponse:
         """Make a request to Github's REST API.
 
@@ -145,7 +154,7 @@ def request(self, method: str, query: str, **kwargs) -> RequestsResponse:
             Dict: The JSON result of the request.
         """
         url = f"{GITHUB_API_ENDPOINT}/{query.lstrip('/')}"
-        session = self._get_requests_session()
+        session = self._get_retry_session()
 
         with session.request(method, url, **kwargs) as resp:
             return resp

test/test_client.py

@@ -204,6 +204,18 @@ def test_sync_client_rest(responses, sync_client):
         resp.raise_for_status()
 
 
+def test_sync_client_retries_on_5xx(responses, sync_client):
+    client = sync_client
+    url = f"{GITHUB_API_ENDPOINT}/octocat"
+
+    responses.get(url, status=502)
+    responses.get(url, status=200, json={"answer": 42})
+
+    resp = client.get("/octocat")
+    result = resp.json()
+    assert result == {"answer": 42}
+
+
 @pytest.mark.asyncio
 async def test_async_client_rest_with_text(aioresponses, async_client):
     client = async_client