praetorian-cli 2.2.5__tar.gz → 2.2.7__tar.gz

{praetorian_cli-2.2.5/praetorian_cli.egg-info → praetorian_cli-2.2.7}/PKG-INFO

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

{praetorian_cli-2.2.5 → praetorian_cli-2.2.7}/praetorian_cli/sdk/entities/risks.py

@@ -1,5 +1,5 @@
 from praetorian_cli.sdk.model.globals import Kind
-from praetorian_cli.sdk.model.query import Relationship, Node, Query, risk_of_key, ASSET_NODE, ATTRIBUTE_NODE
+from praetorian_cli.sdk.model.query import Relationship, Node, Query, risk_of_key, ASSET_NODE, ATTRIBUTE_NODE, Filter
 
 
 class Risks:
@@ -85,12 +85,12 @@ class Risks:
 
         return self.api.delete_by_key('risk', key, body)
 
-    def list(self, prefix_filter='', offset=None, pages=100000) -> tuple:
+    def list(self, contains_filter='', offset=None, pages=100000) -> tuple:
         """
         List risks with optional filtering and pagination.
 
-        :param prefix_filter: Prefix filter to apply after the "#risk#" portion of the key. Risk keys follow format '#risk#{asset_dns}#{risk_name}'
-        :type prefix_filter: str
+        :param contains_filter: Filter to apply to the risk key. Ensure the risk's key contains the filter.
+        :type contains_filter: str
         :param offset: The offset of the page you want to retrieve results. If not supplied, retrieves from the first page
         :type offset: str or None
         :param pages: The number of pages of results to retrieve. <mcp>Start with one page of results unless specifically requested.</mcp>
@@ -98,7 +98,12 @@ class Risks:
         :return: A tuple containing (list of matching risks, next page offset)
         :rtype: tuple
         """
-        return self.api.search.by_key_prefix(f'#risk#{prefix_filter}', offset, pages)
+        filters = []
+        if contains_filter:
+            contains_filter_filter = Filter(field=Filter.Field.KEY, operator=Filter.Operator.CONTAINS, value=contains_filter)
+            filters.append(contains_filter_filter)
+        query = Query(Node([Node.Label.RISK], filters=filters), page=int(offset))
+        return self.api.search.by_query(query, pages)
 
     def attributes(self, key):
         """

{praetorian_cli-2.2.5 → praetorian_cli-2.2.7/praetorian_cli.egg-info}/PKG-INFO

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

{praetorian_cli-2.2.5 → praetorian_cli-2.2.7}/setup.cfg

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