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

pltr/commands/space.py

@@ -26,13 +26,25 @@ formatter = OutputFormatter(console)
 @app.command("create")
 def create_space(
     name: str = typer.Argument(..., help="Space display name"),
-    organization_rid: str = typer.Option(
+    enrollment_rid: str = typer.Option(
         ...,
-        "--organization-rid",
-        "-org",
-        help="Organization Resource Identifier",
+        "--enrollment-rid",
+        "-e",
+        help="Enrollment Resource Identifier",
         autocompletion=complete_rid,
     ),
+    organizations: List[str] = typer.Option(
+        ...,
+        "--organization",
+        "-org",
+        help="Organization RID(s) (can specify multiple)",
+    ),
+    deletion_policy_organizations: List[str] = typer.Option(
+        ...,
+        "--deletion-policy-org",
+        "-dpo",
+        help="Organization RID(s) for deletion policy (can specify multiple)",
+    ),
     description: Optional[str] = typer.Option(
         None, "--description", "-d", help="Space description"
     ),
@@ -54,7 +66,9 @@ def create_space(
         with SpinnerProgressTracker().track_spinner(f"Creating space '{name}'..."):
             space = service.create_space(
                 display_name=name,
-                organization_rid=organization_rid,
+                enrollment_rid=enrollment_rid,
+                organizations=organizations,
+                deletion_policy_organizations=deletion_policy_organizations,
                 description=description,
             )
 
@@ -295,250 +309,6 @@ def delete_space(
         raise typer.Exit(1)
 
 
-@app.command("batch-get")
-def get_spaces_batch(
-    space_rids: List[str] = typer.Argument(
-        ..., help="Space Resource Identifiers (space-separated)"
-    ),
-    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", "-o", help="Output file path"
-    ),
-):
-    """Get multiple spaces in a single request (max 1000)."""
-    try:
-        service = SpaceService(profile=profile)
-
-        with SpinnerProgressTracker().track_spinner(
-            f"Fetching {len(space_rids)} spaces..."
-        ):
-            spaces = service.get_spaces_batch(space_rids)
-
-        # Cache RIDs for future completions
-        for space in spaces:
-            if space.get("rid"):
-                cache_rid(space["rid"])
-
-        # Format output
-        if format == "json":
-            if output:
-                formatter.save_to_file(spaces, output, "json")
-            else:
-                formatter.format_list(spaces)
-        elif format == "csv":
-            if output:
-                formatter.save_to_file(spaces, output, "csv")
-            else:
-                formatter.format_list(spaces)
-        else:
-            _format_spaces_table(spaces)
-
-        if output:
-            formatter.print_success(f"Spaces information saved to {output}")
-
-    except (ProfileNotFoundError, MissingCredentialsError) as e:
-        formatter.print_error(f"Authentication error: {e}")
-        raise typer.Exit(1)
-    except ValueError as e:
-        formatter.print_error(f"Invalid request: {e}")
-        raise typer.Exit(1)
-    except Exception as e:
-        formatter.print_error(f"Failed to get spaces batch: {e}")
-        raise typer.Exit(1)
-
-
-@app.command("list-members")
-def list_space_members(
-    space_rid: str = typer.Argument(
-        ..., help="Space Resource Identifier", autocompletion=complete_rid
-    ),
-    principal_type: Optional[str] = typer.Option(
-        None,
-        "--principal-type",
-        "-t",
-        help="Filter by principal type (User or Group)",
-    ),
-    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", "-o", help="Output file path"
-    ),
-    page_size: Optional[int] = typer.Option(
-        None, "--page-size", help="Number of items per page"
-    ),
-):
-    """Get all members (users/groups) of a space."""
-    try:
-        service = SpaceService(profile=profile)
-
-        filter_desc = f" ({principal_type}s only)" if principal_type else ""
-        with SpinnerProgressTracker().track_spinner(
-            f"Listing members of space {space_rid}{filter_desc}..."
-        ):
-            members = service.get_space_members(
-                space_rid=space_rid,
-                principal_type=principal_type.title() if principal_type else None,
-                page_size=page_size,
-            )
-
-        if not members:
-            formatter.print_info(f"No members found in space {space_rid}.")
-            return
-
-        # Format output
-        if format == "json":
-            if output:
-                formatter.save_to_file(members, output, "json")
-            else:
-                formatter.format_list(members)
-        elif format == "csv":
-            if output:
-                formatter.save_to_file(members, output, "csv")
-            else:
-                formatter.format_list(members)
-        else:
-            _format_space_members_table(members)
-
-        if output:
-            formatter.print_success(f"Space members 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 space members: {e}")
-        raise typer.Exit(1)
-
-
-@app.command("add-member")
-def add_space_member(
-    space_rid: str = typer.Argument(
-        ..., help="Space Resource Identifier", autocompletion=complete_rid
-    ),
-    principal_id: str = typer.Option(
-        ..., "--principal-id", "-p", help="Principal (user/group) identifier"
-    ),
-    principal_type: str = typer.Option(
-        ...,
-        "--principal-type",
-        "-t",
-        help="Principal type (User or Group)",
-    ),
-    role_name: str = typer.Option(..., "--role", "-r", help="Role name to grant"),
-    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,
-    ),
-):
-    """Add a member to a space with a specific role."""
-    try:
-        service = SpaceService(profile=profile)
-
-        with SpinnerProgressTracker().track_spinner(
-            f"Adding {principal_type} '{principal_id}' to space {space_rid} with role '{role_name}'..."
-        ):
-            member = service.add_space_member(
-                space_rid=space_rid,
-                principal_id=principal_id,
-                principal_type=principal_type.title(),
-                role_name=role_name,
-            )
-
-        formatter.print_success(
-            f"Successfully added {principal_type} '{principal_id}' to space with role '{role_name}'"
-        )
-
-        # Format output
-        if format == "json":
-            formatter.format_dict(member)
-        elif format == "csv":
-            formatter.format_list([member])
-        else:
-            _format_space_member_table(member)
-
-    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 space member: {e}")
-        raise typer.Exit(1)
-
-
-@app.command("remove-member")
-def remove_space_member(
-    space_rid: str = typer.Argument(
-        ..., help="Space Resource Identifier", autocompletion=complete_rid
-    ),
-    principal_id: str = typer.Option(
-        ..., "--principal-id", "-p", help="Principal (user/group) identifier"
-    ),
-    principal_type: str = typer.Option(
-        ...,
-        "--principal-type",
-        "-t",
-        help="Principal type (User or Group)",
-    ),
-    profile: Optional[str] = typer.Option(
-        None, "--profile", help="Profile name", autocompletion=complete_profile
-    ),
-    confirm: bool = typer.Option(False, "--confirm", help="Skip confirmation prompt"),
-):
-    """Remove a member from a space."""
-    try:
-        if not confirm:
-            confirm_remove = typer.confirm(
-                f"Are you sure you want to remove {principal_type} '{principal_id}' from space {space_rid}?"
-            )
-            if not confirm_remove:
-                formatter.print_info("Member removal cancelled.")
-                return
-
-        service = SpaceService(profile=profile)
-
-        with SpinnerProgressTracker().track_spinner(
-            f"Removing {principal_type} '{principal_id}' from space {space_rid}..."
-        ):
-            service.remove_space_member(
-                space_rid=space_rid,
-                principal_id=principal_id,
-                principal_type=principal_type.title(),
-            )
-
-        formatter.print_success(
-            f"Successfully removed {principal_type} '{principal_id}' from space"
-        )
-
-    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 space member: {e}")
-        raise typer.Exit(1)
-
-
 def _format_space_table(space: dict):
     """Format space information as a table."""
     table = Table(title="Space Information", show_header=True, header_style="bold cyan")
@@ -581,57 +351,20 @@ def _format_spaces_table(spaces: List[dict]):
     console.print(f"\nTotal: {len(spaces)} spaces")
 
 
-def _format_space_member_table(member: dict):
-    """Format space member information as a table."""
-    table = Table(
-        title="Space Member Information", show_header=True, header_style="bold cyan"
-    )
-    table.add_column("Property", style="cyan")
-    table.add_column("Value")
-
-    table.add_row("Space RID", member.get("space_rid", "N/A"))
-    table.add_row("Principal ID", member.get("principal_id", "N/A"))
-    table.add_row("Principal Type", member.get("principal_type", "N/A"))
-    table.add_row("Role Name", member.get("role_name", "N/A"))
-    table.add_row("Added By", member.get("added_by", "N/A"))
-    table.add_row("Added Time", member.get("added_time", "N/A"))
-
-    console.print(table)
-
-
-def _format_space_members_table(members: List[dict]):
-    """Format multiple space members as a table."""
-    table = Table(title="Space Members", show_header=True, header_style="bold cyan")
-    table.add_column("Principal Type")
-    table.add_column("Principal ID")
-    table.add_column("Role Name")
-    table.add_column("Added By")
-    table.add_column("Added Time")
-
-    for member in members:
-        table.add_row(
-            member.get("principal_type", "N/A"),
-            member.get("principal_id", "N/A"),
-            member.get("role_name", "N/A"),
-            member.get("added_by", "N/A"),
-            member.get("added_time", "N/A"),
-        )
-
-    console.print(table)
-    console.print(f"\nTotal: {len(members)} members")
-
-
 @app.callback()
 def main():
     """
     Space operations using foundry-platform-sdk.
 
     Manage spaces in the Foundry filesystem. Create, retrieve, update, and delete
-    spaces, and manage space membership using Resource Identifiers (RIDs).
+    spaces using Resource Identifiers (RIDs).
 
     Examples:
-        # Create a space in an organization
-        pltr space create "My Space" --organization-rid ri.compass.main.organization.xyz123
+        # Create a space with required parameters
+        pltr space create "My Space" \\
+            --enrollment-rid ri.enrollment.main.enrollment.xyz123 \\
+            --organization ri.compass.main.organization.abc456 \\
+            --deletion-policy-org ri.compass.main.organization.abc456
 
         # List all spaces
         pltr space list
@@ -645,17 +378,6 @@ def main():
         # Update space
         pltr space update ri.compass.main.space.abc456 --name "Updated Name"
 
-        # List space members
-        pltr space list-members ri.compass.main.space.abc456
-
-        # Add a user to a space
-        pltr space add-member ri.compass.main.space.abc456 \\
-            --principal-id user123 --principal-type User --role viewer
-
-        # Remove a user from a space
-        pltr space remove-member ri.compass.main.space.abc456 \\
-            --principal-id user123 --principal-type User
-
         # Delete space
         pltr space delete ri.compass.main.space.abc456
     """

pltr/commands/sql.py

@@ -42,8 +42,15 @@ def execute_query(
     fallback_branches: Optional[str] = typer.Option(
         None, "--fallback-branches", help="Comma-separated list of fallback branch IDs"
     ),
+    preview: bool = typer.Option(
+        True,
+        "--preview/--no-preview",
+        help="Enable preview mode (required for SQL API)",
+    ),
 ) -> None:
-    """Execute a SQL query and display results."""
+    """Execute a SQL query and display results.
+
+    Note: SQL queries are currently in preview and may be modified or removed at any time."""
     console = Console()
     formatter = OutputFormatter()
 
@@ -62,6 +69,7 @@ def execute_query(
                 fallback_branch_ids=fallback_branch_ids,
                 timeout=timeout,
                 format="table" if output_format in ["table", "csv"] else "json",
+                preview=preview,
             )
 
         # Extract results
@@ -92,8 +100,15 @@ def submit_query(
     fallback_branches: Optional[str] = typer.Option(
         None, "--fallback-branches", help="Comma-separated list of fallback branch IDs"
     ),
+    preview: bool = typer.Option(
+        True,
+        "--preview/--no-preview",
+        help="Enable preview mode (required for SQL API)",
+    ),
 ) -> None:
-    """Submit a SQL query without waiting for completion."""
+    """Submit a SQL query without waiting for completion.
+
+    Note: SQL queries are currently in preview and may be modified or removed at any time."""
     console = Console()
     formatter = OutputFormatter()
 
@@ -108,7 +123,7 @@ def submit_query(
 
         with SpinnerProgressTracker().track_spinner("Submitting SQL query..."):
             result = service.submit_query(
-                query=query, fallback_branch_ids=fallback_branch_ids
+                query=query, fallback_branch_ids=fallback_branch_ids, preview=preview
             )
 
         console.print("[green]Query submitted successfully[/green]")
@@ -136,6 +151,11 @@ def get_query_status(
     profile: Optional[str] = typer.Option(
         None, "--profile", help="Auth profile to use"
     ),
+    preview: bool = typer.Option(
+        True,
+        "--preview/--no-preview",
+        help="Enable preview mode (required for SQL API)",
+    ),
 ) -> None:
     """Get the status of a submitted query."""
     console = Console()
@@ -145,7 +165,7 @@ def get_query_status(
         service = SqlService(profile=profile)
 
         with SpinnerProgressTracker().track_spinner("Checking query status..."):
-            result = service.get_query_status(query_id)
+            result = service.get_query_status(query_id, preview=preview)
 
         console.print(f"Query ID: [bold]{query_id}[/bold]")
 
@@ -183,6 +203,11 @@ def get_query_results(
     output_file: Optional[Path] = typer.Option(
         None, "--output", help="Save results to file"
     ),
+    preview: bool = typer.Option(
+        True,
+        "--preview/--no-preview",
+        help="Enable preview mode (required for SQL API)",
+    ),
 ) -> None:
     """Get the results of a completed query."""
     console = Console()
@@ -195,6 +220,7 @@ def get_query_results(
             result = service.get_query_results(
                 query_id,
                 format="table" if output_format in ["table", "csv"] else "json",
+                preview=preview,
             )
 
         # Display or save results
@@ -217,6 +243,11 @@ def cancel_query(
     profile: Optional[str] = typer.Option(
         None, "--profile", help="Auth profile to use"
     ),
+    preview: bool = typer.Option(
+        True,
+        "--preview/--no-preview",
+        help="Enable preview mode (required for SQL API)",
+    ),
 ) -> None:
     """Cancel a running query."""
     console = Console()
@@ -226,7 +257,7 @@ def cancel_query(
         service = SqlService(profile=profile)
 
         with SpinnerProgressTracker().track_spinner("Canceling query..."):
-            result = service.cancel_query(query_id)
+            result = service.cancel_query(query_id, preview=preview)
 
         console.print(f"Query ID: [bold]{query_id}[/bold]")
 
@@ -259,8 +290,15 @@ def export_query_results(
     fallback_branches: Optional[str] = typer.Option(
         None, "--fallback-branches", help="Comma-separated list of fallback branch IDs"
     ),
+    preview: bool = typer.Option(
+        True,
+        "--preview/--no-preview",
+        help="Enable preview mode (required for SQL API)",
+    ),
 ) -> None:
-    """Execute a SQL query and export results to a file."""
+    """Execute a SQL query and export results to a file.
+
+    Note: SQL queries are currently in preview and may be modified or removed at any time."""
     console = Console()
     formatter = OutputFormatter()
 
@@ -289,6 +327,7 @@ def export_query_results(
                 fallback_branch_ids=fallback_branch_ids,
                 timeout=timeout,
                 format="table" if output_format in ["table", "csv"] else "json",
+                preview=preview,
             )
 
         # Save results to file
@@ -321,6 +360,11 @@ def wait_for_query(
     output_file: Optional[Path] = typer.Option(
         None, "--output", help="Save results to file when completed"
     ),
+    preview: bool = typer.Option(
+        True,
+        "--preview/--no-preview",
+        help="Enable preview mode (required for SQL API)",
+    ),
 ) -> None:
     """Wait for a query to complete and optionally display results."""
     console = Console()
@@ -330,7 +374,9 @@ def wait_for_query(
         service = SqlService(profile=profile)
 
         with SpinnerProgressTracker().track_spinner("Waiting for query to complete..."):
-            status_result = service.wait_for_completion(query_id, timeout)
+            status_result = service.wait_for_completion(
+                query_id, timeout, preview=preview
+            )
 
         console.print(f"Query ID: [bold]{query_id}[/bold]")
         console.print(
@@ -343,6 +389,7 @@ def wait_for_query(
                 result = service.get_query_results(
                     query_id,
                     format="table" if output_format in ["table", "csv"] else "json",
+                    preview=preview,
                 )
 
             if output_file: