clerk-sdk 0.5.3__tar.gz → 0.5.5__tar.gz

{clerk_sdk-0.5.3 → clerk_sdk-0.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clerk-sdk
-Version: 0.5.3
+Version: 0.5.5
 Summary: Library for interacting with Clerk
 Project-URL: Homepage, https://github.com/F-ONE-Group/clerk_pypi
 Author-email: F-One <contact@f-one.group>

{clerk_sdk-0.5.3 → clerk_sdk-0.5.5}/clerk/__init__.py

@@ -1,4 +1,4 @@
 from .client import Clerk
 
 
-__version__ = "0.5.3"
+__version__ = "0.5.5"

{clerk_sdk-0.5.3 → clerk_sdk-0.5.5}/clerk/gui_automation/client_actor/client_actor.py

@@ -19,7 +19,12 @@ from .model import (
 )
 
 from .model import PerformActionResponse, ActionStates
-from .exception import PerformActionException, GetScreenError
+from .exception import (
+    PerformActionException,
+    GetScreenError,
+    AckTimeoutError,
+    ActionTimeoutError,
+)
 
 
 async def before_retry(details: Any):
@@ -35,6 +40,13 @@ async def before_retry(details: Any):
     max_tries=5,
     on_backoff=before_retry,
 )
+@backoff.on_exception(
+    backoff.constant,
+    (ActionTimeoutError, AckTimeoutError),
+    interval=10,
+    max_tries=2,
+    on_backoff=before_retry,
+)
 async def _perform_action_ws(payload: Dict[str, Any]) -> PerformActionResponse:
     """Perform an action over a WebSocket connection.
 
@@ -57,15 +69,21 @@ async def _perform_action_ws(payload: Dict[str, Any]) -> PerformActionResponse:
         # 2. wait for ack message
         try:
             ack = await asyncio.wait_for(global_ws.recv(), 10)
-            if ack == "OK":
-                action_info = await asyncio.wait_for(global_ws.recv(), 10)
-                return PerformActionResponse(**json.loads(action_info))
-            else:
-                raise RuntimeError("Received ACK != OK")
+        except asyncio.TimeoutError:
+            raise AckTimeoutError("Timeout waiting for ACK message")
         except WebSocketException as e:
             raise e
-        except asyncio.TimeoutError:
-            raise RuntimeError("The ack message did not arrive.")
+
+        if ack == "OK":
+            try:
+                action_info = await asyncio.wait_for(global_ws.recv(), 60)
+            except asyncio.TimeoutError:
+                raise ActionTimeoutError("Timeout waiting for action response")
+            except WebSocketException as e:
+                raise e
+            return PerformActionResponse(**json.loads(action_info))
+        else:
+            raise RuntimeError("Received ACK != OK")
     else:
         raise RuntimeError("The Websocket has not been initiated.")
 

{clerk_sdk-0.5.3 → clerk_sdk-0.5.5}/clerk/gui_automation/client_actor/exception.py

@@ -20,3 +20,19 @@ class WebSocketConnectionFailed(Exception):
     """
 
     pass
+
+
+class AckTimeoutError(Exception):
+    """
+    Timeout waiting for ACK message from WebSocket.
+    """
+
+    pass
+
+
+class ActionTimeoutError(Exception):
+    """
+    Timeout waiting for action response from WebSocket.
+    """
+
+    pass

{clerk_sdk-0.5.3 → clerk_sdk-0.5.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "clerk-sdk"
-version = "0.5.3"
+version = "0.5.5"
 description = "Library for interacting with Clerk"
 readme = "README.md"
 authors = [

{clerk_sdk-0.5.3 → clerk_sdk-0.5.5}/setup.py

@@ -13,7 +13,7 @@ gui_requirements = get_requirements("./clerk/gui_automation")
 
 setup(
     name="clerk-sdk",
-    version="0.5.3",
+    version="0.5.5",
     description="Library for interacting with Clerk",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",