wherobots-python-dbapi 0.23.0__tar.gz → 0.23.2__tar.gz

{wherobots_python_dbapi-0.23.0 → wherobots_python_dbapi-0.23.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wherobots-python-dbapi
-Version: 0.23.0
+Version: 0.23.2
 Summary: Python DB-API driver for Wherobots DB
 Project-URL: Homepage, https://github.com/wherobots/wherobots-python-dbapi-driver
 Project-URL: Tracker, https://github.com/wherobots/wherobots-python-dbapi-driver/issues
@@ -13,7 +13,7 @@ Requires-Dist: packaging
 Requires-Dist: pandas
 Requires-Dist: pandas-stubs>=2.0.3.230814
 Requires-Dist: pyarrow>=14.0.2
-Requires-Dist: requests>=2.31.0
+Requires-Dist: requests>=2.32.5
 Requires-Dist: strenum<0.5,>=0.4.15
 Requires-Dist: tenacity>=8.2.3
 Requires-Dist: types-requests>=2.32.0.20241016

{wherobots_python_dbapi-0.23.0 → wherobots_python_dbapi-0.23.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wherobots-python-dbapi"
-version = "0.23.0"
+version = "0.23.2"
 description = "Python DB-API driver for Wherobots DB"
 authors = [{ name = "Maxime Petazzoni", email = "max@wherobots.com" }]
 requires-python = ">=3.10, <4"
@@ -8,7 +8,7 @@ readme = "README.md"
 license = "Apache-2.0"
 dependencies = [
     "packaging",
-    "requests>=2.31.0",
+    "requests>=2.32.5",
     "websockets>=13.0",
     "tenacity>=8.2.3",
     "cbor2>=5.6.3",

{wherobots_python_dbapi-0.23.0 → wherobots_python_dbapi-0.23.2}/wherobots/db/driver.py

@@ -46,6 +46,10 @@ apilevel = "2.0"
 threadsafety = 1
 paramstyle: Final[str] = PARAM_STYLE
 
+# HTTP status codes that indicate transient server-side issues and should be retried.
+# This follows the industry-standard set used by urllib3.util.Retry's status_forcelist.
+TRANSIENT_HTTP_STATUS_CODES = {429, 502, 503, 504}
+
 
 def gen_user_agent_header():
     try:
@@ -135,9 +139,16 @@ def connect(
     @tenacity.retry(
         stop=tenacity.stop_after_delay(wait_timeout),
         wait=tenacity.wait_exponential(multiplier=1, min=1, max=5),
-        retry=tenacity.retry_if_not_exception_type(
-            (requests.HTTPError, OperationalError)
+        retry=(
+            tenacity.retry_if_exception(
+                lambda e: (
+                    isinstance(e, requests.HTTPError)
+                    and e.response.status_code in TRANSIENT_HTTP_STATUS_CODES
+                )
+            )
+            | tenacity.retry_if_exception_type(tenacity.TryAgain)
         ),
+        reraise=True,
     )
     def get_session_uri() -> str:
         r = requests.get(session_id_url, headers=headers)

{wherobots_python_dbapi-0.23.0 → wherobots_python_dbapi-0.23.2}/wherobots/db/runtime.py

@@ -2,6 +2,7 @@ from enum import Enum
 
 
 class Runtime(Enum):
+    # CPU/General purpose runtimes
     MICRO = "micro"
     TINY = "tiny"
     SMALL = "small"
@@ -11,18 +12,24 @@ class Runtime(Enum):
     XX_LARGE = "2x-large"
     XXXX_LARGE = "4x-large"
 
-    # HIMEM
+    # HIMEM/Memory-optimized runtimes
     MEDIUM_HIMEM = "medium-himem"
     LARGE_HIMEM = "large-himem"
     X_LARGE_HIMEM = "x-large-himem"
     XX_LARGE_HIMEM = "2x-large-himem"
     XXXX_LARGE_HIMEM = "4x-large-himem"
 
-    # GPU
+    # HICPU/Compute-optimized runtimes
+    X_LARGE_HICPU = "x-large-hicpu"
+    XX_LARGE_HICPU = "2x-large-hicpu"
+
+    # GPU-accelerated runtimes
     MICRO_A10_GPU = "micro-a10-gpu"
     TINY_A10_GPU = "tiny-a10-gpu"
     SMALL_A10_GPU = "small-a10-gpu"
     MEDIUM_A10_GPU = "medium-a10-gpu"
+    LARGE_A10_GPU = "large-a10-gpu"
+    X_LARGE_A10_GPU = "x-large-a10-gpu"
 
 
 _NAME_TO_ENUM = {runtime.value: runtime for runtime in Runtime}