pltr-cli 0.8.0__py3-none-any.whl → 0.9.1__py3-none-any.whl

pltr/__init__.py

@@ -1 +1 @@
-__version__ = "0.8.0"
+__version__ = "0.9.1"

pltr/commands/dataset.py

@@ -93,11 +93,17 @@ def get_schema(
         None, "--output", "-o", help="Output file path"
     ),
 ):
-    """Get the schema of a dataset."""
+    """Get the schema of a dataset (requires API preview access)."""
     try:
         cache_rid(dataset_rid)
         service = DatasetService(profile=profile)
 
+        formatter.print_warning(
+            "Note: This command requires API preview access. "
+            "If you encounter an 'ApiFeaturePreviewUsageOnly' error, "
+            "use 'pltr dataset schema apply' instead to infer/apply schema."
+        )
+
         with SpinnerProgressTracker().track_spinner(
             f"Fetching schema for {dataset_rid}..."
         ):
@@ -120,7 +126,54 @@ def get_schema(
         formatter.print_error(f"Authentication error: {e}")
         raise typer.Exit(1)
     except Exception as e:
-        formatter.print_error(f"Failed to get schema: {e}")
+        if "ApiFeaturePreviewUsageOnly" in str(e):
+            formatter.print_error(
+                "This command requires API preview access. "
+                "Please use 'pltr dataset schema apply' instead."
+            )
+        else:
+            formatter.print_error(f"Failed to get schema: {e}")
+        raise typer.Exit(1)
+
+
+@schema_app.command("apply")
+def apply_schema(
+    dataset_rid: str = typer.Argument(
+        ..., help="Dataset Resource Identifier", autocompletion=complete_rid
+    ),
+    branch: str = typer.Option("master", "--branch", "-b", help="Dataset branch name"),
+    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,
+    ),
+):
+    """Apply/infer schema for a dataset."""
+    try:
+        cache_rid(dataset_rid)
+        service = DatasetService(profile=profile)
+
+        with SpinnerProgressTracker().track_spinner(
+            f"Applying schema to dataset {dataset_rid} on branch '{branch}'..."
+        ):
+            result = service.apply_schema(dataset_rid, branch)
+
+        formatter.print_success(f"Schema applied successfully to branch '{branch}'")
+
+        # Display result if available
+        if result.get("result"):
+            formatter._format_json(result.get("result"))
+
+    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 apply schema: {e}")
         raise typer.Exit(1)
 
 

pltr/services/base.py

@@ -2,10 +2,13 @@
 Base service class for Foundry API wrappers.
 """
 
-from typing import Any, Optional
+from typing import Any, Optional, Dict
 from abc import ABC, abstractmethod
+import requests
 
 from ..auth.manager import AuthManager
+from ..auth.storage import CredentialStorage
+from ..config.profiles import ProfileManager
 
 
 class BaseService(ABC):
@@ -60,3 +63,67 @@ class BaseService(ABC):
             Configured service instance
         """
         return self._get_service()
+
+    def _make_request(
+        self,
+        method: str,
+        endpoint: str,
+        data: Optional[Dict] = None,
+        json_data: Optional[Dict] = None,
+        headers: Optional[Dict] = None,
+    ) -> requests.Response:
+        """
+        Make a direct HTTP request to Foundry API.
+
+        Args:
+            method: HTTP method (GET, POST, PUT, DELETE)
+            endpoint: API endpoint path (e.g., '/foundry-schema-inference/api/...')
+            data: Form data to send
+            json_data: JSON data to send
+            headers: Additional headers
+
+        Returns:
+            Response object
+
+        Raises:
+            requests.HTTPError: If request fails
+        """
+        # Get credentials for authentication
+        storage = CredentialStorage()
+        profile_manager = ProfileManager()
+        profile_name = self.profile or profile_manager.get_active_profile()
+        if not profile_name:
+            from ..auth.base import ProfileNotFoundError
+
+            raise ProfileNotFoundError(
+                "No profile specified and no default profile configured. "
+                "Run 'pltr configure configure' to set up authentication."
+            )
+        credentials = storage.get_profile(profile_name)
+
+        # Build full URL
+        host = credentials.get("host", "").rstrip("/")
+        url = f"{host}{endpoint}"
+
+        # Set up headers with authentication
+        request_headers = {
+            "Authorization": f"Bearer {credentials.get('token')}",
+            "Content-Type": "application/json",
+            "Accept": "application/json",
+        }
+        if headers:
+            request_headers.update(headers)
+
+        # Make the request
+        response = requests.request(
+            method=method,
+            url=url,
+            data=data,
+            json=json_data,
+            headers=request_headers,
+        )
+
+        # Raise an error for bad status codes
+        response.raise_for_status()
+
+        return response

