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

pltr/__init__.py

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

pltr/services/dataset.py

@@ -149,14 +149,12 @@ class DatasetService(BaseService):
             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,
-                )
+            result = self.service.Dataset.File.upload(
+                dataset_rid=dataset_rid,
+                file_path=str(file_path),
+                branch_name=branch,
+                transaction_rid=transaction_rid,
+            )
 
             return {
                 "dataset_rid": dataset_rid,
@@ -196,8 +194,8 @@ class DatasetService(BaseService):
             # 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
+            file_content = self.service.Dataset.File.read(
+                dataset_rid=dataset_rid, file_path=file_path, branch_name=branch
             )
 
             # Write file content to disk
@@ -236,7 +234,9 @@ class DatasetService(BaseService):
             List of file information dictionaries
         """
         try:
-            files = self.service.list_files(dataset_rid=dataset_rid, branch=branch)
+            files = self.service.Dataset.File.list(
+                dataset_rid=dataset_rid, branch_name=branch
+            )
 
             return [
                 {
@@ -322,11 +322,11 @@ class DatasetService(BaseService):
             Transaction information dictionary
         """
         try:
-            # Try to use the v2 API's Dataset.create_transaction method
-            transaction = self.service.Dataset.create_transaction(
+            # Use the v2 API's Dataset.Transaction.create method
+            transaction = self.service.Dataset.Transaction.create(
                 dataset_rid=dataset_rid,
-                branch=branch,
                 transaction_type=transaction_type,
+                branch_name=branch,
             )
 
             return {
@@ -338,27 +338,6 @@ class DatasetService(BaseService):
                 "created_time": getattr(transaction, "created_time", None),
                 "created_by": getattr(transaction, "created_by", None),
             }
-        except AttributeError:
-            # Fallback to service.create_transaction if available
-            try:
-                transaction = self.service.create_transaction(
-                    dataset_rid=dataset_rid,
-                    branch=branch,
-                    transaction_type=transaction_type,
-                )
-                return {
-                    "transaction_rid": getattr(transaction, "rid", None),
-                    "dataset_rid": dataset_rid,
-                    "branch": branch,
-                    "transaction_type": transaction_type,
-                    "status": getattr(transaction, "status", "OPEN"),
-                    "created_time": getattr(transaction, "created_time", None),
-                    "created_by": getattr(transaction, "created_by", None),
-                }
-            except Exception as e:
-                raise RuntimeError(
-                    f"Failed to create transaction for dataset {dataset_rid}: {e}"
-                )
         except Exception as e:
             raise RuntimeError(
                 f"Failed to create transaction for dataset {dataset_rid}: {e}"
@@ -378,8 +357,8 @@ class DatasetService(BaseService):
             Transaction commit result
         """
         try:
-            # Try the v2 API first (Dataset.commit_transaction)
-            self.service.Dataset.commit_transaction(
+            # Use the v2 API Dataset.Transaction.commit method
+            self.service.Dataset.Transaction.commit(
                 dataset_rid=dataset_rid, transaction_rid=transaction_rid
             )
 
@@ -390,23 +369,6 @@ class DatasetService(BaseService):
                 "committed_time": None,  # Would need to get this from a status call
                 "success": True,
             }
-        except AttributeError:
-            # Fallback to service.commit_transaction
-            try:
-                self.service.commit_transaction(
-                    dataset_rid=dataset_rid, transaction_rid=transaction_rid
-                )
-                return {
-                    "transaction_rid": transaction_rid,
-                    "dataset_rid": dataset_rid,
-                    "status": "COMMITTED",
-                    "committed_time": None,
-                    "success": True,
-                }
-            except Exception as e:
-                raise RuntimeError(
-                    f"Failed to commit transaction {transaction_rid}: {e}"
-                )
         except Exception as e:
             raise RuntimeError(f"Failed to commit transaction {transaction_rid}: {e}")
 
@@ -424,8 +386,8 @@ class DatasetService(BaseService):
             Transaction abort result
         """
         try:
-            # Try the v2 API first (Dataset.abort_transaction)
-            self.service.Dataset.abort_transaction(
+            # Use the v2 API Dataset.Transaction.abort method
+            self.service.Dataset.Transaction.abort(
                 dataset_rid=dataset_rid, transaction_rid=transaction_rid
             )
 
@@ -436,23 +398,6 @@ class DatasetService(BaseService):
                 "aborted_time": None,  # Would need to get this from a status call
                 "success": True,
             }
-        except AttributeError:
-            # Fallback to service.abort_transaction
-            try:
-                self.service.abort_transaction(
-                    dataset_rid=dataset_rid, transaction_rid=transaction_rid
-                )
-                return {
-                    "transaction_rid": transaction_rid,
-                    "dataset_rid": dataset_rid,
-                    "status": "ABORTED",
-                    "aborted_time": None,
-                    "success": True,
-                }
-            except Exception as e:
-                raise RuntimeError(
-                    f"Failed to abort transaction {transaction_rid}: {e}"
-                )
         except Exception as e:
             raise RuntimeError(f"Failed to abort transaction {transaction_rid}: {e}")
 
@@ -470,8 +415,8 @@ class DatasetService(BaseService):
             Transaction status information
         """
         try:
-            # Try the v2 API first (Dataset.get_transaction)
-            transaction = self.service.Dataset.get_transaction(
+            # Use the v2 API Dataset.Transaction.get method
+            transaction = self.service.Dataset.Transaction.get(
                 dataset_rid=dataset_rid, transaction_rid=transaction_rid
             )
 
@@ -486,27 +431,6 @@ class DatasetService(BaseService):
                 "committed_time": getattr(transaction, "committed_time", None),
                 "aborted_time": getattr(transaction, "aborted_time", None),
             }
-        except AttributeError:
-            # Fallback to service.get_transaction
-            try:
-                transaction = self.service.get_transaction(
-                    dataset_rid=dataset_rid, transaction_rid=transaction_rid
-                )
-                return {
-                    "transaction_rid": transaction_rid,
-                    "dataset_rid": dataset_rid,
-                    "status": getattr(transaction, "status", None),
-                    "transaction_type": getattr(transaction, "transaction_type", None),
-                    "branch": getattr(transaction, "branch", None),
-                    "created_time": getattr(transaction, "created_time", None),
-                    "created_by": getattr(transaction, "created_by", None),
-                    "committed_time": getattr(transaction, "committed_time", None),
-                    "aborted_time": getattr(transaction, "aborted_time", None),
-                }
-            except Exception as e:
-                raise RuntimeError(
-                    f"Failed to get transaction status {transaction_rid}: {e}"
-                )
         except Exception as e:
             raise RuntimeError(
                 f"Failed to get transaction status {transaction_rid}: {e}"
@@ -526,9 +450,10 @@ class DatasetService(BaseService):
             List of transaction information dictionaries
         """
         try:
-            # Try the v2 API first (Dataset.list_transactions)
-            transactions = self.service.Dataset.list_transactions(
-                dataset_rid=dataset_rid, branch=branch
+            # Use the v2 API Dataset.Transaction for transaction listing
+            # Note: This method may not exist in all SDK versions
+            transactions = self.service.Dataset.Transaction.list(
+                dataset_rid=dataset_rid, branch_name=branch
             )
 
             return [
@@ -545,37 +470,11 @@ class DatasetService(BaseService):
                 for transaction in transactions
             ]
         except AttributeError:
-            # Fallback to service.list_transactions
-            try:
-                transactions = self.service.list_transactions(
-                    dataset_rid=dataset_rid, branch=branch
-                )
-
-                return [
-                    {
-                        "transaction_rid": getattr(transaction, "rid", None),
-                        "created_time": getattr(transaction, "created_time", None),
-                        "created_by": getattr(transaction, "created_by", None),
-                        "status": getattr(transaction, "status", None),
-                        "transaction_type": getattr(
-                            transaction, "transaction_type", None
-                        ),
-                        "branch": getattr(transaction, "branch", branch),
-                        "committed_time": getattr(transaction, "committed_time", None),
-                        "aborted_time": getattr(transaction, "aborted_time", None),
-                    }
-                    for transaction in transactions
-                ]
-            except AttributeError:
-                # Method not available in this SDK version
-                raise NotImplementedError(
-                    "Transaction listing is not supported by this SDK version. "
-                    "This feature may require a newer version of foundry-platform-python."
-                )
-            except Exception as e:
-                raise RuntimeError(
-                    f"Failed to get transactions for dataset {dataset_rid}: {e}"
-                )
+            # Method not available in this SDK version
+            raise NotImplementedError(
+                "Transaction listing is not supported by this SDK version. "
+                "This feature may require a newer version of foundry-platform-python."
+            )
         except Exception as e:
             raise RuntimeError(
                 f"Failed to get transactions for dataset {dataset_rid}: {e}"

{pltr_cli-0.5.1.dist-info → pltr_cli-0.5.2.dist-info}/METADATA

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

@@ -1,4 +1,4 @@
-pltr/__init__.py,sha256=LBK46heutvn3KmsCrKIYu8RQikbfnjZaj2xFrXaeCzQ,22
+pltr/__init__.py,sha256=isJrmDBLRag7Zc2UK9ZovWGOv7ji1Oh-zJtJMNJFkXw,22
 pltr/__main__.py,sha256=HWJ49UoAYBQCf8kjuySPmBTuUjTZrOx-y6PzMTyS1KE,879
 pltr/cli.py,sha256=DikRsWsU7QWvRWHgB6wZIct916ebWyaub7PlAjKJXws,2664
 pltr/auth/__init__.py,sha256=G0V-Rh25FaJsH2nhrf146XQQG_ApdbyPJNuHJC25kgk,38
@@ -33,7 +33,7 @@ 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/connectivity.py,sha256=34kazXhue5gNi1_2s2R5Ma4VQe6jP25CO-ztiPhCeZw,10548
-pltr/services/dataset.py,sha256=L_SZ0yFIq3Zlx_eHtOwpRYZcUh1xGG7pIlbCRnYjrbs,25175
+pltr/services/dataset.py,sha256=OeKRkkTRT_3uALsvipVzaFIOYRatqXUxLbLkbyCiqLc,20478
 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=MuIgi8dSqvhzik7H0YQOq1sYPnY_poZOYeTVvrIY2Jk,44235
 pltr/utils/progress.py,sha256=BKYbiLO61uhQbibabU7pxvvbAWMRLRmqk4pZldBQK_g,9053
-pltr_cli-0.5.1.dist-info/METADATA,sha256=eQiiY1B0njuRJbz5-rFJ9qsFkeIwVD2aPmGHIPmajhY,17714
-pltr_cli-0.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pltr_cli-0.5.1.dist-info/entry_points.txt,sha256=8tvEcW04kA_oAE2Dwwu-Og9efjl4ESJvs4AzlP2KBdQ,38
-pltr_cli-0.5.1.dist-info/licenses/LICENSE,sha256=6VUFd_ytnOBD2O1tmkKrA-smigi9QEhYr_tge4h4z8Y,1070
-pltr_cli-0.5.1.dist-info/RECORD,,
+pltr_cli-0.5.2.dist-info/METADATA,sha256=vWl7v5qxGJIOJYHLsRjY4fHdCPk6sWav2BTn-xS0F1Y,17714
+pltr_cli-0.5.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pltr_cli-0.5.2.dist-info/entry_points.txt,sha256=8tvEcW04kA_oAE2Dwwu-Og9efjl4ESJvs4AzlP2KBdQ,38
+pltr_cli-0.5.2.dist-info/licenses/LICENSE,sha256=6VUFd_ytnOBD2O1tmkKrA-smigi9QEhYr_tge4h4z8Y,1070
+pltr_cli-0.5.2.dist-info/RECORD,,