clerk-sdk 0.2.1__py3-none-any.whl → 0.2.4__py3-none-any.whl

clerk/gui_automation/decorators/gui_automation.py

@@ -24,8 +24,10 @@ global_ws: Union[ClientConnection, None] = None
 clerk_client = RPAClerk()
 wss_uri = "wss://agent-manager.f-one.group/action"
 
-CLIENT_TIMEOUT_MINS = 1
-MAX_CLIENT_AVAILABILITY_RETRIES = 60
+REMOTE_DEVICE_ALLOCATION_TIMEOUT = int(os.getenv("REMOTE_DEVICE_ALLOCATION_TIMEOUT", 1))
+REMOTE_DEVICE_ALLOCATION_MAX_TRIES = int(
+    os.getenv("REMOTE_DEVICE_ALLOCATION_MAX_TRIES", 60)
+)
 
 
 def _allocate_remote_device(
@@ -39,21 +41,21 @@ def _allocate_remote_device(
             remote_device = clerk_client.allocate_remote_device(
                 group_name=group_name, run_id=run_id
             )
+            os.environ["REMOTE_DEVICE_ID"] = remote_device.id
+            os.environ["REMOTE_DEVICE_NAME"] = remote_device.name
+            logger.debug(f"Remote device allocated: {remote_device.name}")
+            return remote_device
+
         except NoClientsAvailable:
             logger.warning(
-                f"No clients are available for {group_name} group. Initiating a {CLIENT_TIMEOUT_MINS} minute wait. Retry count: {retries}"
+                f"No clients are available for {group_name} group. Initiating a {REMOTE_DEVICE_ALLOCATION_TIMEOUT} minute wait. Retry count: {retries}"
             )
-            retries += 1
-            if retries == MAX_CLIENT_AVAILABILITY_RETRIES:
+            if retries >= REMOTE_DEVICE_ALLOCATION_MAX_TRIES:
                 raise ClientAvailabilityTimeout(
-                    f"No clients available for {group_name} group after {CLIENT_TIMEOUT_MINS*MAX_CLIENT_AVAILABILITY_RETRIES} minutes"
+                    f"No clients available for {group_name} group after {REMOTE_DEVICE_ALLOCATION_TIMEOUT * REMOTE_DEVICE_ALLOCATION_MAX_TRIES} minutes"
                 )
-            time.sleep(CLIENT_TIMEOUT_MINS)
-
-        os.environ["REMOTE_DEVICE_ID"] = remote_device.id
-        os.environ["REMOTE_DEVICE_NAME"] = remote_device.name
-        logger.debug(f"Remote device allocated: {remote_device.name}")
-        return remote_device
+            time.sleep(REMOTE_DEVICE_ALLOCATION_TIMEOUT * 60)
+            retries += 1
 
 
 def _deallocate_target(
@@ -65,7 +67,9 @@ def _deallocate_target(
     os.environ.pop("REMOTE_DEVICE_NAME", None)
 
 
-def gui_automation():
+def gui_automation(
+    reserve_client: bool = False,
+):
     """
     Decorator that:
       • Allocates a remote device,
@@ -88,6 +92,7 @@ def gui_automation():
         @functools.wraps(func)
         def wrapper(payload: ClerkCodePayload, *args, **kwargs):
             global global_ws
+            force_deallocate = False
             os.environ["PROC_ID"] = payload.run_id
 
             remote_device = _allocate_remote_device(
@@ -116,9 +121,11 @@ def gui_automation():
 
             except Exception as e:
                 os.environ.pop("PROC_ID", None)
+                force_deallocate = True
                 raise
             finally:
-                _deallocate_target(clerk_client, remote_device, payload.run_id)
+                if not reserve_client or force_deallocate:
+                    _deallocate_target(clerk_client, remote_device, payload.run_id)
 
                 if global_ws and global_ws.state is State.OPEN:
                     close_task = event_loop.create_task(close_ws_connection(global_ws))

{clerk_sdk-0.2.1.dist-info → clerk_sdk-0.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clerk-sdk
-Version: 0.2.1
+Version: 0.2.4
 Summary: Library for interacting with Clerk
 Home-page: https://github.com/F-ONE-Group/clerk_pypi
 Author: F-ONE Group

{clerk_sdk-0.2.1.dist-info → clerk_sdk-0.2.4.dist-info}/RECORD

@@ -14,7 +14,7 @@ clerk/gui_automation/client_actor/client_actor.py,sha256=Me3NHc0nMA7Y5CUcnE0ZHT1
 clerk/gui_automation/client_actor/exception.py,sha256=zdnImHZ88yf52Xq3aMHivEU3aJg-r2c-r8x8XZnI3ic,407
 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=efB38zIcHsDubfnRgc_o1MgYEJAH7Sfv9oqKB7toZW4,4690
+clerk/gui_automation/decorators/gui_automation.py,sha256=O0zZTNeJT_eH1AfWt3h7BWWsuZhqoGo_eIzsu0pmzoQ,5061
 clerk/gui_automation/exceptions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clerk/gui_automation/exceptions/agent_manager.py,sha256=KFWFWQ7X_8Z9XG__SMzb1jKI3yNoVTPJAXzW5O-fSXc,101
 clerk/gui_automation/exceptions/websocket.py,sha256=-MdwSwlf1hbnu55aDgk3L1znkTZ6xJS6po5VpEGD9ck,117
@@ -42,8 +42,8 @@ clerk/models/response_model.py,sha256=R62daUN1YVOwgnrh_epvFRsQcOwT7R4u97l73egvm-
 clerk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clerk/utils/logger.py,sha256=_jYOSEc1WCsY9DvU8cIWUPmz41QNjpHTEZ7WrUnmFPA,3650
 clerk/utils/save_artifact.py,sha256=Lwl3l3QqKnSFLFpfEEa-dJ-dV9OxvCU3ADkuYZ18U-Q,1117
-clerk_sdk-0.2.1.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
-clerk_sdk-0.2.1.dist-info/METADATA,sha256=QEgrEZmgLaFSscyy0ob9Sr5Y4npmMXFbxF-kPxsk8lE,3508
-clerk_sdk-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-clerk_sdk-0.2.1.dist-info/top_level.txt,sha256=99eQiU6d05_-f41tmSFanfI_SIJeAdh7u9m3LNSfcv4,6
-clerk_sdk-0.2.1.dist-info/RECORD,,
+clerk_sdk-0.2.4.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
+clerk_sdk-0.2.4.dist-info/METADATA,sha256=QXyRQO8-rueo0kDZJqTiFqQotcnfzQG2JD7O-_LtdUI,3508
+clerk_sdk-0.2.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+clerk_sdk-0.2.4.dist-info/top_level.txt,sha256=99eQiU6d05_-f41tmSFanfI_SIJeAdh7u9m3LNSfcv4,6
+clerk_sdk-0.2.4.dist-info/RECORD,,