uipath 2.1.85__py3-none-any.whl → 2.1.86__py3-none-any.whl

uipath/_cli/_utils/_project_files.py

@@ -1,13 +1,22 @@
 # type: ignore
+import hashlib
 import json
 import os
 import re
+from pathlib import Path
 from typing import Any, Dict, Optional, Tuple
 
+import click
 from pydantic import BaseModel
 
 from .._utils._console import ConsoleLogger
 from ._constants import is_binary_file
+from ._studio_project import (
+    ProjectFile,
+    ProjectFolder,
+    StudioClient,
+    get_folder_by_name,
+)
 
 try:
     import tomllib
@@ -431,3 +440,116 @@ def files_to_include(
                     )
                 )
     return extra_files
+
+
+def compute_normalized_hash(content: str) -> str:
+    """Compute hash of normalized content.
+
+    Args:
+        content: Content to hash
+
+    Returns:
+        str: SHA256 hash of the normalized content
+    """
+    try:
+        # Try to parse as JSON to handle formatting
+        json_content = json.loads(content)
+        normalized = json.dumps(json_content, indent=2)
+    except json.JSONDecodeError:
+        # Not JSON, normalize line endings
+        normalized = content.replace("\r\n", "\n").replace("\r", "\n")
+
+    return hashlib.sha256(normalized.encode("utf-8")).hexdigest()
+
+
+def collect_files_from_folder(
+    folder: ProjectFolder, base_path: str, files_dict: Dict[str, ProjectFile]
+) -> None:
+    """Recursively collect all files from a folder and its subfolders.
+
+    Args:
+        folder: The folder to collect files from
+        base_path: Base path for file paths
+        files_dict: Dictionary to store collected files
+    """
+    # Add files from current folder
+    for file in folder.files:
+        file_path = os.path.join(base_path, file.name)
+        files_dict[file_path] = file
+
+    # Recursively process subfolders
+    for subfolder in folder.folders:
+        subfolder_path = os.path.join(base_path, subfolder.name)
+        collect_files_from_folder(subfolder, subfolder_path, files_dict)
+
+
+async def download_folder_files(
+    studio_client: StudioClient,
+    folder: ProjectFolder,
+    base_path: Path,
+) -> None:
+    """Download files from a folder recursively.
+
+    Args:
+        studio_client: Studio client
+        folder: The folder to download files from
+        base_path: Base path for local file storage
+    """
+    files_dict: Dict[str, ProjectFile] = {}
+    collect_files_from_folder(folder, "", files_dict)
+    for file_path, remote_file in files_dict.items():
+        local_path = base_path / file_path
+        local_path.parent.mkdir(parents=True, exist_ok=True)
+
+        # Download remote file
+        response = await studio_client.download_file_async(remote_file.id)
+        remote_content = response.read().decode("utf-8")
+        remote_hash = compute_normalized_hash(remote_content)
+
+        if os.path.exists(local_path):
+            # Read and hash local file
+            with open(local_path, "r", encoding="utf-8") as f:
+                local_content = f.read()
+                local_hash = compute_normalized_hash(local_content)
+
+            # Compare hashes
+            if local_hash != remote_hash:
+                styled_path = click.style(str(file_path), fg="cyan")
+                console.warning(f"File {styled_path}" + " differs from remote version.")
+                response = click.prompt("Do you want to overwrite it? (y/n)", type=str)
+                if response.lower() == "y":
+                    with open(local_path, "w", encoding="utf-8", newline="\n") as f:
+                        f.write(remote_content)
+                    console.success(f"Updated {click.style(str(file_path), fg='cyan')}")
+                else:
+                    console.info(f"Skipped {click.style(str(file_path), fg='cyan')}")
+            else:
+                console.info(
+                    f"File {click.style(str(file_path), fg='cyan')} is up to date"
+                )
+        else:
+            # File doesn't exist locally, create it
+            with open(local_path, "w", encoding="utf-8", newline="\n") as f:
+                f.write(remote_content)
+            console.success(f"Downloaded {click.style(str(file_path), fg='cyan')}")
+
+
+async def pull_project(project_id: str, download_configuration: dict[str, Path]):
+    studio_client = StudioClient(project_id)
+
+    with console.spinner("Pulling UiPath project files..."):
+        try:
+            structure = await studio_client.get_project_structure_async()
+            for source_key, destination in download_configuration.items():
+                source_folder = get_folder_by_name(structure, source_key)
+                if source_folder:
+                    await download_folder_files(
+                        studio_client,
+                        source_folder,
+                        destination,
+                    )
+                else:
+                    console.warning(f"No {source_key} folder found in remote project")
+
+        except Exception as e:
+            console.error(f"Failed to pull UiPath project: {str(e)}")

