prelude-sdk 2.6.32__py3-none-any.whl → 2.6.34__py3-none-any.whl

prelude_sdk/controllers/scm_controller.py

@@ -593,3 +593,72 @@ class ScmController(HttpController):
                 ],
             )
         return history
+
+    @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,
+        )
+        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,
+        )
+        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,
+        )
+        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,
+        )
+        return res.json()
+
+    @verify_credentials
+    def get_chart_data(
+        self,
+        scm_category: SCMCategory,
+        sort_by: str,
+        group_by: str,
+        group_limit: int,
+        scopes: str = None,
+        filter: str = None,
+    ):
+        """Get SCM chart data"""
+        params = {
+            "scm_category": scm_category.name,
+            "sort_by": sort_by,
+            "group_by": group_by,
+            "group_limit": group_limit,
+            "scopes": scopes,
+        }
+        if filter:
+            params["$filter"] = filter
+        res = self.get(
+            f"{self.account.hq}/scm/chart_data",
+            headers=self.account.headers,
+            params=params,
+            timeout=30,
+        )
+        return res.json()

prelude_sdk/models/codes.py

@@ -236,6 +236,68 @@ class Control(Enum, metaclass=MissingItem):
             case _:
                 return []
 
+    @property
+    def display_name(self):
+        match self:
+            case Control.CROWDSTRIKE:
+                return "CrowdStrike"
+            case Control.DEFENDER:
+                return "Microsoft Defender"
+            case Control.SPLUNK:
+                return "Splunk"
+            case Control.SENTINELONE:
+                return "SentinelOne"
+            case Control.VECTR:
+                return "VECTR"
+            case Control.S3:
+                return "Amazon S3"
+            case Control.INTUNE:
+                return "Microsoft Intune"
+            case Control.SERVICENOW:
+                return "ServiceNow"
+            case Control.OKTA:
+                return "Okta"
+            case Control.M365:
+                return "Microsoft 365"
+            case Control.ENTRA:
+                return "Microsoft Entra ID"
+            case Control.JAMF:
+                return "Jamf"
+            case Control.GMAIL:
+                return "Gmail"
+            case Control.GOOGLE_IDENTITY:
+                return "Google Cloud Identity Platform"
+            case Control.DEFENDER_DISCOVERY:
+                return "Microsoft Defender Discovery"
+            case Control.TENABLE:
+                return "Tenable"
+            case Control.EC2:
+                return "Amazon EC2"
+            case Control.AWS_SSM:
+                return "Amazon SSM"
+            case Control.AZURE_VM:
+                return "Azure VM"
+            case Control.GITHUB:
+                return "GitHub"
+            case Control.TENABLE_DISCOVERY:
+                return "Tenable Discovery"
+            case Control.QUALYS:
+                return "Qualys"
+            case Control.QUALYS_DISCOVERY:
+                return "Qualys Discovery"
+            case Control.RAPID7:
+                return "Rapid7"
+            case Control.RAPID7_DISCOVERY:
+                return "Rapid7 Discovery"
+            case Control.INTEL_INTUNE:
+                return "Intel"
+            case Control.CISCO_MERAKI:
+                return "Cisco Meraki"
+            case Control.CISCO_MERAKI_IDENTITY:
+                return "Cisco Meraki Identity"
+            case _:
+                return "Unknown Control"
+
 
 class ControlCategory(Enum, metaclass=MissingItem):
     INVALID = -1
@@ -308,6 +370,34 @@ class ControlCategory(Enum, metaclass=MissingItem):
             ],
         }
 
+    @property
+    def display_name(self):
+        match self:
+            case ControlCategory.CLOUD:
+                return "Cloud"
+            case ControlCategory.EMAIL:
+                return "Email"
+            case ControlCategory.IDENTITY:
+                return "Identity Provider"
+            case ControlCategory.NETWORK:
+                return "Network"
+            case ControlCategory.XDR:
+                return "EDR"
+            case ControlCategory.ASSET_MANAGER:
+                return "Endpoint Management"
+            case ControlCategory.DISCOVERED_DEVICES:
+                return "Discovered Devices"
+            case ControlCategory.VULN_MANAGER:
+                return "Vulnerability Management"
+            case ControlCategory.SIEM:
+                return "SIEM"
+            case ControlCategory.PRIVATE_REPO:
+                return "Private Repository"
+            case ControlCategory.HARDWARE:
+                return "Client Hardware Security"
+            case _:
+                return "Unknown Control Category"
+
 
 class SCMCategory(Enum, metaclass=MissingItem):
     INVALID = -1

{prelude_sdk-2.6.32.dist-info → prelude_sdk-2.6.34.dist-info}/METADATA

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

{prelude_sdk-2.6.32.dist-info → prelude_sdk-2.6.34.dist-info}/RECORD

@@ -9,12 +9,12 @@ prelude_sdk/controllers/iam_controller.py,sha256=Ya5jg7F6t3cAYzanTlBQfthkVV081aV
 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=HAtamZUQy5fvNxpN8wV0EsmukAN5ap3XyyFn4NTkRmg,18300
+prelude_sdk/controllers/scm_controller.py,sha256=tivAboJLmS0ztdZnyMD70fsOC5mRxz4FfqXf7iHQydw,20231
 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=h5X7X-Zqh_W6gQPVBz7uu627DTytdKuzxTMEgHXGY2c,15442
-prelude_sdk-2.6.32.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
-prelude_sdk-2.6.32.dist-info/METADATA,sha256=O0P3LSJVnONJQATdHgeSPzVjnAVIID8QnVQ5ofz9mdA,1187
-prelude_sdk-2.6.32.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-prelude_sdk-2.6.32.dist-info/top_level.txt,sha256=6O7C8nl-yK7FsVpsPaka_GV8PYy2uvAJtus8Tlzw4dE,12
-prelude_sdk-2.6.32.dist-info/RECORD,,
+prelude_sdk/models/codes.py,sha256=owX-h3DNEU9buAQ8u4_GhgINOIgPNAG01o8E7eyZLuU,18672
+prelude_sdk-2.6.34.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
+prelude_sdk-2.6.34.dist-info/METADATA,sha256=UGE_rQSeBIp5_WA0pCyul2KPpYudr2OkDPtNh9cOLoI,1187
+prelude_sdk-2.6.34.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+prelude_sdk-2.6.34.dist-info/top_level.txt,sha256=6O7C8nl-yK7FsVpsPaka_GV8PYy2uvAJtus8Tlzw4dE,12
+prelude_sdk-2.6.34.dist-info/RECORD,,