Add retries to API calls

Author: RealOrangeOne

heroku_scheduled_scaling/__main__.py

@@ -5,7 +5,7 @@
 import sentry_sdk
 
 from .scale import scale_app
-from .utils import get_heroku_apps, get_heroku_client
+from .utils import get_heroku_apps, requests_session
 
 
 def main() -> None:
@@ -15,7 +15,7 @@ def main() -> None:
     apps = get_heroku_apps()
 
     requests_pool_size = (
-        get_heroku_client()._session.adapters["https://"]._pool_connections  # type: ignore[attr-defined]
+        requests_session.adapters["https://"]._pool_connections  # type: ignore[attr-defined]
     )
 
     with concurrent.futures.ThreadPoolExecutor(

heroku_scheduled_scaling/utils.py

@@ -4,8 +4,19 @@
 from typing import Iterable
 
 import heroku3
+import requests
+import requests.adapters
 from heroku3.models.app import App
 
+requests_retry = requests.adapters.Retry(total=3, backoff_factor=0.1)
+requests_session = requests.Session()
+requests_session.mount(
+    "https://", requests.adapters.HTTPAdapter(max_retries=requests_retry)
+)
+requests_session.mount(
+    "http://", requests.adapters.HTTPAdapter(max_retries=requests_retry)
+)
+
 
 def get_apps_for_teams(heroku: heroku3.core.Heroku, teams: list[str]) -> Iterable[App]:
     for team in teams:
@@ -15,7 +26,7 @@ def get_apps_for_teams(heroku: heroku3.core.Heroku, teams: list[str]) -> Iterabl
 
 
 def get_heroku_client() -> heroku3.core.Heroku:
-    return heroku3.from_key(os.environ["HEROKU_API_KEY"])
+    return heroku3.from_key(os.environ["HEROKU_API_KEY"], session=requests_session)
 
 
 def get_heroku_apps() -> list[App]:

stubs/heroku3/__init__.pyi

@@ -1,3 +1,5 @@
+from requests import Session
+
 from .core import Heroku
 
-def from_key(api_key: str) -> Heroku: ...
+def from_key(api_key: str, session: Session) -> Heroku: ...