clerk-sdk 0.2.2__tar.gz → 0.2.5__tar.gz

{clerk_sdk-0.2.2/clerk_sdk.egg-info → clerk_sdk-0.2.5}/PKG-INFO

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

{clerk_sdk-0.2.2 → clerk_sdk-0.2.5}/clerk/gui_automation/decorators/gui_automation.py

@@ -24,8 +24,12 @@ 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", 60)
+)
+REMOTE_DEVICE_ALLOCATION_MAX_TRIES = int(
+    os.getenv("REMOTE_DEVICE_ALLOCATION_MAX_TRIES", 60)
+)
 
 
 def _allocate_remote_device(
@@ -46,14 +50,14 @@ def _allocate_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} seconds 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} seconds"
                 )
-            time.sleep(CLIENT_TIMEOUT_MINS * 60)
+            time.sleep(REMOTE_DEVICE_ALLOCATION_TIMEOUT)
+            retries += 1
 
 
 def _deallocate_target(
@@ -65,7 +69,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 +94,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 +123,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.2 → clerk_sdk-0.2.5/clerk_sdk.egg-info}/PKG-INFO

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

{clerk_sdk-0.2.2 → clerk_sdk-0.2.5}/setup.py

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