pltr-cli 0.11.0__py3-none-any.whl → 0.13.0__py3-none-any.whl

pltr/commands/cp.py

@@ -0,0 +1,103 @@
+"""
+Copy resources between Compass folders.
+"""
+
+from __future__ import annotations
+
+from typing import Optional
+
+import typer
+from rich.console import Console
+
+from ..auth.base import MissingCredentialsError, ProfileNotFoundError
+from ..services.copy import CopyService
+from ..utils.completion import complete_profile, complete_rid
+from ..utils.formatting import OutputFormatter
+
+
+def cp_command(
+    source_rid: str = typer.Argument(
+        ...,
+        help="Resource Identifier to copy (dataset or folder)",
+        autocompletion=complete_rid,
+    ),
+    target_folder_rid: str = typer.Argument(
+        ...,
+        help="Destination Compass folder RID",
+        autocompletion=complete_rid,
+    ),
+    profile: Optional[str] = typer.Option(
+        None,
+        "--profile",
+        help="Authentication profile",
+        autocompletion=complete_profile,
+    ),
+    recursive: bool = typer.Option(
+        False,
+        "--recursive",
+        "-r",
+        help="Copy folder contents recursively (required when SOURCE_RID is a folder)",
+    ),
+    branch: str = typer.Option(
+        "master",
+        "--branch",
+        "-b",
+        help="Dataset branch to copy file-backed datasets from",
+    ),
+    name_suffix: str = typer.Option(
+        "-copy",
+        "--name-suffix",
+        help="Suffix appended to cloned folder/dataset names",
+    ),
+    copy_schema: bool = typer.Option(
+        True,
+        "--schema/--no-schema",
+        help="Copy dataset schemas when available",
+    ),
+    dry_run: bool = typer.Option(
+        False,
+        "--dry-run",
+        help="Log actions without creating any resources",
+    ),
+    fail_fast: bool = typer.Option(
+        False,
+        "--fail-fast",
+        help="Stop immediately on first error when copying folders recursively",
+    ),
+    debug: bool = typer.Option(
+        False,
+        "--debug",
+        help="Print stack traces when errors occur",
+    ),
+):
+    """Copy a resource identified by RID into another Compass folder."""
+    console = Console()
+    formatter = OutputFormatter(console)
+
+    try:
+        service = CopyService(
+            profile=profile,
+            branch=branch,
+            name_suffix=name_suffix,
+            copy_schema=copy_schema,
+            dry_run=dry_run,
+            fail_fast=fail_fast,
+            debug=debug,
+            console=console,
+        )
+        summary = service.copy_resource(
+            source_rid, target_folder_rid, recursive=recursive
+        )
+        formatter.print_success("Copy operation completed")
+        formatter.print_info(
+            "Folders copied: {folders_copied} | Datasets copied: {datasets_copied} | "
+            "Skipped: {skipped} | Errors: {errors}".format(**summary)
+        )
+    except (ProfileNotFoundError, MissingCredentialsError) as exc:
+        formatter.print_error(f"Authentication error: {exc}")
+        raise typer.Exit(1)
+    except Exception as exc:
+        if debug:
+            raise
+        formatter.print_error(f"Failed to copy resource: {exc}")
+        raise typer.Exit(1)

pltr/commands/dataset.py

@@ -8,6 +8,7 @@ from rich.console import Console
 
 from ..services.dataset import DatasetService
 from ..utils.formatting import OutputFormatter
+from ..utils.pagination import PaginationConfig
 from ..utils.progress import SpinnerProgressTracker
 from ..auth.base import ProfileNotFoundError, MissingCredentialsError
 from ..utils.completion import (
@@ -73,6 +74,55 @@ def get_dataset(
         raise typer.Exit(1)
 
 
+@app.command("preview")
+def preview_dataset(
+    dataset_rid: str = typer.Argument(
+        ..., help="Dataset Resource Identifier", autocompletion=complete_rid
+    ),
+    limit: int = typer.Option(
+        10, "--limit", "-n", help="Number of rows to display", min=1
+    ),
+    profile: Optional[str] = typer.Option(
+        None, "--profile", "-p", help="Profile name", autocompletion=complete_profile
+    ),
+    format: str = typer.Option(
+        "table",
+        "--format",
+        "-f",
+        help="Output format (table, json, csv)",
+        autocompletion=complete_output_format,
+    ),
+    output: Optional[str] = typer.Option(
+        None, "--output", "-o", help="Output file path"
+    ),
+):
+    """Preview dataset contents."""
+    try:
+        cache_rid(dataset_rid)
+        service = DatasetService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Fetching preview of {dataset_rid} (limit: {limit})..."
+        ):
+            data = service.preview_data(dataset_rid, limit=limit)
+
+        if not data:
+            formatter.print_warning("Dataset is empty or has no readable data")
+            return
+
+        formatter.format_output(data, format, output)
+
+        if output:
+            formatter.print_success(f"Preview saved to {output}")
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        formatter.print_error(f"Authentication error: {e}")
+        raise typer.Exit(1)
+    except Exception as e:
+        formatter.print_error(f"Failed to preview dataset: {e}")
+        raise typer.Exit(1)
+
+
 # Schema commands
 @schema_app.command("get")
 def get_schema(
@@ -566,21 +616,71 @@ def list_files(
     output: Optional[str] = typer.Option(
         None, "--output", "-o", help="Output file path"
     ),
+    page_size: Optional[int] = typer.Option(
+        None, "--page-size", help="Number of files per page (default: from settings)"
+    ),
+    max_pages: Optional[int] = typer.Option(
+        1, "--max-pages", help="Maximum number of pages to fetch (default: 1)"
+    ),
+    page_token: Optional[str] = typer.Option(
+        None, "--page-token", help="Page token to resume from previous response"
+    ),
+    all: bool = typer.Option(
+        False, "--all", help="Fetch all available pages (overrides --max-pages)"
+    ),
 ):
-    """List files in a dataset."""
+    """
+    List files in a dataset with pagination support.
+
+    By default, fetches only the first page of results. Use --all to fetch all files,
+    or --max-pages to control how many pages to fetch. Critical for large datasets.
+
+    Examples:
+        # List first page of files (default)
+        pltr dataset files list DATASET_RID
+
+        # List all files
+        pltr dataset files list DATASET_RID --all
+
+        # List first 3 pages
+        pltr dataset files list DATASET_RID --max-pages 3
+    """
     try:
         cache_rid(dataset_rid)
         service = DatasetService(profile=profile)
 
+        # Create pagination config
+        config = PaginationConfig(
+            page_size=page_size,
+            max_pages=max_pages,
+            page_token=page_token,
+            fetch_all=all,
+        )
+
         with SpinnerProgressTracker().track_spinner(
             f"Fetching files from {dataset_rid} (branch: {branch})..."
         ):
-            files = service.list_files(dataset_rid, branch)
-
-        formatter.format_files(files, format, output)
+            result = service.list_files_paginated(dataset_rid, branch, config)
 
+        # Format and display paginated results
         if output:
+            formatter.format_paginated_output(
+                result,
+                format,
+                output,
+                formatter_fn=lambda data, fmt, out: formatter.format_files(
+                    data, fmt, out
+                ),
+            )
             formatter.print_success(f"Files information saved to {output}")
+        else:
+            formatter.format_paginated_output(
+                result,
+                format,
+                formatter_fn=lambda data, fmt, out: formatter.format_files(
+                    data, fmt, out
+                ),
+            )
 
     except (ProfileNotFoundError, MissingCredentialsError) as e:
         formatter.print_error(f"Authentication error: {e}")