uipath/_cli/cli_pull.py

@@ -11,133 +11,27 @@ It handles:
 
 # type: ignore
 import asyncio
-import hashlib
-import json
 import os
-from typing import Dict, Set
+from pathlib import Path
 
 import click
 
 from ..telemetry import track
 from ._utils._console import ConsoleLogger
 from ._utils._constants import UIPATH_PROJECT_ID
-from ._utils._studio_project import (
-    ProjectFile,
-    ProjectFolder,
-    StudioClient,
-    get_folder_by_name,
-)
+from ._utils._project_files import pull_project
 
 console = ConsoleLogger()
 
 
-def compute_normalized_hash(content: str) -> str:
-    """Compute hash of normalized content.
-
-    Args:
-        content: Content to hash
-
-    Returns:
-        str: SHA256 hash of the normalized content
-    """
-    try:
-        # Try to parse as JSON to handle formatting
-        json_content = json.loads(content)
-        normalized = json.dumps(json_content, indent=2)
-    except json.JSONDecodeError:
-        # Not JSON, normalize line endings
-        normalized = content.replace("\r\n", "\n").replace("\r", "\n")
-
-    return hashlib.sha256(normalized.encode("utf-8")).hexdigest()
-
-
-def collect_files_from_folder(
-    folder: ProjectFolder, base_path: str, files_dict: Dict[str, ProjectFile]
-) -> None:
-    """Recursively collect all files from a folder and its subfolders.
-
-    Args:
-        folder: The folder to collect files from
-        base_path: Base path for file paths
-        files_dict: Dictionary to store collected files
-    """
-    # Add files from current folder
-    for file in folder.files:
-        file_path = os.path.join(base_path, file.name)
-        files_dict[file_path] = file
-
-    # Recursively process subfolders
-    for subfolder in folder.folders:
-        subfolder_path = os.path.join(base_path, subfolder.name)
-        collect_files_from_folder(subfolder, subfolder_path, files_dict)
-
-
-async def download_folder_files(
-    studio_client: StudioClient,
-    folder: ProjectFolder,
-    base_path: str,
-    processed_files: Set[str],
-) -> None:
-    """Download files from a folder recursively.
-
-    Args:
-        studio_client: Studio client
-        folder: The folder to download files from
-        base_path: Base path for local file storage
-        processed_files: Set to track processed files
-    """
-    files_dict: Dict[str, ProjectFile] = {}
-    collect_files_from_folder(folder, "", files_dict)
-
-    for file_path, remote_file in files_dict.items():
-        local_path = os.path.join(base_path, file_path)
-        local_dir = os.path.dirname(local_path)
-
-        # Create directory if it doesn't exist
-        if not os.path.exists(local_dir):
-            os.makedirs(local_dir)
-
-        # Download remote file
-        response = await studio_client.download_file_async(remote_file.id)
-        remote_content = response.read().decode("utf-8")
-        remote_hash = compute_normalized_hash(remote_content)
-
-        if os.path.exists(local_path):
-            # Read and hash local file
-            with open(local_path, "r", encoding="utf-8") as f:
-                local_content = f.read()
-                local_hash = compute_normalized_hash(local_content)
-
-            # Compare hashes
-            if local_hash != remote_hash:
-                styled_path = click.style(str(file_path), fg="cyan")
-                console.warning(f"File {styled_path}" + " differs from remote version.")
-                response = click.prompt("Do you want to overwrite it? (y/n)", type=str)
-                if response.lower() == "y":
-                    with open(local_path, "w", encoding="utf-8", newline="\n") as f:
-                        f.write(remote_content)
-                    console.success(f"Updated {click.style(str(file_path), fg='cyan')}")
-                else:
-                    console.info(f"Skipped {click.style(str(file_path), fg='cyan')}")
-            else:
-                console.info(
-                    f"File {click.style(str(file_path), fg='cyan')} is up to date"
-                )
-        else:
-            # File doesn't exist locally, create it
-            with open(local_path, "w", encoding="utf-8", newline="\n") as f:
-                f.write(remote_content)
-            console.success(f"Downloaded {click.style(str(file_path), fg='cyan')}")
-
-        processed_files.add(file_path)
-
-
 @click.command()
 @click.argument(
-    "root", type=click.Path(exists=True, file_okay=False, dir_okay=True), default="."
+    "root",
+    type=click.Path(exists=False, file_okay=False, dir_okay=True, path_type=Path),
+    default=Path("."),
 )
 @track
