[feat] Add tenacity retry for async paths with max retry of 5

kiritowu · Nov 7, 2024 · 65b1173 · 65b1173
1 parent 1e84b83
commit 65b1173
Show file tree

Hide file tree

Showing 6 changed files with 78 additions and 12 deletions.
diff --git a/nlb_catalogue_client/api/catalogue/get_get_availability_info.py b/nlb_catalogue_client/api/catalogue/get_get_availability_info.py
@@ -2,7 +2,7 @@
 from typing import Any, Dict, Optional, Union
 
 import httpx
-from tenacity import retry, retry_if_result, wait_exponential
+from tenacity import retry, retry_if_result, stop_after_attempt, wait_exponential
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
@@ -125,7 +125,12 @@ def _build_response(
     )
 
 
-@retry(retry=retry_if_result(lambda x: x.status_code == 429), wait=wait_exponential())
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 def sync_detailed(
     *,
     client: AuthenticatedClient,
@@ -245,6 +250,12 @@ def sync(
     ).parsed
 
 
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,

diff --git a/nlb_catalogue_client/api/catalogue/get_get_most_checkouts_trends_titles.py b/nlb_catalogue_client/api/catalogue/get_get_most_checkouts_trends_titles.py
@@ -2,7 +2,7 @@
 from typing import Any, Dict, Optional, Union
 
 import httpx
-from tenacity import retry, retry_if_result, wait_exponential
+from tenacity import retry, retry_if_result, stop_after_attempt, wait_exponential
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
@@ -113,7 +113,12 @@ def _build_response(
     )
 
 
-@retry(retry=retry_if_result(lambda x: x.status_code == 429), wait=wait_exponential())
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 def sync_detailed(
     *,
     client: AuthenticatedClient,
@@ -201,6 +206,12 @@ def sync(
     ).parsed
 
 
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,

diff --git a/nlb_catalogue_client/api/catalogue/get_get_new_titles.py b/nlb_catalogue_client/api/catalogue/get_get_new_titles.py
@@ -2,7 +2,7 @@
 from typing import Any, Dict, List, Optional, Union
 
 import httpx
-from tenacity import retry, retry_if_result, wait_exponential
+from tenacity import retry, retry_if_result, stop_after_attempt, wait_exponential
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
@@ -162,7 +162,12 @@ def _build_response(
     )
 
 
-@retry(retry=retry_if_result(lambda x: x.status_code == 429), wait=wait_exponential())
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 def sync_detailed(
     *,
     client: AuthenticatedClient,
@@ -314,6 +319,12 @@ def sync(
     ).parsed
 
 
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,

diff --git a/nlb_catalogue_client/api/catalogue/get_get_title_details.py b/nlb_catalogue_client/api/catalogue/get_get_title_details.py
@@ -2,7 +2,7 @@
 from typing import Any, Dict, Optional, Union
 
 import httpx
-from tenacity import retry, retry_if_result, wait_exponential
+from tenacity import retry, retry_if_result, stop_after_attempt, wait_exponential
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
@@ -113,7 +113,12 @@ def _build_response(
     )
 
 
-@retry(retry=retry_if_result(lambda x: x.status_code == 429), wait=wait_exponential())
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 def sync_detailed(
     *,
     client: AuthenticatedClient,
@@ -209,6 +214,12 @@ def sync(
     ).parsed
 
 
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,

diff --git a/nlb_catalogue_client/api/catalogue/get_get_titles.py b/nlb_catalogue_client/api/catalogue/get_get_titles.py
@@ -2,7 +2,7 @@
 from typing import Any, Dict, Optional, Union
 
 import httpx
-from tenacity import retry, wait_exponential, retry_if_result
+from tenacity import retry, retry_if_result, stop_after_attempt, wait_exponential
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
@@ -134,7 +134,12 @@ def _build_response(
     )
 
 
-@retry(retry=retry_if_result(lambda x: x.status_code == 429), wait=wait_exponential())
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 def sync_detailed(
     *,
     client: AuthenticatedClient,
@@ -268,6 +273,12 @@ def sync(
     ).parsed
 
 
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,

diff --git a/nlb_catalogue_client/api/catalogue/get_search_titles.py b/nlb_catalogue_client/api/catalogue/get_search_titles.py
@@ -2,7 +2,7 @@
 from typing import Any, Dict, List, Optional, Union
 
 import httpx
-from tenacity import retry, retry_if_result, wait_exponential
+from tenacity import retry, retry_if_result, stop_after_attempt, wait_exponential
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
@@ -169,7 +169,12 @@ def _build_response(
     )
 
 
-@retry(retry=retry_if_result(lambda x: x.status_code == 429), wait=wait_exponential())
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 def sync_detailed(
     *,
     client: AuthenticatedClient,
@@ -323,6 +328,12 @@ def sync(
     ).parsed
 
 
+@retry(
+    retry=retry_if_result(lambda x: x.status_code == 429),
+    wait=wait_exponential(),
+    stop=stop_after_attempt(5),
+    retry_error_callback=lambda x: x.outcome.result() if x.outcome else None,
+)
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,