wherobots-python-dbapi 0.25.4__tar.gz → 0.26.1__tar.gz

{wherobots_python_dbapi-0.25.4 → wherobots_python_dbapi-0.26.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wherobots-python-dbapi
-Version: 0.25.4
+Version: 0.26.1
 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
@@ -225,6 +225,7 @@ You must also specify in which region your SQL session should execute
 into. Wherobots Cloud supports the following compute regions:
 
 * `aws-us-east-1`: AWS US East 1 (N. Virginia)
+* `aws-us-east-2`: AWS US East 2 (Ohio)
 * `aws-us-west-2`: AWS US West 2 (Oregon)
 * `aws-eu-west-1`: AWS EU West 1 (Ireland)
 * `aws-ap-south-1`: AWS AP South 1 (Mumbai)

{wherobots_python_dbapi-0.25.4 → wherobots_python_dbapi-0.26.1}/README.md

@@ -201,6 +201,7 @@ You must also specify in which region your SQL session should execute
 into. Wherobots Cloud supports the following compute regions:
 
 * `aws-us-east-1`: AWS US East 1 (N. Virginia)
+* `aws-us-east-2`: AWS US East 2 (Ohio)
 * `aws-us-west-2`: AWS US West 2 (Oregon)
 * `aws-eu-west-1`: AWS EU West 1 (Ireland)
 * `aws-ap-south-1`: AWS AP South 1 (Mumbai)

{wherobots_python_dbapi-0.25.4 → wherobots_python_dbapi-0.26.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wherobots-python-dbapi"
-version = "0.25.4"
+version = "0.26.1"
 description = "Python DB-API driver for Wherobots DB"
 authors = [{ name = "Maxime Petazzoni", email = "max@wherobots.com" }]
 requires-python = ">=3.10, <4"

{wherobots_python_dbapi-0.25.4 → wherobots_python_dbapi-0.26.1}/wherobots/db/driver.py

@@ -13,6 +13,7 @@ import requests
 import tenacity
 from typing import Final, Union, Dict
 import urllib.parse
+import websockets.exceptions
 import websockets.sync.client
 import certifi
 
@@ -200,17 +201,33 @@ def connect_direct(
     geometry_representation: Union[GeometryRepresentation, None] = None,
 ) -> Connection:
     uri_with_protocol = f"{uri}/{protocol}"
+    ssl_context = ssl.create_default_context()
+    ssl_context.load_verify_locations(certifi.where())
 
-    try:
+    @tenacity.retry(
+        stop=tenacity.stop_after_attempt(5),
+        wait=tenacity.wait_exponential(multiplier=1, min=1, max=5),
+        retry=tenacity.retry_if_exception_type(
+            (
+                ConnectionRefusedError,
+                ConnectionResetError,
+                TimeoutError,
+                websockets.exceptions.InvalidHandshake,
+            )
+        ),
+        reraise=True,
+    )
+    def ws_connect() -> websockets.sync.client.ClientConnection:
         logging.info("Connecting to SQL session at %s ...", uri_with_protocol)
-        ssl_context = ssl.create_default_context()
-        ssl_context.load_verify_locations(certifi.where())
-        ws = websockets.sync.client.connect(
+        return websockets.sync.client.connect(
             uri=uri_with_protocol,
             additional_headers=headers,
             max_size=MAX_MESSAGE_SIZE,
             ssl=ssl_context,
         )
+
+    try:
+        ws = ws_connect()
     except Exception as e:
         raise InterfaceError("Failed to connect to SQL session!") from e
 

{wherobots_python_dbapi-0.25.4 → wherobots_python_dbapi-0.26.1}/wherobots/db/region.py

@@ -4,6 +4,7 @@ from enum import Enum
 class Region(Enum):
     # Americas
     AWS_US_EAST_1 = "aws-us-east-1"
+    AWS_US_EAST_2 = "aws-us-east-2"
     AWS_US_WEST_2 = "aws-us-west-2"
 
     # EMEA