-def pull(root: str) -> None:
+def pull(root: Path) -> None:
     """Pull remote project files from Studio Web Project.
 
     This command pulls the remote project files from a UiPath Studio Web project.
@@ -158,42 +52,8 @@ def pull(root: str) -> None:
     if not (project_id := os.getenv(UIPATH_PROJECT_ID, False)):
         console.error("UIPATH_PROJECT_ID environment variable not found.")
 
-    studio_client = StudioClient(project_id)
-
-    with console.spinner("Pulling UiPath project files..."):
-        try:
-            structure = asyncio.run(studio_client.get_project_structure_async())
-
-            processed_files: Set[str] = set()
-
-            # Process source_code folder
-            source_code_folder = get_folder_by_name(structure, "source_code")
-            if source_code_folder:
-                asyncio.run(
-                    download_folder_files(
-                        studio_client,
-                        source_code_folder,
-                        root,
-                        processed_files,
-                    )
-                )
-            else:
-                console.warning("No source_code folder found in remote project")
-
-            # Process evals folder
-            evals_folder = get_folder_by_name(structure, "evals")
-            if evals_folder:
-                evals_path = os.path.join(root, "evals")
-                asyncio.run(
-                    download_folder_files(
-                        studio_client,
-                        evals_folder,
-                        evals_path,
-                        processed_files,
-                    )
-                )
-            else:
-                console.warning("No evals folder found in remote project")
-
-        except Exception as e:
-            console.error(f"Failed to pull UiPath project: {str(e)}")
+    default_download_configuration = {
+        "source_code": root,
+        "evals": root / "evals",
+    }
+    asyncio.run(pull_project(project_id, default_download_configuration))

{uipath-2.1.85.dist-info → uipath-2.1.86.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath
-Version: 2.1.85
+Version: 2.1.86
 Summary: Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools.
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-python

{uipath-2.1.85.dist-info → uipath-2.1.86.dist-info}/RECORD

@@ -15,7 +15,7 @@ uipath/_cli/cli_invoke.py,sha256=m-te-EjhDpk_fhFDkt-yQFzmjEHGo5lQDGEQWxSXisQ,439
 uipath/_cli/cli_new.py,sha256=9378NYUBc9j-qKVXV7oja-jahfJhXBg8zKVyaon7ctY,2102
 uipath/_cli/cli_pack.py,sha256=NmwZTfwZ2fURiHyiX1BM0juAtBOjPB1Jmcpu-rD7p-4,11025
 uipath/_cli/cli_publish.py,sha256=DgyfcZjvfV05Ldy0Pk5y_Le_nT9JduEE_x-VpIc_Kq0,6471
-uipath/_cli/cli_pull.py,sha256=PZ2hkfsfN-ElNa3FHjNetTux8XH03tDY5kWWqydQ2OY,6832
+uipath/_cli/cli_pull.py,sha256=pLUzS4wrSsTzO69wvJa6C-l9qTomySc15GOwliPPyHU,1785
 uipath/_cli/cli_push.py,sha256=-j-gDIbT8GyU2SybLQqFl5L8KI9nu3CDijVtltDgX20,3132
 uipath/_cli/cli_run.py,sha256=1FKv20EjxrrP1I5rNSnL_HzbWtOAIMjB3M--4RPA_Yo,3709
 uipath/_cli/middlewares.py,sha256=0D9a-wphyetnH9T97F08o7-1OKWF1lMweFHHAR0xiOw,4979
@@ -82,7 +82,7 @@ uipath/_cli/_utils/_folders.py,sha256=RsYrXzF0NA1sPxgBoLkLlUY3jDNLg1V-Y8j71Q8a8H
 uipath/_cli/_utils/_input_args.py,sha256=3LGNqVpJItvof75VGm-ZNTUMUH9-c7-YgleM5b2YgRg,5088
 uipath/_cli/_utils/_parse_ast.py,sha256=8Iohz58s6bYQ7rgWtOTjrEInLJ-ETikmOMZzZdIY2Co,20072
 uipath/_cli/_utils/_processes.py,sha256=q7DfEKHISDWf3pngci5za_z0Pbnf_shWiYEcTOTCiyk,1855
-uipath/_cli/_utils/_project_files.py,sha256=62VwZrroeKjlnqMqYSy3Aex11n9qYb7J8n3a8IVdo3I,15156
+uipath/_cli/_utils/_project_files.py,sha256=1DQ0dY1oUyCP_y7i1PbqJv_JcDQmHzzsJy1bKcr3xYk,19671
 uipath/_cli/_utils/_studio_project.py,sha256=8WYwi_CiTPRqo8KV2bsvj0H_KBFxTEN0Q2cXoZb-NnM,17030
 uipath/_cli/_utils/_tracing.py,sha256=2igb03j3EHjF_A406UhtCKkPfudVfFPjUq5tXUEG4oo,1541
 uipath/_cli/_utils/_uv_helpers.py,sha256=6SvoLnZPoKIxW0sjMvD1-ENV_HOXDYzH34GjBqwT138,3450
@@ -173,8 +173,8 @@ uipath/tracing/_utils.py,sha256=X-LFsyIxDeNOGuHPvkb6T5o9Y8ElYhr_rP3CEBJSu4s,1383
 uipath/utils/__init__.py,sha256=VD-KXFpF_oWexFg6zyiWMkxl2HM4hYJMIUDZ1UEtGx0,105
 uipath/utils/_endpoints_manager.py,sha256=iRTl5Q0XAm_YgcnMcJOXtj-8052sr6jpWuPNz6CgT0Q,8408
 uipath/utils/dynamic_schema.py,sha256=w0u_54MoeIAB-mf3GmwX1A_X8_HDrRy6p998PvX9evY,3839
-uipath-2.1.85.dist-info/METADATA,sha256=jcwWDDU5E1A7FmIZgfvvmg8tWpkZ3KZuu91Ou0aWrwM,6593
-uipath-2.1.85.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath-2.1.85.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
-uipath-2.1.85.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
-uipath-2.1.85.dist-info/RECORD,,
+uipath-2.1.86.dist-info/METADATA,sha256=G_FdE5vnRC3-2kjfKVvUdE_W8-SiIDzcsNSj2bTowXs,6593
+uipath-2.1.86.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath-2.1.86.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
+uipath-2.1.86.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
+uipath-2.1.86.dist-info/RECORD,,