unique_sdk 0.9.12__tar.gz → 0.9.13__tar.gz

{unique_sdk-0.9.12 → unique_sdk-0.9.13}/CHANGELOG.md

@@ -5,6 +5,9 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.9.13] - 2024-10-23
+- Add retry for `5xx` errors, add additional error message.
+
 ## [0.9.12] - 2024-11-21
 - Include original error message in returned exceptions
 

{unique_sdk-0.9.12 → unique_sdk-0.9.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.9.12
+Version: 0.9.13
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -899,6 +899,9 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.9.13] - 2024-10-23
+- Add retry for `5xx` errors, add additional error message.
+
 ## [0.9.12] - 2024-11-21
 - Include original error message in returned exceptions
 

{unique_sdk-0.9.12 → unique_sdk-0.9.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_sdk"
-version = "0.9.12"
+version = "0.9.13"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

{unique_sdk-0.9.12 → unique_sdk-0.9.13}/unique_sdk/_api_resource.py

@@ -22,10 +22,12 @@ retry_dict = {
     "error_messages": [
         "problem proxying the request",
         "Upstream service reached a hard timeout",
+        "Invalid response body from API",
     ],
     "max_retries": 3,
     "backoff_factor": 2,
     "initial_delay": 1,
+    "should_retry_5xx": True,
 }
 
 

{unique_sdk-0.9.12 → unique_sdk-0.9.13}/unique_sdk/_util.py

@@ -200,6 +200,7 @@ def retry_on_error(
     initial_delay: int = 1,
     backoff_factor: int = 2,
     error_class=APIError,
+    should_retry_5xx=False,
 ):
     def decorator(func: Callable) -> Callable:
         @wraps(func)
@@ -209,10 +210,15 @@ def retry_on_error(
                 try:
                     return await func(*args, **kwargs)
                 except Exception as e:
-                    if not any(
+                    should_retry = any(
                         err_msg.lower() in str(e).lower() for err_msg in error_messages
-                    ):
-                        raise e  # Raise the error if none of the messages match
+                    )
+                    # Add 5xx check if `should_retry_5xx` is True
+                    if should_retry_5xx and hasattr(e, "status_code"):
+                        should_retry = should_retry or (500 <= e.status_code < 600)
+
+                    if not should_retry:
+                        raise e  # Raise the error if no retry condition is met
 
                     attempts += 1
                     if attempts >= max_retries:
@@ -229,10 +235,15 @@ def retry_on_error(
                 try:
                     return func(*args, **kwargs)
                 except Exception as e:
-                    if not any(
+                    should_retry = any(
                         err_msg.lower() in str(e).lower() for err_msg in error_messages
-                    ):
-                        raise e  # Raise the error if none of the messages match
+                    )
+                    # Add 5xx check if `should_retry_5xx` is True
+                    if should_retry_5xx and hasattr(e, "status_code"):
+                        should_retry = should_retry or (500 <= e.status_code < 600)
+
+                    if not should_retry:
+                        raise e  # Raise the error if no retry condition is met
 
                     attempts += 1
                     if attempts >= max_retries:

unique_sdk-0.9.13/unique_sdk/_version.py

@@ -0,0 +1 @@
+VERSION = "0.9.13"

unique_sdk-0.9.12/unique_sdk/_version.py

@@ -1 +0,0 @@
-VERSION = "0.9.8"