prelude-sdk 2.6.39__py3-none-any.whl → 2.6.40__py3-none-any.whl

prelude_sdk/controllers/scm_controller.py

@@ -21,12 +21,8 @@ class ScmController(HttpController):
     def endpoints(self, filter: str = None, orderby: str = None, top: int = None):
         """List endpoints with SCM analysis"""
         params = {"$filter": filter, "$orderby": orderby, "$top": top}
-        res = self.get(
-            f"{self.account.hq}/scm/endpoints",
-            headers=self.account.headers,
-            params=params,
-            timeout=30,
-        )
+
+        res = self.get(f"{self.account.hq}/scm/endpoints", params=params, timeout=30)
         data = res.json()
         if self.account.resolve_enums:
             self.resolve_enums(
@@ -44,12 +40,8 @@ class ScmController(HttpController):
     def inboxes(self, filter: str = None, orderby: str = None, top: int = None):
         """List inboxes with SCM analysis"""
         params = {"$filter": filter, "$orderby": orderby, "$top": top}
-        res = self.get(
-            f"{self.account.hq}/scm/inboxes",
-            headers=self.account.headers,
-            params=params,
-            timeout=30,
-        )
+
+        res = self.get(f"{self.account.hq}/scm/inboxes", params=params, timeout=30)
         data = res.json()
         if self.account.resolve_enums:
             self.resolve_enums(
@@ -67,11 +59,9 @@ class ScmController(HttpController):
     def network_devices(self, filter: str = None, orderby: str = None, top: int = None):
         """List network_devices with SCM analysis"""
         params = {"$filter": filter, "$orderby": orderby, "$top": top}
+
         res = self.get(
-            f"{self.account.hq}/scm/network_devices",
-            headers=self.account.headers,
-            params=params,
-            timeout=30,
+            f"{self.account.hq}/scm/network_devices", params=params, timeout=30
         )
         data = res.json()
         if self.account.resolve_enums:
@@ -90,12 +80,8 @@ class ScmController(HttpController):
     def users(self, filter: str = None, orderby: str = None, top: int = None):
         """List users with SCM analysis"""
         params = {"$filter": filter, "$orderby": orderby, "$top": top}
-        res = self.get(
-            f"{self.account.hq}/scm/users",
-            headers=self.account.headers,
-            params=params,
-            timeout=30,
-        )
+
+        res = self.get(f"{self.account.hq}/scm/users", params=params, timeout=30)
         data = res.json()
         if self.account.resolve_enums:
             self.resolve_enums(
@@ -115,7 +101,6 @@ class ScmController(HttpController):
         res = self.get(
             f"{self.account.hq}/scm/technique_summary",
             params=dict(techniques=techniques),
-            headers=self.account.headers,
             timeout=30,
         )
         return res.json()
@@ -136,11 +121,9 @@ class ScmController(HttpController):
         )
         if techniques:
             params["techniques"] = techniques
+
         res = self.get(
-            f"{self.account.hq}/scm/evaluation_summary",
-            params=params,
-            headers=self.account.headers,
-            timeout=30,
+            f"{self.account.hq}/scm/evaluation_summary", params=params, timeout=30
         )
         data = res.json()
         if self.account.resolve_enums:
@@ -164,10 +147,10 @@ class ScmController(HttpController):
             params["policy_types"] = policy_types
         if techniques:
             params["techniques"] = techniques
+
         res = self.get(
             f"{self.account.hq}/scm/evaluations/{partner.name}/{instance_id}",
             params=params,
-            headers=self.account.headers,
             timeout=30,
         )
         data = res.json()
@@ -179,9 +162,7 @@ class ScmController(HttpController):
     def update_evaluation(self, partner: Control, instance_id: str):
         """Update policy evaluations for given partner"""
         res = self.post(
-            f"{self.account.hq}/scm/evaluations/{partner.name}/{instance_id}",
-            headers=self.account.headers,
-            timeout=60,
+            f"{self.account.hq}/scm/evaluations/{partner.name}/{instance_id}"
         )
         return res.json()
 
@@ -189,12 +170,8 @@ class ScmController(HttpController):
     def list_partner_groups(self, filter: str = None, orderby: str = None):
         """List groups"""
         params = {"$filter": filter, "$orderby": orderby}
-        res = self.get(
-            f"{self.account.hq}/scm/groups",
-            headers=self.account.headers,
-            params=params,
-            timeout=10,
-        )
+
+        res = self.get(f"{self.account.hq}/scm/groups", params=params)
         groups = res.json()
         if self.account.resolve_enums:
             self.resolve_enums(groups, [(Control, "control")])
@@ -206,22 +183,16 @@ class ScmController(HttpController):
     ):
         """Update groups"""
         body = dict(group_ids=group_ids)
+
         res = self.post(
-            f"{self.account.hq}/scm/groups/{partner.name}/{instance_id}",
-            headers=self.account.headers,
-            json=body,
-            timeout=10,
+            f"{self.account.hq}/scm/groups/{partner.name}/{instance_id}", json=body
         )
         return res.json()
 
     @verify_credentials
     def list_object_exceptions(self):
         """List object exceptions"""
-        res = self.get(
-            f"{self.account.hq}/scm/exceptions/objects",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.get(f"{self.account.hq}/scm/exceptions/objects")
         exceptions = res.json()
         if self.account.resolve_enums:
             self.resolve_enums(exceptions, [(ControlCategory, "category")])
@@ -229,60 +200,50 @@ class ScmController(HttpController):
 
     @verify_credentials
     def create_object_exception(
-        self, category: ControlCategory, filter, name=None, expires: str = None
+        self, category: ControlCategory, filter, comment=None, name=None, expires: str = None
     ):
         """Create an object exception"""
         body = dict(category=category.name, filter=filter)
+        if comment:
+            body["comment"] = comment
         if name:
             body["name"] = name
         if expires:
             body["expires"] = expires
-        res = self.post(
-            f"{self.account.hq}/scm/exceptions/objects",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
+
+        res = self.post(f"{self.account.hq}/scm/exceptions/objects", json=body)
         return res.json()
 
     @verify_credentials
     def update_object_exception(
-        self, exception_id, expires=default, filter=None, name=None
+        self, exception_id, comment=None, expires=default, filter=None, name=None
     ):
         """Update an object exception"""
         body = dict()
+        if comment:
+            body["comment"] = comment
         if expires != self.default:
             body["expires"] = expires
         if filter:
             body["filter"] = filter
         if name:
             body["name"] = name
+
         res = self.post(
-            f"{self.account.hq}/scm/exceptions/objects/{exception_id}",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
+            f"{self.account.hq}/scm/exceptions/objects/{exception_id}", json=body
         )
         return res.json()
 
     @verify_credentials
     def delete_object_exception(self, exception_id):
         """Delete an object exception"""
-        res = self.delete(
-            f"{self.account.hq}/scm/exceptions/objects/{exception_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.delete(f"{self.account.hq}/scm/exceptions/objects/{exception_id}")
         return res.json()
 
     @verify_credentials
     def list_policy_exceptions(self):
         """List policy exceptions"""
-        res = self.get(
-            f"{self.account.hq}/scm/exceptions/policies",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.get(f"{self.account.hq}/scm/exceptions/policies")
         exceptions = res.json()
         if self.account.resolve_enums:
             self.resolve_enums(
@@ -292,22 +253,19 @@ class ScmController(HttpController):
 
     @verify_credentials
     def create_policy_exception(
-        self, partner: Control, instance_id: str, policy_id, setting_names, expires=None
+        self, partner: Control, instance_id: str, policy_id, setting_names, comment=None, expires=None
     ):
         """Create policy exceptions"""
         body = dict(
             control=partner.name,
+            comment=comment,
             expires=expires,
             instance_id=instance_id,
             policy_id=policy_id,
             setting_names=setting_names,
         )
-        res = self.post(
-            f"{self.account.hq}/scm/exceptions/policies",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
+
+        res = self.post(f"{self.account.hq}/scm/exceptions/policies", json=body)
         return res.json()
 
     @verify_credentials
@@ -316,43 +274,34 @@ class ScmController(HttpController):
         partner: Control,
         instance_id: str,
         policy_id,
+        comment=None,
         expires=default,
         setting_names=None,
     ):
         """Update policy exceptions"""
         body = dict(control=partner.name, instance_id=instance_id, policy_id=policy_id)
+        if comment:
+            body["comment"] = comment
         if expires != self.default:
             body["expires"] = expires
         if setting_names:
             body["setting_names"] = setting_names
-        res = self.put(
-            f"{self.account.hq}/scm/exceptions/policies",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
+
+        res = self.put(f"{self.account.hq}/scm/exceptions/policies", json=body)
         return res.json()
 
     @verify_credentials
     def delete_policy_exception(self, instance_id: str, policy_id: str):
         """Delete policy exceptions"""
         body = dict(instance_id=instance_id, policy_id=policy_id)
-        res = self.delete(
-            f"{self.account.hq}/scm/exceptions/policies",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
+
+        res = self.delete(f"{self.account.hq}/scm/exceptions/policies", json=body)
         return res.json()
 
     @verify_credentials
     def list_views(self):
         """List views"""
-        res = self.get(
-            f"{self.account.hq}/scm/views",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.get(f"{self.account.hq}/scm/views")
         views = res.json()
         if self.account.resolve_enums:
             self.resolve_enums(views, [(ControlCategory, "category")])
@@ -362,12 +311,8 @@ class ScmController(HttpController):
     def create_view(self, category: ControlCategory, filter: str, name: str):
         """Create a view"""
         body = dict(category=category.name, filter=filter, name=name)
-        res = self.post(
-            f"{self.account.hq}/scm/views",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
+
+        res = self.post(f"{self.account.hq}/scm/views", json=body)
         return res.json()
 
     @verify_credentials
@@ -382,22 +327,14 @@ class ScmController(HttpController):
             body["filter"] = filter
         if name:
             body["name"] = name
-        res = self.post(
-            f"{self.account.hq}/scm/views/{view_id}",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
+
+        res = self.post(f"{self.account.hq}/scm/views/{view_id}", json=body)
         return res.json()
 
     @verify_credentials
     def delete_view(self, view_id):
         """Delete a view"""
-        res = self.delete(
-            f"{self.account.hq}/scm/views/{view_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.delete(f"{self.account.hq}/scm/views/{view_id}")
         return res.json()
 
     @verify_credentials
@@ -429,46 +366,32 @@ class ScmController(HttpController):
         if techniques:
             body["techniques"] = techniques
 
-        res = self.post(
-            f"{self.account.hq}/scm/threats",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.post(f"{self.account.hq}/scm/threats", json=body)
         return res.json()
 
     @verify_credentials
     def delete_threat(self, id):
         """Delete an existing scm threat"""
-        res = self.delete(
-            f"{self.account.hq}/scm/threats/{id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.delete(f"{self.account.hq}/scm/threats/{id}")
         return res.json()
 
     @verify_credentials
     def get_threat(self, id):
         """Get specific scm threat"""
-        res = self.get(
-            f"{self.account.hq}/scm/threats/{id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.get(f"{self.account.hq}/scm/threats/{id}")
         return res.json()
 
     @verify_credentials
     def list_threats(self):
         """List all scm threats"""
-        res = self.get(
-            f"{self.account.hq}/scm/threats", headers=self.account.headers, timeout=10
-        )
+        res = self.get(f"{self.account.hq}/scm/threats")
         return res.json()
 
     @verify_credentials
     def parse_threat_intel(self, file: str):
         with open(file, "rb") as f:
             body = f.read()
+
         res = self.post(
             f"{self.account.hq}/scm/threat-intel",
             data=body,
@@ -480,9 +403,9 @@ class ScmController(HttpController):
     @verify_credentials
     def parse_from_partner_advisory(self, partner: Control, advisory_id: str):
         params = dict(advisory_id=advisory_id)
+
         res = self.post(
             f"{self.account.hq}/scm/partner-advisories/{partner.name}",
-            headers=self.account.headers,
             json=params,
             timeout=30,
         )
@@ -490,11 +413,7 @@ class ScmController(HttpController):
 
     @verify_credentials
     def list_notifications(self):
-        res = self.get(
-            f"{self.account.hq}/scm/notifications",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.get(f"{self.account.hq}/scm/notifications")
         notifications = res.json()
         if self.account.resolve_enums:
             self.resolve_enums(
@@ -509,11 +428,7 @@ class ScmController(HttpController):
 
     @verify_credentials
     def delete_notification(self, notification_id: str):
-        res = self.delete(
-            f"{self.account.hq}/scm/notifications/{notification_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.delete(f"{self.account.hq}/scm/notifications/{notification_id}")
         return res.json()
 
     @verify_credentials
@@ -551,22 +466,14 @@ class ScmController(HttpController):
             body["slack"] = dict(hook_urls=slack_urls, message=message)
         if teams_urls:
             body["teams"] = dict(hook_urls=teams_urls, message=message)
-        res = self.put(
-            f"{self.account.hq}/scm/notifications",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
+
+        res = self.put(f"{self.account.hq}/scm/notifications", json=body)
         return res.json()
 
     @verify_credentials
     def list_notations(self):
         """List notations"""
-        res = self.get(
-            f"{self.account.hq}/scm/notations",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.get(f"{self.account.hq}/scm/notations")
         notations = res.json()
         if self.account.resolve_enums:
             self.resolve_enums(notations, [(NotationType, "event")])
@@ -578,12 +485,8 @@ class ScmController(HttpController):
     ):
         """List history"""
         params = {"start_date": start_date, "end_date": end_date, "$filter": filter}
-        res = self.get(
-            f"{self.account.hq}/scm/history",
-            headers=self.account.headers,
-            params=params,
-            timeout=10,
-        )
+
+        res = self.get(f"{self.account.hq}/scm/history", params=params)
         history = res.json()
         if self.account.resolve_enums:
             self.resolve_enums(
@@ -599,42 +502,27 @@ class ScmController(HttpController):
     @verify_credentials
     def get_report(self, report_id: str):
         """Get SCM report by ID"""
-        res = self.get(
-            f"{self.account.hq}/scm/reports/{report_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.get(f"{self.account.hq}/scm/reports/{report_id}")
         return res.json()
 
     @verify_credentials
     def list_reports(self):
         """List SCM reports"""
-        res = self.get(
-            f"{self.account.hq}/scm/reports",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.get(f"{self.account.hq}/scm/reports")
         return res.json()
 
     @verify_credentials
     def delete_report(self, report_id: str):
         """Delete SCM report by ID"""
-        res = self.delete(
-            f"{self.account.hq}/scm/reports/{report_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
+        res = self.delete(f"{self.account.hq}/scm/reports/{report_id}")
         return res.json()
 
     @verify_credentials
     def put_report(self, report_data: dict, report_id: str = None):
         """Put SCM report by ID"""
-        res = self.put(
-            f"{self.account.hq}/scm/reports",
-            headers=self.account.headers,
-            json=dict(report=report_data, id=report_id),
-            timeout=10,
-        )
+        body = dict(report=report_data, id=report_id)
+
+        res = self.put(f"{self.account.hq}/scm/reports", json=body)
         return res.json()
 
     @verify_credentials
@@ -659,10 +547,6 @@ class ScmController(HttpController):
         }
         if odata_filter:
             body["$filter"] = odata_filter
-        res = self.post(
-            f"{self.account.hq}/scm/reports/data",
-            headers=self.account.headers,
-            json=body,
-            timeout=30,
-        )
+
+        res = self.post(f"{self.account.hq}/scm/reports/data", json=body, timeout=30)
         return res.json()

prelude_sdk/models/codes.py

@@ -53,6 +53,7 @@ class Permission(Enum, metaclass=MissingItem):
     BUILD = 2
     SERVICE = 3
     SUPPORT = 5
+    SCHEDULER = 6
 
     @classmethod
     def _missing_(cls, value):

{prelude_sdk-2.6.39.dist-info → prelude_sdk-2.6.40.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prelude-sdk
-Version: 2.6.39
+Version: 2.6.40
 Summary: For interacting with the Prelude API
 Home-page: https://github.com/preludeorg
 Author: Prelude Research

prelude_sdk-2.6.40.dist-info/RECORD

@@ -0,0 +1,20 @@
+prelude_sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+prelude_sdk/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+prelude_sdk/controllers/build_controller.py,sha256=CGVhQG0Pjp8FTkWnYvT0G9TmzurgODyDshhKg-WiY9U,7540
+prelude_sdk/controllers/detect_controller.py,sha256=SQO2cHHQFeqnFpdk68x_ctazKWYj5ttnzxUFX9af-bM,6409
+prelude_sdk/controllers/export_controller.py,sha256=rITSEId7OE3xik5WK2W7BQlPyzoDt8zZ3fKY1vMaLys,741
+prelude_sdk/controllers/generate_controller.py,sha256=LJm0XAsz_JqGQSNWxCKk3NXv_HnCWutb19-CxSh4bng,1175
+prelude_sdk/controllers/http_controller.py,sha256=TN9FrcTrY31znIumFu73oOEuyd5E5I-oD42ewpLQv84,3840
+prelude_sdk/controllers/iam_controller.py,sha256=PNifSPdrrOvBx_CZ65vL-9dUfM2Pqk-JGj0ixPNRuTE,6673
+prelude_sdk/controllers/jobs_controller.py,sha256=2INTCETApEGJTD6wJuJcG6-WO71Qm9_ofa-rn32bckw,1021
+prelude_sdk/controllers/partner_controller.py,sha256=z4Z36vHctD5k3AOWi9xQOKR2on3RUyciQR4oyjQrHq0,4185
+prelude_sdk/controllers/probe_controller.py,sha256=Skdj3joA7C_H72LyIiAQYIX9D9-KgxzMdbYOeEA9G7I,372
+prelude_sdk/controllers/scm_controller.py,sha256=_uRiJgdXgD7Px5dwAXBspCRPjwznzAGuPwlORS92uy0,17541
+prelude_sdk/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+prelude_sdk/models/account.py,sha256=dRKhX6_mrqDTyqyfIsvHFVavcedgZ5hZAIDYVF7YlZI,8962
+prelude_sdk/models/codes.py,sha256=UcaPL1aQJJN3ehHIL4AvIlYtni5ssINtusixp76f0CI,19345
+prelude_sdk-2.6.40.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
+prelude_sdk-2.6.40.dist-info/METADATA,sha256=pnWpI1dapymSAZVji8OG_ynjZEqU8K1o5jmiV5Fc8W8,1187
+prelude_sdk-2.6.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+prelude_sdk-2.6.40.dist-info/top_level.txt,sha256=6O7C8nl-yK7FsVpsPaka_GV8PYy2uvAJtus8Tlzw4dE,12
+prelude_sdk-2.6.40.dist-info/RECORD,,

prelude_sdk-2.6.39.dist-info/RECORD

@@ -1,20 +0,0 @@
-prelude_sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prelude_sdk/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prelude_sdk/controllers/build_controller.py,sha256=PUBBPzMCq78tF_0DUiryZwDCW4lPlVAAZcAcJtheWTs,9084
-prelude_sdk/controllers/detect_controller.py,sha256=zQvdhAi0KoFHUrxYNn8Moi0t_zkFjCD8sp4h-BnDuS8,8106
-prelude_sdk/controllers/export_controller.py,sha256=lIRmO7b2sMLyA8WAsGFqtXuCp79r25zgopJGM5KnxvA,847
-prelude_sdk/controllers/generate_controller.py,sha256=JUYlBdMh8alSMvin32qb3hB1JNGbT-BarEgxhMhg9PI,1304
-prelude_sdk/controllers/http_controller.py,sha256=zomZo8OCWcJ-8zQIlBtukXbnJeqM9PkQn303APojLSA,3160
-prelude_sdk/controllers/iam_controller.py,sha256=QLPo_5AqFU2a_gA9KBFS51xtbn_WolyaXQXEpvF8RKs,8370
-prelude_sdk/controllers/jobs_controller.py,sha256=TECogrNcbW4uY5uDIDndAwxGHSA7RLnFZWmeXhvAHSM,1150
-prelude_sdk/controllers/partner_controller.py,sha256=63P7d7fwVc1iWIUMxQv5PQOQ-e-VIpDql9HYNtLOZNo,4742
-prelude_sdk/controllers/probe_controller.py,sha256=rwBnqkFJa1KuVwZMluM-pfo8bhQFB6JKSRSeK7s-kew,406
-prelude_sdk/controllers/scm_controller.py,sha256=STGwaE8uD3GGbPL6PU0a66cpasrXc3Ige_Td_lTAf1w,20406
-prelude_sdk/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prelude_sdk/models/account.py,sha256=dRKhX6_mrqDTyqyfIsvHFVavcedgZ5hZAIDYVF7YlZI,8962
-prelude_sdk/models/codes.py,sha256=WKb8dKElp-X2QqVDATowngWNSFSmpGcFVooyXzGaulE,19327
-prelude_sdk-2.6.39.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
-prelude_sdk-2.6.39.dist-info/METADATA,sha256=2RRk7ugWtM122Amx71S6AbfWEg51Hg-CWaTuAyYCeag,1187
-prelude_sdk-2.6.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-prelude_sdk-2.6.39.dist-info/top_level.txt,sha256=6O7C8nl-yK7FsVpsPaka_GV8PYy2uvAJtus8Tlzw4dE,12
-prelude_sdk-2.6.39.dist-info/RECORD,,