clerk-sdk 0.5.4__py3-none-any.whl → 0.5.6__py3-none-any.whl

clerk/__init__.py

@@ -1,4 +1,4 @@
 from .client import Clerk
 
 
-__version__ = "0.5.4"
+__version__ = "0.5.6"

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.
 
@@ -56,16 +68,22 @@ 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(), 60)
-                return PerformActionResponse(**json.loads(action_info))
-            else:
-                raise RuntimeError("Received ACK != OK")
+            ack = await asyncio.wait_for(global_ws.recv(), 120)
+        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(), 120)
+            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/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.4.dist-info → clerk_sdk-0.5.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clerk-sdk
-Version: 0.5.4
+Version: 0.5.6
 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.4.dist-info → clerk_sdk-0.5.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-clerk/__init__.py,sha256=sQnCEW-ctm8zuDSgan3OWyjXaHvo3pZCgzRADiavzbI,50
+clerk/__init__.py,sha256=htiVGLGz-FhPfTF3lUtC9DGzouJJEiqE9G8SbxLOHDk,50
 clerk/base.py,sha256=lbFTdpdDfsmYIQUFH93S1aw0-L6GNJwAcubW1tdMFX4,3967
 clerk/client.py,sha256=JGhDUb4m8R7ZTHLxFsAwAw0vNA1kxewg5-GC6qOFYDk,5244
 clerk/decorator/__init__.py,sha256=yGGcS17VsZ7cZ-hVGCm3I3vGDJMiJIAqmDGzriIi0DI,65
@@ -30,8 +30,8 @@ clerk/gui_automation/action_model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeR
 clerk/gui_automation/action_model/model.py,sha256=yzaCyEMOH3YMkPBf6IwUMuu69-xyf78HzmthiewgWQY,3811
 clerk/gui_automation/action_model/utils.py,sha256=xzFxgN-bTK6HKGS7J-esQZ-ePj_yG72T-2ZVhcWvKjw,798
 clerk/gui_automation/client_actor/__init__.py,sha256=SVuL6-oo1Xc0oJkjMKrO6mJwpPGjrCLKhDV6r2Abtf8,66
-clerk/gui_automation/client_actor/client_actor.py,sha256=05gju2A7kvklO1vjc2v6UJ_jKLFUXDQuH7952Qq6loE,5475
-clerk/gui_automation/client_actor/exception.py,sha256=zdnImHZ88yf52Xq3aMHivEU3aJg-r2c-r8x8XZnI3ic,407
+clerk/gui_automation/client_actor/client_actor.py,sha256=MaejYln5Ya21L5M_DNmE8xsoW9LwnkqGnd_SL0CNwCA,5882
+clerk/gui_automation/client_actor/exception.py,sha256=Z66TwaoplRL3VKy3y6V-G0tnQFkjytXEFUQMhQDHwJo,642
 clerk/gui_automation/client_actor/model.py,sha256=wVpFCi1w2kh4kAV8oNx489vf_SLUQnqhc02rFD5NIJA,6335
 clerk/gui_automation/decorators/__init__.py,sha256=OCgXStEumscgT-RyVy5OKS7ml1w9y-lEnjCVnxuRnQs,43
 clerk/gui_automation/decorators/gui_automation.py,sha256=CwfDmTP9d4BOEsQS4_YGwOoz1gbDwYsA-JVnXz25Cto,3680
@@ -63,8 +63,8 @@ clerk/models/ui_operator.py,sha256=mKTJUFZgv7PeEt5oys28HVZxHOJsofmRQOcRpqj0dbU,2
 clerk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clerk/utils/logger.py,sha256=NrMIlJfVmRjjRw_N_Jngkl0qqv7btXUbg5wxcRmFEH4,3800
 clerk/utils/save_artifact.py,sha256=94aYkYNVGcSUaSWZmdjiY6Oc-3yCKb2XWCZ56IAXQqk,1158
-clerk_sdk-0.5.4.dist-info/METADATA,sha256=bn-CQWNQvCxpMBSl2mrPC-10sVtwvmUrt0ekLQ0FcfU,9635
-clerk_sdk-0.5.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-clerk_sdk-0.5.4.dist-info/entry_points.txt,sha256=VoUmW07sRRSioms5pqQ4A6CYxNEyhGA93GtyBlB_wGw,53
-clerk_sdk-0.5.4.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
-clerk_sdk-0.5.4.dist-info/RECORD,,
+clerk_sdk-0.5.6.dist-info/METADATA,sha256=gzzcJIy0xp2JJxoyby4AahaXMIgVpbbxA0iYR8HRjXc,9635
+clerk_sdk-0.5.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+clerk_sdk-0.5.6.dist-info/entry_points.txt,sha256=VoUmW07sRRSioms5pqQ4A6CYxNEyhGA93GtyBlB_wGw,53
+clerk_sdk-0.5.6.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
+clerk_sdk-0.5.6.dist-info/RECORD,,