praetorian-cli 2.2.15__tar.gz → 2.2.17__tar.gz

{praetorian_cli-2.2.15/praetorian_cli.egg-info → praetorian_cli-2.2.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praetorian-cli
-Version: 2.2.15
+Version: 2.2.17
 Summary: For interacting with the Chariot API
 Home-page: https://github.com/praetorian-inc/praetorian-cli
 Author: Praetorian

{praetorian_cli-2.2.15 → praetorian_cli-2.2.17}/praetorian_cli/sdk/chariot.py

@@ -66,11 +66,9 @@ class Chariot:
 
     def chariot_request(self, method: str, url: str, headers: dict = {}, **kwargs) -> requests.Response:
         """
-        Centralized wrapper around requests.request. Take care of proxy, beta flag, and 
+        Centralized wrapper around requests.request. Takes care of proxy and 
         supplies the authentication headers
         """
-        self.add_beta_url_param(kwargs)
-
         if self.proxy:
             kwargs['proxies'] = {'http': self.proxy, 'https': self.proxy}
             kwargs['verify'] = False
@@ -78,12 +76,6 @@ class Chariot:
         return requests.request(method, url, headers=(headers | self.keychain.headers()), **kwargs)
 
 
-    def add_beta_url_param(self, kwargs: dict):
-        if 'params' in kwargs:
-            kwargs['params']['beta'] = 'true'
-        else:
-            kwargs['params'] = {'beta': 'true'}
-
     def my(self, params: dict, pages=1) -> dict:
         final_resp = dict()
 

{praetorian_cli-2.2.15 → praetorian_cli-2.2.17}/praetorian_cli/sdk/entities/preseeds.py

@@ -85,9 +85,9 @@ class Preseeds:
         """
         return self.api.search.by_exact_key(key, details)
 
-    def update(self, key, status):
+    def update(self, key, status, comment=None):
         """Update the status of an existing preseed.
-        
+
         Only the status field can be meaningfully updated for preseeds. This allows
         you to activate, freeze, or mark preseeds for deletion without recreating them.
         The preseed must exist before it can be updated.
@@ -101,24 +101,29 @@ class Preseeds:
                       - 'FR' (FROZEN_REJECTED): Preseed is frozen and rejected
                       - 'P' (PENDING): Preseed is pending activation
         :type status: str
+        :param comment: Optional comment to include with the update (stored in History field)
+        :type comment: str or None
         :return: The updated preseed object with new status
         :rtype: dict
         :raises: Prints error message if preseed with the specified key is not found
 
         **Example Usage:**
-        
+
         .. code-block:: python
-        
+
+            # Update status only
             sdk.preseeds.update(
                 "#preseed#whois+company#Example Company#example company",
                 "A"
             )
-            
+
+            # Update status with comment
             sdk.preseeds.update(
                 "#preseed#dns+subdomain#API Subdomain#api.example.com",
-                "F"
+                "F",
+                comment="Temporarily frozen pending review"
             )
-            
+
             sdk.preseeds.update(
                 "#preseed#whois+company#Old Company#old company",
                 "D"
@@ -130,7 +135,13 @@ class Preseeds:
         """
         preseed = self.api.search.by_exact_key(key)
         if preseed:
-            return self.api.update('preseed', dict(key=key, status=status))
+            update_payload = {'key': key, 'status': status}
+
+            # Include comment if provided
+            if comment is not None:
+                update_payload['comment'] = comment
+
+            return self.api.update('preseed', update_payload)
         else:
             error(f'Pre-seed {key} is not found.')
 

{praetorian_cli-2.2.15 → praetorian_cli-2.2.17}/praetorian_cli/sdk/entities/risks.py

@@ -201,4 +201,4 @@ def get_note_entry_indices(history):
 
 
 def is_note_entry(entry):
-    return entry.get('comment') and entry.get('from') is None
+    return entry.get('comment') is not None

{praetorian_cli-2.2.15 → praetorian_cli-2.2.17}/praetorian_cli/sdk/entities/seeds.py

@@ -69,26 +69,46 @@ class Seeds:
             return None
         return results[0]
 
-    def update(self, key, status=None):
+    def update(self, key, status=None, comment=None):
         """
         Update seed fields dynamically.
-        
+
         :param key: Seed/Asset key (e.g., '#seed#domain#example.com' or '#asset#domain#example.com')
         :type key: str
         :param status: Status for backward compatibility (can be positional)
         :type status: str or None
+        :param comment: Optional comment to include with the update (stored in History field)
+        :type comment: str or None
         :param kwargs: Fields to update
         :return: The updated seed, or None if the seed was not found
         :rtype: dict or None
+
+        **Example Usage:**
+
+        .. code-block:: python
+
+            # Update status only
+            sdk.seeds.update("#asset#domain#example.com", "A")
+
+            # Update status with comment
+            sdk.seeds.update(
+                "#asset#domain#example.com",
+                "A",
+                comment="Activated after verification"
+            )
         """
-            
+
         seed = self.get(key)  # This already handles old key format conversion
         if seed:
             update_payload = {
                 'key': key,
                 'status': status
             }
-            
+
+            # Include comment if provided
+            if comment is not None:
+                update_payload['comment'] = comment
+
             return self.api.upsert('seed', update_payload)
         else:
             error(f'Seed {key} not found.')

{praetorian_cli-2.2.15 → praetorian_cli-2.2.17}/praetorian_cli/sdk/test/test_preseed.py

@@ -27,5 +27,15 @@ class TestPreseed:
         assert len(preseeds) > 0
         assert preseeds[0]['status'] == Preseed.FROZEN_REJECTED.value
 
+    def test_update_preseed_with_comment(self):
+        # Update preseed with comment parameter
+        self.sdk.preseeds.update(self.preseed_key, Preseed.ACTIVE.value, comment="Test comment for preseed update")
+        preseed = self.sdk.preseeds.get(self.preseed_key, details=True)
+        assert preseed['status'] == Preseed.ACTIVE.value
+        # Verify comment was stored in history
+        assert 'history' in preseed
+        assert len(preseed['history']) > 0
+        assert preseed['history'][0]['comment'] == "Test comment for preseed update"
+
     def teardown_class(self):
         clean_test_entities(self.sdk, self)

{praetorian_cli-2.2.15 → praetorian_cli-2.2.17}/praetorian_cli/sdk/test/test_seed.py

@@ -29,6 +29,16 @@ class TestSeed:
         self.sdk.seeds.update(self.seed_asset_key, Seed.ACTIVE.value)
         assert self.get_seed()['status'] == Seed.ACTIVE.value
 
+    def test_update_seed_with_comment(self):
+        # Update seed with comment parameter
+        self.sdk.seeds.update(self.seed_asset_key, Seed.FROZEN.value, comment="Test comment for seed update")
+        seed = self.get_seed()
+        assert seed['status'] == Seed.FROZEN.value
+        # Verify comment was stored in history
+        assert 'history' in seed
+        assert len(seed['history']) > 0
+        assert seed['history'][0]['comment'] == "Test comment for seed update"
+
     def test_delete_seed(self):
         self.sdk.seeds.delete(self.seed_asset_key)
         assert self.sdk.seeds.get(self.seed_asset_key)['status'] == Seed.DELETED.value

{praetorian_cli-2.2.15 → praetorian_cli-2.2.17/praetorian_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praetorian-cli
-Version: 2.2.15
+Version: 2.2.17
 Summary: For interacting with the Chariot API
 Home-page: https://github.com/praetorian-inc/praetorian-cli
 Author: Praetorian

{praetorian_cli-2.2.15 → praetorian_cli-2.2.17}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = praetorian-cli
-version = 2.2.15
+version = 2.2.17
 author = Praetorian
 author_email = support@praetorian.com
 description = For interacting with the Chariot API