pltr-cli 0.10.0__py3-none-any.whl → 0.12.0__py3-none-any.whl

pltr/commands/orchestration.py

@@ -9,6 +9,7 @@ from rich.console import Console
 
 from ..services.orchestration import OrchestrationService
 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 (
@@ -226,7 +227,7 @@ def get_build_jobs(
 
         if response.get("next_page_token"):
             formatter.print_info(
-                "More results available. Use pagination token to fetch next page."
+                f"More results available. Use --page-token {response['next_page_token']} to fetch next page."
             )
 
     except (ProfileNotFoundError, MissingCredentialsError) as e:
@@ -241,7 +242,16 @@ def get_build_jobs(
 def search_builds(
     profile: Optional[str] = typer.Option(None, "--profile", "-p", help="Profile name"),
     page_size: Optional[int] = typer.Option(
-        None, "--page-size", help="Number of results per page"
+        None, "--page-size", help="Number of builds 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)"
     ),
     format: str = typer.Option(
         "table", "--format", "-f", help="Output format (table, json, csv)"
@@ -250,12 +260,102 @@ def search_builds(
         None, "--output", "-o", help="Output file path"
     ),
 ):
-    """Search for builds."""
+    """
+    Search for builds with pagination support.
+
+    By default, fetches only the first page of results. Use --all to fetch all builds,
+    or --max-pages to control how many pages to fetch.
+
+    Examples:
+        # Search first page of builds (default)
+        pltr orchestration builds search
+
+        # Search all builds
+        pltr orchestration builds search --all
+
+        # Search first 3 pages
+        pltr orchestration builds search --max-pages 3
+
+        # Resume from a specific page
+        pltr orchestration builds search --page-token abc123
+    """
     try:
         service = OrchestrationService(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("Searching builds..."):
-            response = service.search_builds(page_size=page_size)
+            result = service.search_builds_paginated(config)
+
+        if not result.data:
+            formatter.print_warning("No builds found")
+            return
+
+        # Format and display paginated results
+        if output:
+            formatter.format_paginated_output(
+                result,
+                format,
+                output,
+                formatter_fn=lambda data, fmt, out: formatter.format_builds_list(
+                    data, fmt, out
+                ),
+            )
+            formatter.print_success(f"Builds list saved to {output}")
+        else:
+            formatter.format_paginated_output(
+                result,
+                format,
+                formatter_fn=lambda data, fmt, out: formatter.format_builds_list(
+                    data, fmt, out
+                ),
+            )
+
+    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 search builds: {e}")
+        raise typer.Exit(1)
+
+
+@builds_app.command("get-batch")
+def get_builds_batch(
+    build_rids: str = typer.Argument(
+        ..., help="Comma-separated list of Build RIDs (max 100)"
+    ),
+    profile: Optional[str] = typer.Option(None, "--profile", "-p", help="Profile name"),
+    format: str = typer.Option(
+        "table", "--format", "-f", help="Output format (table, json, csv)"
+    ),
+    output: Optional[str] = typer.Option(
+        None, "--output", "-o", help="Output file path"
+    ),
+):
+    """Get multiple builds in batch (max 100)."""
+    try:
+        rids_list = [rid.strip() for rid in build_rids.split(",") if rid.strip()]
+
+        if not rids_list:
+            formatter.print_error("No valid RIDs provided")
+            raise typer.Exit(1)
+
+        if len(rids_list) > 100:
+            formatter.print_error("Maximum batch size is 100 builds")
+            raise typer.Exit(1)
+
+        service = OrchestrationService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Fetching {len(rids_list)} builds..."
+        ):
+            response = service.get_builds_batch(rids_list)
 
         builds = response.get("builds", [])
 
@@ -266,18 +366,13 @@ def search_builds(
         formatter.format_builds_list(builds, format, output)
 
         if output:
-            formatter.print_success(f"Builds list saved to {output}")
-
-        if response.get("next_page_token"):
-            formatter.print_info(
-                "More results available. Use pagination token to fetch next page."
-            )
+            formatter.print_success(f"Builds information 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 search builds: {e}")
+        formatter.print_error(f"Failed to get builds batch: {e}")
         raise typer.Exit(1)
 
 
@@ -655,6 +750,67 @@ def replace_schedule(
         raise typer.Exit(1)
 
 
+@schedules_app.command("runs")
+def get_schedule_runs(
+    schedule_rid: str = typer.Argument(
+        ..., help="Schedule Resource Identifier", autocompletion=complete_rid
+    ),
+    profile: Optional[str] = typer.Option(
+        None, "--profile", "-p", help="Profile name", autocompletion=complete_profile
+    ),
+    page_size: Optional[int] = typer.Option(
+        None, "--page-size", help="Number of results per page"
+    ),
+    page_token: Optional[str] = typer.Option(
+        None, "--page-token", help="Pagination token for next page"
+    ),
+    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"
+    ),
+):
+    """List recent execution runs for a schedule."""
+    try:
+        cache_rid(schedule_rid)
+        service = OrchestrationService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Fetching runs for schedule {schedule_rid}..."
+        ):
+            response = service.get_schedule_runs(
+                schedule_rid, page_size=page_size, page_token=page_token
+            )
+
+        runs = response.get("runs", [])
+
+        if not runs:
+            formatter.print_warning("No runs found for this schedule")
+            return
+
+        formatter.format_schedule_runs_list(runs, format, output)
+
+        if output:
+            formatter.print_success(f"Runs list saved to {output}")
+
+        if response.get("next_page_token"):
+            formatter.print_info(
+                f"More results available. Use --page-token {response['next_page_token']} to fetch next page."
+            )
+
+    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 get schedule runs: {e}")
+        raise typer.Exit(1)
+
+
 # ============================================================================
 # Main app setup
 # ============================================================================

pltr/commands/project.py

@@ -366,6 +366,229 @@ def get_projects_batch(
         raise typer.Exit(1)
 
 
+# ==================== Organization Operations ====================
+
+
+@app.command("add-orgs")
+def add_organizations(
+    project_rid: str = typer.Argument(
+        ..., help="Project Resource Identifier", autocompletion=complete_rid
+    ),
+    organization_rids: List[str] = typer.Option(
+        ...,
+        "--org",
+        "-o",
+        help="Organization RID(s) to add (can specify multiple)",
+    ),
+    profile: Optional[str] = typer.Option(
+        None, "--profile", help="Profile name", autocompletion=complete_profile
+    ),
+):
+    """Add organizations to a project."""
+    try:
+        service = ProjectService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Adding {len(organization_rids)} organization(s) to project {project_rid}..."
+        ):
+            service.add_organizations(project_rid, organization_rids)
+
+        formatter.print_success(
+            f"Successfully added {len(organization_rids)} organization(s) to project {project_rid}"
+        )
+
+    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 add organizations: {e}")
+        raise typer.Exit(1)
+
+
+@app.command("remove-orgs")
+def remove_organizations(
+    project_rid: str = typer.Argument(
+        ..., help="Project Resource Identifier", autocompletion=complete_rid
+    ),
+    organization_rids: List[str] = typer.Option(
+        ...,
+        "--org",
+        "-o",
+        help="Organization RID(s) to remove (can specify multiple)",
+    ),
+    profile: Optional[str] = typer.Option(
+        None, "--profile", help="Profile name", autocompletion=complete_profile
+    ),
+):
+    """Remove organizations from a project."""
+    try:
+        service = ProjectService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Removing {len(organization_rids)} organization(s) from project {project_rid}..."
+        ):
+            service.remove_organizations(project_rid, organization_rids)
+
+        formatter.print_success(
+            f"Successfully removed {len(organization_rids)} organization(s) from project {project_rid}"
+        )
+
+    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 remove organizations: {e}")
+        raise typer.Exit(1)
+
+
+@app.command("list-orgs")
+def list_organizations(
+    project_rid: str = typer.Argument(
+        ..., help="Project Resource Identifier", autocompletion=complete_rid
+    ),
+    profile: Optional[str] = typer.Option(
+        None, "--profile", 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", help="Output file path"),
+    page_size: Optional[int] = typer.Option(
+        None, "--page-size", help="Number of items per page"
+    ),
+):
+    """List organizations directly applied to a project."""
+    try:
+        service = ProjectService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Fetching organizations for project {project_rid}..."
+        ):
+            organizations = service.list_organizations(project_rid, page_size=page_size)
+
+        if not organizations:
+            formatter.print_info("No organizations found on this project.")
+            return
+
+        # Format output
+        if format == "json":
+            if output:
+                formatter.save_to_file(organizations, output, "json")
+            else:
+                formatter.format_list(organizations)
+        elif format == "csv":
+            if output:
+                formatter.save_to_file(organizations, output, "csv")
+            else:
+                formatter.format_list(organizations)
+        else:
+            _format_organizations_table(organizations)
+
+        if output:
+            formatter.print_success(f"Organizations list 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 list organizations: {e}")
+        raise typer.Exit(1)
+
+
+# ==================== Template Operations ====================
+
+
+@app.command("create-from-template")
+def create_from_template(
+    template_rid: str = typer.Option(
+        ...,
+        "--template-rid",
+        "-t",
+        help="Template Resource Identifier",
+        autocompletion=complete_rid,
+    ),
+    variable: List[str] = typer.Option(
+        ...,
+        "--var",
+        "-v",
+        help="Variable values in format 'name=value' (can specify multiple)",
+    ),
+    description: Optional[str] = typer.Option(
+        None, "--description", "-d", help="Project description"
+    ),
+    organization_rids: Optional[List[str]] = typer.Option(
+        None,
+        "--org",
+        "-o",
+        help="Organization RIDs (can be specified multiple times)",
+    ),
+    profile: Optional[str] = typer.Option(
+        None, "--profile", help="Profile name", autocompletion=complete_profile
+    ),
+    format: str = typer.Option(
+        "table",
+        "--format",
+        "-f",
+        help="Output format (table, json, csv)",
+        autocompletion=complete_output_format,
+    ),
+):
+    """Create a project from a template."""
+    try:
+        # Parse variable values
+        variable_values = {}
+        for var in variable:
+            if "=" not in var:
+                formatter.print_error(
+                    f"Invalid variable format: '{var}'. Use 'name=value' format."
+                )
+                raise typer.Exit(1)
+            name, value = var.split("=", 1)
+            name = name.strip()
+            if not name:
+                formatter.print_error(f"Variable name cannot be empty: '{var}'")
+                raise typer.Exit(1)
+            variable_values[name] = value
+
+        service = ProjectService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Creating project from template {template_rid}..."
+        ):
+            project = service.create_project_from_template(
+                template_rid=template_rid,
+                variable_values=variable_values,
+                organization_rids=organization_rids,
+                project_description=description,
+            )
+
+        # Cache the RID for future completions
+        if project.get("rid"):
+            cache_rid(project["rid"])
+
+        formatter.print_success("Successfully created project from template")
+        formatter.print_info(f"Project RID: {project.get('rid', 'unknown')}")
+
+        # Format output
+        if format == "json":
+            formatter.format_dict(project)
+        elif format == "csv":
+            formatter.format_list([project])
+        else:
+            _format_project_table(project)
+
+    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 create project from template: {e}")
+        raise typer.Exit(1)
+
+
 def _format_project_table(project: dict):
     """Format project information as a table."""
     table = Table(
@@ -410,6 +633,24 @@ def _format_projects_table(projects: List[dict]):
     console.print(f"\nTotal: {len(projects)} projects")
 
 
+def _format_organizations_table(organizations: List[dict]):
+    """Format organizations as a table."""
+    table = Table(title="Organizations", show_header=True, header_style="bold cyan")
+    table.add_column("Organization RID")
+    table.add_column("Display Name")
+    table.add_column("Description")
+
+    for org in organizations:
+        table.add_row(
+            org.get("organization_rid", "N/A"),
+            org.get("display_name", "N/A"),
+            org.get("description", "N/A"),
+        )
+
+    console.print(table)
+    console.print(f"\nTotal: {len(organizations)} organizations")
+
+
 @app.callback()
 def main():
     """
@@ -436,5 +677,13 @@ def main():
 
         # Delete project
         pltr project delete ri.compass.main.project.abc456
+
+        # Organization operations
+        pltr project add-orgs ri.compass.main.project.abc456 -o org-rid-1 -o org-rid-2
+        pltr project remove-orgs ri.compass.main.project.abc456 -o org-rid-1
+        pltr project list-orgs ri.compass.main.project.abc456
+
+        # Create from template
+        pltr project create-from-template -t template-rid -v "name=MyProject" -v "desc=Description"
     """
     pass