praetorian-cli 2.2.15__py3-none-any.whl → 2.2.17__py3-none-any.whl

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/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/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/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/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/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.dist-info → praetorian_cli-2.2.17.dist-info}/METADATA

@@ -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.dist-info → praetorian_cli-2.2.17.dist-info}/RECORD

@@ -25,7 +25,7 @@ praetorian_cli/scripts/utils.py,sha256=lGCf4trEpsfECa9U42pDJ-f48EimlS-hG6AjnKjNt
 praetorian_cli/scripts/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 praetorian_cli/scripts/commands/nmap-example.py,sha256=varKTkHKG4DAs9Ssf0c6SygP9GfuCG01aFxhfvixLM0,2727
 praetorian_cli/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-praetorian_cli/sdk/chariot.py,sha256=amMgs7FxX9DY0tWNoizg7XfrVYWyLixbqqcaBQsuAKg,14626
+praetorian_cli/sdk/chariot.py,sha256=XRF0iWXw4ylGtlWouc6wr4g9oSR0KfzPfe8f5-3AdVY,14386
 praetorian_cli/sdk/keychain.py,sha256=KCz2mOTv09jgg6mrZQooRg1VrnvF9W0_BjRlEQhbzqU,7468
 praetorian_cli/sdk/mcp_server.py,sha256=8UoTotD4UVl-tp1gqMjkOVpO__KeGCvy7mIpKXVc8Rg,8750
 praetorian_cli/sdk/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -42,12 +42,12 @@ praetorian_cli/sdk/entities/files.py,sha256=Gw9Yt_Cm2xR-Vu8vjaekMHbGldhe930WJjje
 praetorian_cli/sdk/entities/integrations.py,sha256=NVaW_dWbnMkMIs-EYr2W7QAeamPVwLhmM2ppdMJmsK0,3176
 praetorian_cli/sdk/entities/jobs.py,sha256=k4QFw4qR5MdVyMAYstfnRIWr_ZH4q1PwDyUEGC-qBSM,8269
 praetorian_cli/sdk/entities/keys.py,sha256=PgoGa3xyLMzWrIIQ8zgi7bfZiUFFumPtMDo64GjhdjE,6089
-praetorian_cli/sdk/entities/preseeds.py,sha256=SeSY4K6diJMQzsjCBxYK3N9Lz0fUz3B_LMBOAAcBSLg,8890
-praetorian_cli/sdk/entities/risks.py,sha256=_MQlMo0y82DunwI04IjscZU2bNWyu8UM9YTazCGf_Jk,8291
+praetorian_cli/sdk/entities/preseeds.py,sha256=C_FdoUnU5afwEgd7TWpeh4DMMXgEdlaIVTVJGqPGqO8,9298
+praetorian_cli/sdk/entities/risks.py,sha256=D2A-Ikl91QeNiRrWwozLFD61uQ0LJ6oYeCW-8qy5iyo,8273
 praetorian_cli/sdk/entities/scanners.py,sha256=QCr5QlBy4jfBh8HRvZt9CoZTgNqLNnKNrI4sdfJf0jE,423
 praetorian_cli/sdk/entities/schema.py,sha256=CPVws1CdRHyOAI7oT9A20WGOCZozTFqZnfo5ox3v0HQ,807
 praetorian_cli/sdk/entities/search.py,sha256=9vTy9HZY2BTlqf5Zwpdl8HCRIfSEvWDRij2_-rAp2Ng,16938
-praetorian_cli/sdk/entities/seeds.py,sha256=eYDFtfgzIX_o9c5wuPiz9I2xDfTz7k9PZPzL7A7x1CM,5404
+praetorian_cli/sdk/entities/seeds.py,sha256=4YMOOGDJHsSO4O7GMTm8p2nkTOdUTBxXr8V7y6e-Yjc,6010
 praetorian_cli/sdk/entities/settings.py,sha256=F-pRCA6UvbdtnjHOLpEG2lN9ws8dcnBNcep-DFlXeTY,2750
 praetorian_cli/sdk/entities/statistics.py,sha256=gtX-NN7r_RsNjDjlQ-zspmzG_0bzBqBFarCuM4NO-EA,7085
 praetorian_cli/sdk/entities/webhook.py,sha256=7Bqt20GlJFbZTlmQwYTuUadsUQvydym6S4kGn9zYa50,6220