pltr/services/dataset.py

@@ -56,6 +56,61 @@ class DatasetService(BaseService):
         except Exception as e:
             raise RuntimeError(f"Failed to get schema for dataset {dataset_rid}: {e}")
 
+    def apply_schema(self, dataset_rid: str, branch: str = "master") -> Dict[str, Any]:
+        """
+        Apply/infer schema for a dataset using both schema inference and metadata APIs.
+
+        This method performs two sequential API calls:
+        1. Schema inference to infer the dataset schema
+        2. Schema application to apply the inferred schema to the dataset
+
+        Args:
+            dataset_rid: Dataset Resource Identifier
+            branch: Dataset branch name (default: "master")
+
+        Returns:
+            Schema application result including transaction and version information
+        """
+        try:
+            # Step 1: Call schema inference API to infer the schema
+            inference_endpoint = f"/foundry-schema-inference/api/datasets/{dataset_rid}/branches/{branch}/schema"
+            inference_response = self._make_request(
+                "POST", inference_endpoint, json_data={}
+            )
+
+            # Parse the inference response
+            inference_result = (
+                inference_response.json() if inference_response.text else {}
+            )
+
+            # Extract the foundry schema from the inference result
+            foundry_schema = inference_result.get("data", {}).get("foundrySchema")
+            if not foundry_schema:
+                raise RuntimeError(
+                    "Schema inference failed: No foundrySchema found in response"
+                )
+
+            # Step 2: Call foundry-metadata API to apply the schema
+            metadata_endpoint = f"/foundry-metadata/api/schemas/datasets/{dataset_rid}/branches/{branch}"
+            metadata_response = self._make_request(
+                "POST", metadata_endpoint, json_data=foundry_schema
+            )
+
+            # Parse the metadata response
+            metadata_result = metadata_response.json() if metadata_response.text else {}
+
+            return {
+                "dataset_rid": dataset_rid,
+                "branch": branch,
+                "status": "Schema applied successfully",
+                "inference_result": inference_result,
+                "application_result": metadata_result,
+                "transaction_rid": metadata_result.get("transactionRid"),
+                "version_id": metadata_result.get("versionId"),
+            }
+        except Exception as e:
+            raise RuntimeError(f"Failed to apply schema for dataset {dataset_rid}: {e}")
+
     def put_schema(
         self,
         dataset_rid: str,

{pltr_cli-0.8.0.dist-info → pltr_cli-0.9.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pltr-cli
-Version: 0.8.0
+Version: 0.9.1
 Summary: Command-line interface for Palantir Foundry APIs
 Project-URL: Homepage, https://github.com/anjor/pltr-cli
 Project-URL: Repository, https://github.com/anjor/pltr-cli

{pltr_cli-0.8.0.dist-info → pltr_cli-0.9.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-pltr/__init__.py,sha256=iPlYCcIzuzW7T2HKDkmYlMkRI51dBLfNRxPPiWrfw9U,22
+pltr/__init__.py,sha256=UwJXM8JY2T3tE2id0K2k_lEaVThbRTrGO1mNibyzIz8,22
 pltr/__main__.py,sha256=HWJ49UoAYBQCf8kjuySPmBTuUjTZrOx-y6PzMTyS1KE,879
 pltr/cli.py,sha256=DikRsWsU7QWvRWHgB6wZIct916ebWyaub7PlAjKJXws,2664
 pltr/auth/__init__.py,sha256=G0V-Rh25FaJsH2nhrf146XQQG_ApdbyPJNuHJC25kgk,38
@@ -13,7 +13,7 @@ pltr/commands/alias.py,sha256=r9xMsQNrGvaixSlspzoO2IXQ44LFXuZM4itt8vC0dRc,6862
 pltr/commands/completion.py,sha256=YTxaRL4-rDs5n7aXf3ogFsxbHVJUBo_HiBbd0fbBPZ0,10870
 pltr/commands/configure.py,sha256=oYj-VlOEj3MDwtB2RC4bYOYzI_sXTanPnz7y1GmMTqY,4800
 pltr/commands/connectivity.py,sha256=m8_BYwHij_5IbrYFTU_SYYtbqLCjxA8VIQpbdlWJqHs,14758
-pltr/commands/dataset.py,sha256=nWmIZGpI_KtxY6_c5wbv1gcZOX4__KQg65iDkdObrpg,51403
+pltr/commands/dataset.py,sha256=zuYtBXAGcfRjxE7cP9Hsz2tqSlsdNzdIflGKwytHbVI,53346
 pltr/commands/folder.py,sha256=IAPPA3Smk1IWqThneEtZ08Zp79vDKVUabSkL_nDvUWk,10679
 pltr/commands/mediasets.py,sha256=FXq7OtYU9wLgUxQFcS_fkA4i_CozGnsYKxh8GOSI0ok,15342
 pltr/commands/ontology.py,sha256=zUgSrmv8xi26SQK7GsM3qusgR9Wuka0GyzE7L8DkduE,18317
@@ -31,9 +31,9 @@ pltr/config/profiles.py,sha256=XMUIp6Ez5LNC6rGXZe2JLH7IKepXhARtuc8ASUA9FYA,3431
 pltr/config/settings.py,sha256=bfIiosPqH_W73TOHS71DvgZdAHka4fJDopU1SvBRFuQ,2908
 pltr/services/__init__.py,sha256=zQpgrqPdAkZI-nobi33mctU2-iGNgazzvjBVY8YRbSQ,101
 pltr/services/admin.py,sha256=8FjExmDeIKeVqkAxM83SVvpp_pH9W-Q33cgVs6BHxLQ,9957
-pltr/services/base.py,sha256=R2G781FI-sXtjUyLd91bVnmLb4cYZI3G8U5ndR9NLA4,1593
+pltr/services/base.py,sha256=JF9cyYf7njZuj1ldOLdgzIDhJjOfazBvXPNR-gKVnMY,3682
 pltr/services/connectivity.py,sha256=34kazXhue5gNi1_2s2R5Ma4VQe6jP25CO-ztiPhCeZw,10548
-pltr/services/dataset.py,sha256=GIf73RX6ajdu03-wUWs6SsUYekhRQg-xlfEkcCxMam4,42622
+pltr/services/dataset.py,sha256=LFoK8otJ_MYqH0x5rcmNyIi5X0zr_Q8KP-zG9reArPA,45027
 pltr/services/folder.py,sha256=mWElyvn-wXPB5sv8Ik_dLeW5JM6jZg3g9KKBk6UcrlQ,5389
 pltr/services/mediasets.py,sha256=HgHNFWoG9r-5xupANVOxHg_h5EKsBDl6PsO8hwdbm28,9854
 pltr/services/ontology.py,sha256=iW7qRK8ptlw-u4eAwLNC-mdzLoLZzh7SRqJyok2c3GU,14883
@@ -48,8 +48,8 @@ pltr/utils/alias_resolver.py,sha256=DIF7P1UnUU8kqocJfIDEWjYq4s8_0KfqRZBbECeZEh8,
 pltr/utils/completion.py,sha256=bjeqjleEfB2YcQFpcxvF0GoQ763F6KBbULSZC4FWY_g,4980
 pltr/utils/formatting.py,sha256=38g3G2T5WUFKCCKo42NIBR8_Rdl4pp0ytCEZURoV3l4,50275
 pltr/utils/progress.py,sha256=BKYbiLO61uhQbibabU7pxvvbAWMRLRmqk4pZldBQK_g,9053
-pltr_cli-0.8.0.dist-info/METADATA,sha256=0Z1AZ5LdQgKblhyOy0JaKeoI6QfFcH23hanrwuBXX9U,17714
-pltr_cli-0.8.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pltr_cli-0.8.0.dist-info/entry_points.txt,sha256=8tvEcW04kA_oAE2Dwwu-Og9efjl4ESJvs4AzlP2KBdQ,38
-pltr_cli-0.8.0.dist-info/licenses/LICENSE,sha256=6VUFd_ytnOBD2O1tmkKrA-smigi9QEhYr_tge4h4z8Y,1070
-pltr_cli-0.8.0.dist-info/RECORD,,
+pltr_cli-0.9.1.dist-info/METADATA,sha256=xxIsV9YwPwzv799EzrHcDAqM44xCBmnHREtfy6E-0ek,17714
+pltr_cli-0.9.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pltr_cli-0.9.1.dist-info/entry_points.txt,sha256=8tvEcW04kA_oAE2Dwwu-Og9efjl4ESJvs4AzlP2KBdQ,38
+pltr_cli-0.9.1.dist-info/licenses/LICENSE,sha256=6VUFd_ytnOBD2O1tmkKrA-smigi9QEhYr_tge4h4z8Y,1070
+pltr_cli-0.9.1.dist-info/RECORD,,