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

pltr/services/dataset_full.py

@@ -1,302 +0,0 @@
-"""
-Dataset service wrapper for Foundry SDK.
-"""
-
-from typing import Any, Optional, List, Dict, Union
-from pathlib import Path
-
-from .base import BaseService
-
-
-class DatasetService(BaseService):
-    """Service wrapper for Foundry dataset operations."""
-
-    def _get_service(self) -> Any:
-        """Get the Foundry datasets service."""
-        return self.client.datasets
-
-    # list_datasets method removed - not supported by foundry-platform-sdk v1.27.0
-
-    def get_dataset(self, dataset_rid: str) -> Dict[str, Any]:
-        """
-        Get information about a specific dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            Dataset information dictionary
-        """
-        try:
-            # Use the v2 API's Dataset.get method
-            dataset = self.service.Dataset.get(dataset_rid)
-            return self._format_dataset_info(dataset)
-        except Exception as e:
-            raise RuntimeError(f"Failed to get dataset {dataset_rid}: {e}")
-
-    def get_schema(self, dataset_rid: str) -> Dict[str, Any]:
-        """
-        Get dataset schema.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            Schema information
-        """
-        try:
-            schema = self.service.Dataset.get_schema(dataset_rid)
-            return {
-                "dataset_rid": dataset_rid,
-                "schema": schema,
-                "type": str(type(schema)),
-                "status": "Schema retrieved successfully",
-            }
-        except Exception as e:
-            raise RuntimeError(f"Failed to get schema for dataset {dataset_rid}: {e}")
-
-    def create_dataset(
-        self, name: str, parent_folder_rid: Optional[str] = None
-    ) -> Dict[str, Any]:
-        """
-        Create a new dataset.
-
-        Args:
-            name: Dataset name
-            parent_folder_rid: Parent folder RID (optional)
-
-        Returns:
-            Created dataset information
-        """
-        try:
-            dataset = self.service.create_dataset(
-                name=name, parent_folder_rid=parent_folder_rid
-            )
-            return self._format_dataset_info(dataset)
-        except Exception as e:
-            raise RuntimeError(f"Failed to create dataset '{name}': {e}")
-
-    def delete_dataset(self, dataset_rid: str) -> bool:
-        """
-        Delete a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            True if deletion was successful
-        """
-        try:
-            self.service.delete_dataset(dataset_rid)
-            return True
-        except Exception as e:
-            raise RuntimeError(f"Failed to delete dataset {dataset_rid}: {e}")
-
-    def upload_file(
-        self,
-        dataset_rid: str,
-        file_path: Union[str, Path],
-        branch: str = "master",
-        transaction_rid: Optional[str] = None,
-    ) -> Dict[str, Any]:
-        """
-        Upload a file to a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-            file_path: Path to file to upload
-            branch: Dataset branch name
-            transaction_rid: Transaction RID (optional)
-
-        Returns:
-            Upload result information
-        """
-        file_path = Path(file_path)
-        if not file_path.exists():
-            raise FileNotFoundError(f"File not found: {file_path}")
-
-        try:
-            with open(file_path, "rb") as f:
-                result = self.service.upload_file(
-                    dataset_rid=dataset_rid,
-                    file_path=file_path.name,
-                    file_data=f,
-                    branch=branch,
-                    transaction_rid=transaction_rid,
-                )
-
-            return {
-                "dataset_rid": dataset_rid,
-                "file_path": str(file_path),
-                "branch": branch,
-                "size_bytes": file_path.stat().st_size,
-                "uploaded": True,
-                "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}"
-            )
-
-    def download_file(
-        self,
-        dataset_rid: str,
-        file_path: str,
-        output_path: Union[str, Path],
-        branch: str = "master",
-    ) -> Dict[str, Any]:
-        """
-        Download a file from a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-            file_path: Path of file within dataset
-            output_path: Local path to save the downloaded file
-            branch: Dataset branch name
-
-        Returns:
-            Download result information
-        """
-        output_path = Path(output_path)
-
-        try:
-            # Ensure output directory exists
-            output_path.parent.mkdir(parents=True, exist_ok=True)
-
-            file_content = self.service.download_file(
-                dataset_rid=dataset_rid, file_path=file_path, branch=branch
-            )
-
-            # Write file content to disk
-            with open(output_path, "wb") as f:
-                if hasattr(file_content, "read"):
-                    # If it's a stream
-                    f.write(file_content.read())
-                else:
-                    # If it's bytes
-                    f.write(file_content)
-
-            return {
-                "dataset_rid": dataset_rid,
-                "file_path": file_path,
-                "output_path": str(output_path),
-                "branch": branch,
-                "size_bytes": output_path.stat().st_size,
-                "downloaded": True,
-            }
-        except Exception as e:
-            raise RuntimeError(
-                f"Failed to download file {file_path} from dataset {dataset_rid}: {e}"
-            )
-
-    def list_files(
-        self, dataset_rid: str, branch: str = "master"
-    ) -> List[Dict[str, Any]]:
-        """
-        List files in a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-            branch: Dataset branch name
-
-        Returns:
-            List of file information dictionaries
-        """
-        try:
-            files = self.service.list_files(dataset_rid=dataset_rid, branch=branch)
-
-            return [
-                {
-                    "path": file.path,
-                    "size_bytes": getattr(file, "size_bytes", None),
-                    "last_modified": getattr(file, "last_modified", None),
-                    "transaction_rid": getattr(file, "transaction_rid", None),
-                }
-                for file in files
-            ]
-        except Exception as e:
-            raise RuntimeError(f"Failed to list files in dataset {dataset_rid}: {e}")
-
-    def get_branches(self, dataset_rid: str) -> List[Dict[str, Any]]:
-        """
-        Get list of branches for a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            List of branch information dictionaries
-        """
-        try:
-            branches = self.service.list_branches(dataset_rid=dataset_rid)
-
-            return [
-                {
-                    "name": branch.name,
-                    "transaction_rid": getattr(branch, "transaction_rid", None),
-                    "created_time": getattr(branch, "created_time", None),
-                    "created_by": getattr(branch, "created_by", None),
-                }
-                for branch in branches
-            ]
-        except Exception as e:
-            raise RuntimeError(f"Failed to get branches for dataset {dataset_rid}: {e}")
-
-    def create_branch(
-        self, dataset_rid: str, branch_name: str, parent_branch: str = "master"
-    ) -> Dict[str, Any]:
-        """
-        Create a new branch for a dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-            branch_name: Name for the new branch
-            parent_branch: Parent branch to branch from
-
-        Returns:
-            Created branch information
-        """
-        try:
-            branch = self.service.create_branch(
-                dataset_rid=dataset_rid,
-                branch_name=branch_name,
-                parent_branch=parent_branch,
-            )
-
-            return {
-                "name": branch.name,
-                "dataset_rid": dataset_rid,
-                "parent_branch": parent_branch,
-                "transaction_rid": getattr(branch, "transaction_rid", None),
-            }
-        except Exception as e:
-            raise RuntimeError(
-                f"Failed to create branch '{branch_name}' for dataset {dataset_rid}: {e}"
-            )
-
-    def _format_dataset_info(self, dataset: Any) -> Dict[str, Any]:
-        """
-        Format dataset information for consistent output.
-
-        Args:
-            dataset: Dataset object from Foundry SDK
-
-        Returns:
-            Formatted dataset information dictionary
-        """
-        # The v2 Dataset object has different attributes
-        return {
-            "rid": getattr(dataset, "rid", "unknown"),
-            "name": getattr(dataset, "name", "Unknown"),
-            "description": getattr(dataset, "description", ""),
-            "path": getattr(dataset, "path", None),
-            "created": getattr(dataset, "created", None),
-            "modified": getattr(dataset, "modified", None),
-            # Try to get additional attributes that might exist
-            "created_time": getattr(dataset, "created_time", None),
-            "created_by": getattr(dataset, "created_by", None),
-            "last_modified": getattr(dataset, "last_modified", None),
-            "size_bytes": getattr(dataset, "size_bytes", None),
-            "schema_id": getattr(dataset, "schema_id", None),
-            "parent_folder_rid": getattr(dataset, "parent_folder_rid", None),
-        }

