pltr-cli 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

pltr/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.0"
+__version__ = "0.5.0"

pltr/cli.py

@@ -11,6 +11,10 @@ from pltr.commands import (
     verify,
     dataset,
     folder,
+    project,
+    resource,
+    resource_role,
+    space,
     ontology,
     orchestration,
     sql,
@@ -19,6 +23,7 @@ from pltr.commands import (
     completion,
     alias,
     mediasets,
+    connectivity,
 )
 
 app = typer.Typer(
@@ -32,6 +37,12 @@ app.add_typer(configure.app, name="configure", help="Manage authentication profi
 app.add_typer(verify.app, name="verify", help="Verify authentication")
 app.add_typer(dataset.app, name="dataset", help="Manage datasets")
 app.add_typer(folder.app, name="folder", help="Manage folders")
+app.add_typer(project.app, name="project", help="Manage projects")
+app.add_typer(resource.app, name="resource", help="Manage resources")
+app.add_typer(
+    resource_role.app, name="resource-role", help="Manage resource permissions"
+)
+app.add_typer(space.app, name="space", help="Manage spaces")
 app.add_typer(ontology.app, name="ontology", help="Ontology operations")
 app.add_typer(
     orchestration.app, name="orchestration", help="Manage builds, jobs, and schedules"
@@ -40,6 +51,9 @@ app.add_typer(sql.app, name="sql", help="Execute SQL queries")
 app.add_typer(
     mediasets.app, name="media-sets", help="Manage media sets and media content"
 )
+app.add_typer(
+    connectivity.app, name="connectivity", help="Manage connections and data imports"
+)
 app.add_typer(
     admin.app,
     name="admin",

pltr/commands/connectivity.py

@@ -0,0 +1,432 @@
+"""
+Connectivity management commands for Foundry connections and imports.
+"""
+
+import typer
+import json
+from typing import Optional
+from rich.console import Console
+
+from ..services.connectivity import ConnectivityService
+from ..utils.formatting import OutputFormatter
+from ..utils.progress import SpinnerProgressTracker
+from ..auth.base import ProfileNotFoundError, MissingCredentialsError
+from ..utils.completion import (
+    complete_rid,
+    complete_profile,
+    complete_output_format,
+    cache_rid,
+)
+
+app = typer.Typer()
+connection_app = typer.Typer()
+import_app = typer.Typer()
+console = Console()
+formatter = OutputFormatter(console)
+
+# Add sub-apps
+app.add_typer(connection_app, name="connection", help="Manage connections")
+app.add_typer(import_app, name="import", help="Manage data imports")
+
+
+@connection_app.command("list")
+def list_connections(
+    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"
+    ),
+):
+    """List available connections."""
+    try:
+        with SpinnerProgressTracker().track_spinner("Fetching connections..."):
+            service = ConnectivityService(profile=profile)
+            connections = service.list_connections()
+
+        if not connections:
+            console.print("[yellow]No connections found[/yellow]")
+            return
+
+        formatter.format_output(connections, format, output)
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        console.print(f"[red]Authentication error: {e}[/red]")
+        raise typer.Exit(1)
+    except Exception as e:
+        console.print(f"[red]Error listing connections: {e}[/red]")
+        raise typer.Exit(1)
+
+
+@connection_app.command("get")
+def get_connection(
+    connection_rid: str = typer.Argument(
+        ..., help="Connection Resource Identifier", autocompletion=complete_rid
+    ),
+    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"
+    ),
+):
+    """Get detailed information about a specific connection."""
+    try:
+        cache_rid(connection_rid)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Fetching connection {connection_rid}..."
+        ):
+            service = ConnectivityService(profile=profile)
+            connection = service.get_connection(connection_rid)
+
+        formatter.format_output([connection], format, output)
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        console.print(f"[red]Authentication error: {e}[/red]")
+        raise typer.Exit(1)
+    except Exception as e:
+        console.print(f"[red]Error getting connection: {e}[/red]")
+        raise typer.Exit(1)
+
+
+@import_app.command("file")
+def import_file(
+    connection_rid: str = typer.Argument(
+        ..., help="Connection Resource Identifier", autocompletion=complete_rid
+    ),
+    source_path: str = typer.Argument(..., help="Source file path in the connection"),
+    target_dataset_rid: str = typer.Argument(
+        ..., help="Target dataset RID", autocompletion=complete_rid
+    ),
+    profile: Optional[str] = typer.Option(
+        None, "--profile", "-p", help="Profile name", autocompletion=complete_profile
+    ),
+    config: Optional[str] = typer.Option(
+        None, "--config", "-c", help="Import configuration in JSON format"
+    ),
+    execute: bool = typer.Option(
+        False, "--execute", help="Execute the import immediately after creation"
+    ),
+    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"
+    ),
+):
+    """Create and optionally execute a file import via connection."""
+    try:
+        cache_rid(connection_rid)
+        cache_rid(target_dataset_rid)
+
+        # Parse import configuration if provided
+        import_config = None
+        if config:
+            try:
+                import_config = json.loads(config)
+            except json.JSONDecodeError as e:
+                console.print(f"[red]Invalid JSON configuration: {e}[/red]")
+                raise typer.Exit(1)
+
+        service = ConnectivityService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner("Creating file import..."):
+            file_import = service.create_file_import(
+                connection_rid=connection_rid,
+                source_path=source_path,
+                target_dataset_rid=target_dataset_rid,
+                import_config=import_config,
+            )
+
+        result_data = [file_import]
+
+        # Execute import if requested
+        if execute:
+            import_rid = file_import.get("rid")
+            if import_rid:
+                with SpinnerProgressTracker().track_spinner("Executing file import..."):
+                    execution_result = service.execute_file_import(import_rid)
+                result_data.append({"execution": execution_result})
+                console.print(f"[green]File import executed: {import_rid}[/green]")
+            else:
+                console.print(
+                    "[yellow]Warning: Could not execute - missing import RID[/yellow]"
+                )
+
+        formatter.format_output(result_data, format, output)
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        console.print(f"[red]Authentication error: {e}[/red]")
+        raise typer.Exit(1)
+    except Exception as e:
+        console.print(f"[red]Error creating file import: {e}[/red]")
+        raise typer.Exit(1)
+
+
+@import_app.command("table")
+def import_table(
+    connection_rid: str = typer.Argument(
+        ..., help="Connection Resource Identifier", autocompletion=complete_rid
+    ),
+    source_table: str = typer.Argument(..., help="Source table name in the connection"),
+    target_dataset_rid: str = typer.Argument(
+        ..., help="Target dataset RID", autocompletion=complete_rid
+    ),
+    profile: Optional[str] = typer.Option(
+        None, "--profile", "-p", help="Profile name", autocompletion=complete_profile
+    ),
+    config: Optional[str] = typer.Option(
+        None, "--config", "-c", help="Import configuration in JSON format"
+    ),
+    execute: bool = typer.Option(
+        False, "--execute", help="Execute the import immediately after creation"
+    ),
+    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"
+    ),
+):
+    """Create and optionally execute a table import via connection."""
+    try:
+        cache_rid(connection_rid)
+        cache_rid(target_dataset_rid)
+
+        # Parse import configuration if provided
+        import_config = None
+        if config:
+            try:
+                import_config = json.loads(config)
+            except json.JSONDecodeError as e:
+                console.print(f"[red]Invalid JSON configuration: {e}[/red]")
+                raise typer.Exit(1)
+
+        service = ConnectivityService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner("Creating table import..."):
+            table_import = service.create_table_import(
+                connection_rid=connection_rid,
+                source_table=source_table,
+                target_dataset_rid=target_dataset_rid,
+                import_config=import_config,
+            )
+
+        result_data = [table_import]
+
+        # Execute import if requested
+        if execute:
+            import_rid = table_import.get("rid")
+            if import_rid:
+                with SpinnerProgressTracker().track_spinner(
+                    "Executing table import..."
+                ):
+                    execution_result = service.execute_table_import(import_rid)
+                result_data.append({"execution": execution_result})
+                console.print(f"[green]Table import executed: {import_rid}[/green]")
+            else:
+                console.print(
+                    "[yellow]Warning: Could not execute - missing import RID[/yellow]"
+                )
+
+        formatter.format_output(result_data, format, output)
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        console.print(f"[red]Authentication error: {e}[/red]")
+        raise typer.Exit(1)
+    except Exception as e:
+        console.print(f"[red]Error creating table import: {e}[/red]")
+        raise typer.Exit(1)
+
+
+@import_app.command("list-file")
+def list_file_imports(
+    connection_rid: Optional[str] = typer.Option(
+        None,
+        "--connection",
+        "-c",
+        help="Filter by connection RID",
+        autocompletion=complete_rid,
+    ),
+    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"
+    ),
+):
+    """List file imports, optionally filtered by connection."""
+    try:
+        if connection_rid:
+            cache_rid(connection_rid)
+
+        with SpinnerProgressTracker().track_spinner("Fetching file imports..."):
+            service = ConnectivityService(profile=profile)
+            imports = service.list_file_imports(connection_rid=connection_rid)
+
+        if not imports:
+            console.print("[yellow]No file imports found[/yellow]")
+            return
+
+        formatter.format_output(imports, format, output)
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        console.print(f"[red]Authentication error: {e}[/red]")
+        raise typer.Exit(1)
+    except Exception as e:
+        console.print(f"[red]Error listing file imports: {e}[/red]")
+        raise typer.Exit(1)
+
+
+@import_app.command("list-table")
+def list_table_imports(
+    connection_rid: Optional[str] = typer.Option(
+        None,
+        "--connection",
+        "-c",
+        help="Filter by connection RID",
+        autocompletion=complete_rid,
+    ),
+    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"
+    ),
+):
+    """List table imports, optionally filtered by connection."""
+    try:
+        if connection_rid:
+            cache_rid(connection_rid)
+
+        with SpinnerProgressTracker().track_spinner("Fetching table imports..."):
+            service = ConnectivityService(profile=profile)
+            imports = service.list_table_imports(connection_rid=connection_rid)
+
+        if not imports:
+            console.print("[yellow]No table imports found[/yellow]")
+            return
+
+        formatter.format_output(imports, format, output)
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        console.print(f"[red]Authentication error: {e}[/red]")
+        raise typer.Exit(1)
+    except Exception as e:
+        console.print(f"[red]Error listing table imports: {e}[/red]")
+        raise typer.Exit(1)
+
+
+@import_app.command("get-file")
+def get_file_import(
+    import_rid: str = typer.Argument(
+        ..., help="File import Resource Identifier", autocompletion=complete_rid
+    ),
+    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"
+    ),
+):
+    """Get detailed information about a specific file import."""
+    try:
+        cache_rid(import_rid)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Fetching file import {import_rid}..."
+        ):
+            service = ConnectivityService(profile=profile)
+            file_import = service.get_file_import(import_rid)
+
+        formatter.format_output([file_import], format, output)
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        console.print(f"[red]Authentication error: {e}[/red]")
+        raise typer.Exit(1)
+    except Exception as e:
+        console.print(f"[red]Error getting file import: {e}[/red]")
+        raise typer.Exit(1)
+
+
+@import_app.command("get-table")
+def get_table_import(
+    import_rid: str = typer.Argument(
+        ..., help="Table import Resource Identifier", autocompletion=complete_rid
+    ),
+    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"
+    ),
+):
+    """Get detailed information about a specific table import."""
+    try:
+        cache_rid(import_rid)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Fetching table import {import_rid}..."
+        ):
+            service = ConnectivityService(profile=profile)
+            table_import = service.get_table_import(import_rid)
+
+        formatter.format_output([table_import], format, output)
+
+    except (ProfileNotFoundError, MissingCredentialsError) as e:
+        console.print(f"[red]Authentication error: {e}[/red]")
+        raise typer.Exit(1)
+    except Exception as e:
+        console.print(f"[red]Error getting table import: {e}[/red]")
+        raise typer.Exit(1)