pltr-cli 0.7.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

pltr/__init__.py

@@ -1 +1 @@
-__version__ = "0.7.0"
+__version__ = "0.9.0"

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)
 
 
@@ -585,9 +638,30 @@ def upload_file(
 
     except (ProfileNotFoundError, MissingCredentialsError) as e:
         formatter.print_error(f"Authentication error: {e}")
+        raise typer.Exit(1)
+    except FileNotFoundError as e:
+        formatter.print_error(f"File error: {e}")
+        raise typer.Exit(1)
+    except RuntimeError as e:
+        # RuntimeError from our service layer contains detailed error info
+        error_msg = str(e)
+        formatter.print_error(f"Upload failed: {error_msg}")
+
+        # If it looks like our enhanced error message, extract the suggestion part
+        if ". Suggestions: " in error_msg:
+            main_error, suggestions = error_msg.split(". Suggestions: ", 1)
+            formatter.print_error(main_error)
+            formatter.print_info(f"💡 Suggestions: {suggestions}")
+
         raise typer.Exit(1)
     except Exception as e:
-        formatter.print_error(f"Failed to upload file: {e}")
+        # Fallback for any other exceptions
+        formatter.print_error(
+            f"Unexpected error during file upload: {type(e).__name__}: {e}"
+        )
+        formatter.print_info(
+            "💡 Try running the command again or check your connection"
+        )
         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,33 @@ 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 the schema inference API.
+
+        Args:
+            dataset_rid: Dataset Resource Identifier
+            branch: Dataset branch name (default: "master")
+
+        Returns:
+            Schema inference result
+        """
+        try:
+            endpoint = f"/foundry-schema-inference/api/datasets/{dataset_rid}/branches/{branch}/schema"
+            response = self._make_request("POST", endpoint, json_data={})
+
+            # Parse the response
+            result = response.json() if response.text else {}
+
+            return {
+                "dataset_rid": dataset_rid,
+                "branch": branch,
+                "status": "Schema applied successfully",
+                "result": result,
+            }
+        except Exception as e:
+            raise RuntimeError(f"Failed to apply schema for dataset {dataset_rid}: {e}")
+
     def put_schema(
         self,
         dataset_rid: str,
@@ -319,9 +346,93 @@ class DatasetService(BaseService):
                 "transaction_rid": getattr(result, "transaction_rid", transaction_rid),
             }
         except Exception as e:
-            raise RuntimeError(
-                f"Failed to upload file {file_path} to dataset {dataset_rid}: {e}"
-            )
+            # Try to extract more detailed error information
+            error_msg = str(e).strip()
+            error_type = type(e).__name__
+
+            # Check for common HTTP/API errors
+            if hasattr(e, "response") and hasattr(e.response, "status_code"):
+                status_code = e.response.status_code
+                if hasattr(e.response, "text"):
+                    response_text = e.response.text[:500]  # Limit to 500 chars
+                    error_details = f"HTTP {status_code}: {response_text}"
+                else:
+                    error_details = f"HTTP {status_code}"
+                error_msg = f"{error_details} ({error_type}: {error_msg})"
+            elif hasattr(e, "status_code"):
+                error_msg = f"HTTP {e.status_code}: {error_msg}"
+            elif hasattr(e, "message"):
+                error_msg = f"{error_type}: {e.message}"
+            else:
+                if error_msg:
+                    error_msg = f"{error_type}: {error_msg}"
+                else:
+                    error_msg = f"{error_type} (no additional details available)"
+
+            # Add context about what might have gone wrong
+            context_hints = []
+            error_lower = error_msg.lower()
+
+            if (
+                "permission" in error_lower
+                or "forbidden" in error_lower
+                or "401" in error_msg
+                or "403" in error_msg
+            ):
+                context_hints.append(
+                    "Check your authentication credentials and dataset permissions"
+                )
+            if "not found" in error_lower or "404" in error_msg:
+                context_hints.append(
+                    "Verify the dataset RID and transaction RID are correct"
+                )
+            if "transaction" in error_lower:
+                context_hints.append(
+                    "Check if the transaction is still open and not expired"
+                )
+            if "schema" in error_lower or "validation" in error_lower:
+                context_hints.append(
+                    "The file might not match the expected dataset schema"
+                )
+            if (
+                "invalidparametercombination" in error_lower
+                or "invalid parameter" in error_lower
+            ):
+                context_hints.append(
+                    "The combination of parameters (dataset RID, transaction RID, branch) may be invalid"
+                )
+                context_hints.append(
+                    "Try without --transaction-rid, or verify the transaction belongs to this dataset"
+                )
+            if (
+                "opentransactionalreadyexists" in error_lower
+                or "transaction already exists" in error_lower
+            ):
+                context_hints.append(
+                    "There's already an open transaction for this dataset"
+                )
+                context_hints.append(
+                    "Use the existing transaction with --transaction-rid, or commit/abort it first"
+                )
+                context_hints.append(
+                    "List transactions with: pltr dataset transactions list "
+                    + dataset_rid
+                )
+
+            # Try to get more detailed error information from the exception
+            if hasattr(e, "__dict__"):
+                for attr in ["detail", "details", "error_message", "description"]:
+                    if hasattr(e, attr):
+                        detail = getattr(e, attr)
+                        if detail and str(detail).strip():
+                            error_msg += f" - {detail}"
+                            break
+
+            full_error = f"Failed to upload file {file_path.name} to dataset {dataset_rid}: {error_msg}"
+            if context_hints:
+                full_error += f". Suggestions: {'; '.join(context_hints)}"
+
+            raise RuntimeError(full_error)
 
     def download_file(
         self,

{pltr_cli-0.7.0.dist-info → pltr_cli-0.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pltr-cli
-Version: 0.7.0
+Version: 0.9.0
 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.7.0.dist-info → pltr_cli-0.9.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-pltr/__init__.py,sha256=RaANGbRu5e-vehwXI1-Qe2ggPPfs1TQaZj072JdbLk4,22
+pltr/__init__.py,sha256=H9NWRZb7NbeRRPLP_V1fARmLNXranorVM-OOY-8_2ug,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=NqGGF5IGhLGuy6FZxG-hd0p6yWlrVblzNwVqNjv3z20,50536
+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=23EBeJrFZkUBOU8EJcf5uMZyu-10rmIwMVGDZ4RZ2lI,38827
+pltr/services/dataset.py,sha256=UuPJ11OryJZg2zUSRdtwJq_pj9unE24-mYGr5oBWobU,43602
 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.7.0.dist-info/METADATA,sha256=yuvIf5pOqN1pl_I_s6czGO8IeXVfEgZuyLGFsKjC8mw,17714
-pltr_cli-0.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pltr_cli-0.7.0.dist-info/entry_points.txt,sha256=8tvEcW04kA_oAE2Dwwu-Og9efjl4ESJvs4AzlP2KBdQ,38
-pltr_cli-0.7.0.dist-info/licenses/LICENSE,sha256=6VUFd_ytnOBD2O1tmkKrA-smigi9QEhYr_tge4h4z8Y,1070
-pltr_cli-0.7.0.dist-info/RECORD,,
+pltr_cli-0.9.0.dist-info/METADATA,sha256=Fddmpgwz08LfEHN4sjHymbv0a69Pi9Ss7hA7t-Oxx08,17714
+pltr_cli-0.9.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pltr_cli-0.9.0.dist-info/entry_points.txt,sha256=8tvEcW04kA_oAE2Dwwu-Og9efjl4ESJvs4AzlP2KBdQ,38
+pltr_cli-0.9.0.dist-info/licenses/LICENSE,sha256=6VUFd_ytnOBD2O1tmkKrA-smigi9QEhYr_tge4h4z8Y,1070
+pltr_cli-0.9.0.dist-info/RECORD,,