pltr/services/dataset_v2.py

@@ -1,128 +0,0 @@
-"""
-Dataset service wrapper for Foundry SDK v2 API.
-This is a simplified version that works with the actual foundry_sdk API.
-"""
-
-from typing import Any, Optional, Dict
-
-from .base import BaseService
-
-
-class DatasetServiceV2(BaseService):
-    """Service wrapper for Foundry dataset operations using v2 API."""
-
-    def _get_service(self) -> Any:
-        """Get the Foundry datasets service."""
-        return self.client.datasets
-
-    def get_dataset(self, dataset_rid: str) -> Dict[str, Any]:
-        """
-        Get information about a specific dataset.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            Dataset information dictionary
-        """
-        try:
-            # The v2 API returns a Dataset object
-            dataset = self.service.Dataset.get(dataset_rid)
-
-            # Format the response
-            return {
-                "rid": dataset_rid,
-                "name": getattr(dataset, "name", "Unknown"),
-                "description": getattr(dataset, "description", ""),
-                "path": getattr(dataset, "path", ""),
-                "created": getattr(dataset, "created_time", None),
-                "modified": getattr(dataset, "modified_time", None),
-                # The actual attributes available depend on the SDK version
-                "status": "Retrieved successfully",
-            }
-        except Exception as e:
-            raise RuntimeError(f"Failed to get dataset {dataset_rid}: {e}")
-
-    def create_dataset(
-        self, name: str, parent_folder_rid: Optional[str] = None
-    ) -> Dict[str, Any]:
-        """
-        Create a new dataset.
-
-        Args:
-            name: Dataset name
-            parent_folder_rid: Parent folder RID (optional)
-
-        Returns:
-            Created dataset information
-        """
-        try:
-            # The create method parameters depend on the SDK version
-            dataset = self.service.Dataset.create(
-                name=name, parent_folder_rid=parent_folder_rid
-            )
-
-            return {
-                "rid": getattr(dataset, "rid", "unknown"),
-                "name": name,
-                "parent_folder_rid": parent_folder_rid,
-                "status": "Created successfully",
-            }
-        except Exception as e:
-            raise RuntimeError(f"Failed to create dataset '{name}': {e}")
-
-    def read_table(self, dataset_rid: str, format: str = "arrow") -> Any:
-        """
-        Read dataset as a table.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-            format: Output format (arrow, pandas, etc.)
-
-        Returns:
-            Table data in specified format
-        """
-        try:
-            return self.service.Dataset.read_table(dataset_rid, format=format)
-        except Exception as e:
-            raise RuntimeError(f"Failed to read dataset {dataset_rid}: {e}")
-
-    def get_schema(self, dataset_rid: str) -> Dict[str, Any]:
-        """
-        Get dataset schema.
-
-        Args:
-            dataset_rid: Dataset Resource Identifier
-
-        Returns:
-            Schema information
-        """
-        try:
-            schema = self.service.Dataset.get_schema(dataset_rid)
-
-            # Format schema for display
-            return {
-                "dataset_rid": dataset_rid,
-                "schema": schema,
-                "status": "Schema retrieved successfully",
-            }
-        except Exception as e:
-            raise RuntimeError(f"Failed to get schema for dataset {dataset_rid}: {e}")
-
-    def list_datasets(self, limit: Optional[int] = None) -> list:
-        """
-        List datasets - NOT SUPPORTED BY SDK v2.
-
-        The foundry_sdk v2 API doesn't provide a list_datasets method.
-        Dataset operations are RID-based. You need to know the dataset RID
-        to interact with it.
-
-        Raises:
-            NotImplementedError: This operation is not supported by the SDK
-        """
-        raise NotImplementedError(
-            "Dataset listing is not supported by foundry_sdk v2. "
-            "The SDK requires knowing dataset RIDs in advance. "
-            "Consider using the Foundry web interface to find dataset RIDs, "
-            "or contact your Foundry administrator for dataset information."
-        )