@@ -72,10 +72,10 @@ praetorian_cli/sdk/test/test_file.py,sha256=ZpFBSKbfq9kzfQdy_tmwCH8rwYxW6GB4NlhT
 praetorian_cli/sdk/test/test_job.py,sha256=J7VqnVxRfvbpxNnYN9Rt3LvxmbNvSKwVqsSZxFEW1xc,1916
 praetorian_cli/sdk/test/test_key.py,sha256=yDrR0-JLwMB3eDx-tI-ss8GGbiJaJijE4dAK8-FUuzc,1425
 praetorian_cli/sdk/test/test_mcp.py,sha256=Ltg283j4Q12dcfCgUgDMKk6neRdDaThcndyywZl5XWs,938
-praetorian_cli/sdk/test/test_preseed.py,sha256=VhKSbSB6OmCYnVOrI6RDsYKBs1PekEuv0KOjbTt_k14,1186
+praetorian_cli/sdk/test/test_preseed.py,sha256=ju82GVVcRguNcuGKxmCO6E5KoBXCdDHmPYs_2EEB8cA,1739
 praetorian_cli/sdk/test/test_risk.py,sha256=WuVTPxL9yj84nJx84oXVAlz0KshLxs8GG3tDeSe3QZU,3000
 praetorian_cli/sdk/test/test_search.py,sha256=SB9Tgo_N3CCpfvla898oLB9IZyGK9Dju1r9REK6Wmek,1996
-praetorian_cli/sdk/test/test_seed.py,sha256=WfnEPZwMXHFtt4jyVT-1JitIW1zTrl7rwlX8jXz22vY,1303
+praetorian_cli/sdk/test/test_seed.py,sha256=nGyNsV5bKUGxm5AILY19zp-JNCw6q0IfmefRSLicgsQ,1785
 praetorian_cli/sdk/test/test_setting.py,sha256=hdPQj71rjSYxa-PODG2D-kJd8C9gkAg1jQXnqYU4P6A,1326
 praetorian_cli/sdk/test/test_webhook.py,sha256=FQJY76QQ6Yg2iLCGpxgKiXGI8TtmB4zTpMIM2SpYKCc,2228
 praetorian_cli/sdk/test/test_webpage.py,sha256=jgKrsobD3ONibDIbbOT-yy7V_NmC5-LwEZmEYdYu0LI,1779
@@ -96,9 +96,9 @@ praetorian_cli/ui/aegis/commands/set.py,sha256=WJ0Qt30fBa_hyWEaawyA3h3rSeWHFPbJS
 praetorian_cli/ui/aegis/commands/ssh.py,sha256=KGsNlN0i-Cwp6gWyr-cjML9_L13oE7xFenysF2pC8Rc,3045
 praetorian_cli/ui/conversation/__init__.py,sha256=sNhNN_ZG1Va_7OLTaoXlIFL6ageKHWdufFVYw6F_aV8,90
 praetorian_cli/ui/conversation/textual_chat.py,sha256=gGgEs7HhNAto9rTVrGbz62mG10OqmtArI-aKxFLSfVs,24405
-praetorian_cli-2.2.15.dist-info/licenses/LICENSE,sha256=Zv97QripiVALv-WokW_Elsiz9vtOfbtNt1aLZhhk67I,1067
-praetorian_cli-2.2.15.dist-info/METADATA,sha256=tFrGUx-JASajeDC0bE5j-btCN4Lru3dM93aLyt7hrF0,7752
-praetorian_cli-2.2.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-praetorian_cli-2.2.15.dist-info/entry_points.txt,sha256=uJbDvZdkYaLiCh2DMvXPUGKFm2p5ZfzJCizUK3-PUEE,56
-praetorian_cli-2.2.15.dist-info/top_level.txt,sha256=QbUdRPGEj_TyHO-E7AD5BxFfR8ore37i273jP4Gn43c,15
-praetorian_cli-2.2.15.dist-info/RECORD,,
+praetorian_cli-2.2.17.dist-info/licenses/LICENSE,sha256=Zv97QripiVALv-WokW_Elsiz9vtOfbtNt1aLZhhk67I,1067
+praetorian_cli-2.2.17.dist-info/METADATA,sha256=XOCMfdaoaMsfTLoXCNZCXwx0HBRju2Og-0mX1sm4q_Q,7752
+praetorian_cli-2.2.17.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+praetorian_cli-2.2.17.dist-info/entry_points.txt,sha256=uJbDvZdkYaLiCh2DMvXPUGKFm2p5ZfzJCizUK3-PUEE,56
+praetorian_cli-2.2.17.dist-info/top_level.txt,sha256=QbUdRPGEj_TyHO-E7AD5BxFfR8ore37i273jP4Gn43c,15
+praetorian_cli-2.2.17.dist-info/RECORD,,

{praetorian_cli-2.2.15.dist-info → praetorian_cli-2.2.17.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any