uipath 2.1.3__py3-none-any.whl → 2.1.5__py3-none-any.whl

uipath/_cli/__init__.py

@@ -10,6 +10,7 @@ from .cli_invoke import invoke as invoke  # type: ignore
 from .cli_new import new as new  # type: ignore
 from .cli_pack import pack as pack  # type: ignore
 from .cli_publish import publish as publish  # type: ignore
+from .cli_push import push as push  # type: ignore
 from .cli_run import run as run  # type: ignore
 
 
@@ -63,3 +64,4 @@ cli.add_command(run)
 cli.add_command(deploy)
 cli.add_command(auth)
 cli.add_command(invoke)
+cli.add_command(push)

uipath/_cli/_auth/_auth_server.py

@@ -1,10 +1,12 @@
+import asyncio
 import http.server
 import json
 import os
 import socketserver
+import threading
 import time
+from typing import Optional
 
-import click
 from dotenv import load_dotenv
 
 from ._oidc_utils import get_auth_config
@@ -117,9 +119,11 @@ class HTTPServer:
         """
         self.current_path = os.path.dirname(os.path.abspath(__file__))
         self.port = port
-        self.httpd = None
+        self.httpd: Optional[socketserver.TCPServer] = None
         self.token_data = None
         self.should_shutdown = False
+        self.token_received_event: Optional[asyncio.Event] = None
+        self.loop = None
 
     def token_received_callback(self, token_data):
         """Callback for when a token is received.
@@ -128,7 +132,8 @@ class HTTPServer:
             token_data (dict): The received token data.
         """
         self.token_data = token_data
-        self.should_shutdown = True
+        if self.token_received_event and self.loop:
+            self.loop.call_soon_threadsafe(self.token_received_event.set)
 
     def create_server(self, state, code_verifier, domain):
         """Create and configure the HTTP server.
@@ -149,7 +154,16 @@ class HTTPServer:
         self.httpd = socketserver.TCPServer(("", self.port), handler)
         return self.httpd
 
-    def start(self, state, code_verifier, domain):
+    def _run_server(self):
+        """Run server loop in thread."""
+        try:
+            while not self.should_shutdown and self.httpd:
+                self.httpd.handle_request()
+        except Exception:
+            # Server might be closed, that's fine
+            pass
+
+    async def start(self, state, code_verifier, domain):
         """Start the server.
 
         Args:
@@ -163,12 +177,16 @@ class HTTPServer:
         if not self.httpd:
             self.create_server(state, code_verifier, domain)
 
+        self.token_received_event = asyncio.Event()
+        self.loop = asyncio.get_event_loop()
+
+        # Run server in daemon thread
+        server_thread = threading.Thread(target=self._run_server, daemon=True)
+        server_thread.start()
+
         try:
-            if self.httpd:
-                while not self.should_shutdown:
-                    self.httpd.handle_request()
-        except KeyboardInterrupt:
-            click.echo("Process interrupted by user")
+            # Wait indefinitely for token received event or interrupt
+            await self.token_received_event.wait()
         finally:
             self.stop()
 
@@ -176,6 +194,7 @@ class HTTPServer:
 
     def stop(self):
         """Stop the server gracefully and cleanup resources."""
+        self.should_shutdown = True
         if self.httpd:
             self.httpd.server_close()
             self.httpd = None

uipath/_cli/_auth/auth_config.json

@@ -1,6 +1,6 @@
 {
   "client_id": "36dea5b8-e8bb-423d-8e7b-c808df8f1c00",
   "redirect_uri": "http://localhost:__PY_REPLACE_PORT__/oidc/login",
-  "scope": "offline_access OrchestratorApiUserAccess IdentityServerApi ConnectionService DataService DocumentUnderstanding EnterpriseContextService Directory JamJamApi LLMGateway LLMOps OMS RCS.FolderAuthorization TM.Projects TM.TestCases TM.Requirements TM.TestSets",
+  "scope": "offline_access OrchestratorApiUserAccess StudioWebBackend IdentityServerApi ConnectionService DataService DocumentUnderstanding EnterpriseContextService Directory JamJamApi LLMGateway LLMOps OMS RCS.FolderAuthorization TM.Projects TM.TestCases TM.Requirements TM.TestSets",
   "port": 8104
-}
+}