prelude-sdk-beta 1427__py3-none-any.whl → 1430__py3-none-any.whl

prelude_sdk_beta/models/codes.py

@@ -206,6 +206,10 @@ class Control(Enum, metaclass=MissingItem):
                 return k
         return SCMCategory.NONE
 
+    @property
+    def policy_types(self):
+        return [k for k, v in PolicyType.control_mapping().items() if self in v]
+
     @property
     def parent(self):
         match self:
@@ -439,7 +443,9 @@ class PartnerEvents(Enum, metaclass=MissingItem):
             ],
             PartnerEvents.MISSING_ASR_POLICY: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.MISSING_AV_POLICY: [ControlCategory.XDR],
-            PartnerEvents.MISSING_DEVICE_COMPLIANCE_POLICY: [ControlCategory.ASSET_MANAGER],
+            PartnerEvents.MISSING_DEVICE_COMPLIANCE_POLICY: [
+                ControlCategory.ASSET_MANAGER
+            ],
             PartnerEvents.MISSING_DISK_ENCRYPTION: [ControlCategory.ASSET_MANAGER],
             PartnerEvents.MISSING_DISK_ENCRYPTION_POLICY: [
                 ControlCategory.ASSET_MANAGER
@@ -521,6 +527,41 @@ class PolicyType(Enum, metaclass=MissingItem):
     def _missing_(cls, value):
         return PolicyType.INVALID
 
+    @classmethod
+    def control_mapping(cls):
+        return {
+            PolicyType.EDR: ControlCategory.mapping()[ControlCategory.XDR],
+            PolicyType.AV: [Control.DEFENDER],
+            PolicyType.IDENTITY_PASSWORD: ControlCategory.mapping()[
+                ControlCategory.IDENTITY
+            ],
+            PolicyType.EMAIL_ANTIPHISH: ControlCategory.mapping()[
+                ControlCategory.EMAIL
+            ],
+            PolicyType.EMAIL_OUTBOUND: ControlCategory.mapping()[ControlCategory.EMAIL],
+            PolicyType.EMAIL_CONTENT: ControlCategory.mapping()[ControlCategory.EMAIL],
+            PolicyType.EMAIL_MALWARE: ControlCategory.mapping()[ControlCategory.EMAIL],
+            PolicyType.EMAIL_ATTACHMENT: ControlCategory.mapping()[
+                ControlCategory.EMAIL
+            ],
+            PolicyType.EMAIL_LINKS: ControlCategory.mapping()[ControlCategory.EMAIL],
+            PolicyType.EMAIL_DKIM: ControlCategory.mapping()[ControlCategory.EMAIL],
+            PolicyType.DEVICE_COMPLIANCE: [Control.INTUNE],
+            PolicyType.IDENTITY_MFA: ControlCategory.mapping()[
+                ControlCategory.IDENTITY
+            ],
+            PolicyType.HOST_FIREWALL: [Control.INTUNE],
+            PolicyType.NETWORK_FIREWALL: ControlCategory.mapping()[
+                ControlCategory.NETWORK
+            ],
+            PolicyType.INTEL_BELOW_OS: [Control.INTEL_INTUNE],
+            PolicyType.INTEL_OS: [Control.INTEL_INTUNE],
+            PolicyType.INTEL_TDT: [Control.INTEL_INTUNE],
+            PolicyType.INTEL_CHIP: [Control.INTEL_INTUNE],
+            PolicyType.DISK_ENCRYPTION: [Control.INTUNE],
+            PolicyType.ASR: [Control.INTUNE],
+        }
+
 
 class Platform(Enum, metaclass=MissingItem):
     INVALID = 0

{prelude_sdk_beta-1427.dist-info → prelude_sdk_beta-1430.dist-info}/METADATA

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

{prelude_sdk_beta-1427.dist-info → prelude_sdk_beta-1430.dist-info}/RECORD

@@ -12,9 +12,9 @@ prelude_sdk_beta/controllers/probe_controller.py,sha256=d2Aa74Css1uMNgBKGWqavbkE
 prelude_sdk_beta/controllers/scm_controller.py,sha256=XRGlteBV-Ph5FAJHQaCeEUKD_sBRNDCSUKg0xllbx6k,18315
 prelude_sdk_beta/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prelude_sdk_beta/models/account.py,sha256=dRKhX6_mrqDTyqyfIsvHFVavcedgZ5hZAIDYVF7YlZI,8962
-prelude_sdk_beta/models/codes.py,sha256=yd3GIjCwgqs0bwxZOSNyATvZMV3DadyiOlkezGTZuhQ,15319
-prelude_sdk_beta-1427.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
-prelude_sdk_beta-1427.dist-info/METADATA,sha256=f3ujNl87SP2w8O85KyqCngRtxKRJk1vO815gCmGpQEQ,1190
-prelude_sdk_beta-1427.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-prelude_sdk_beta-1427.dist-info/top_level.txt,sha256=pqXTtEd5ElvJKoO6HAz232H9FW5j6X7gW4kEEakfSFM,17
-prelude_sdk_beta-1427.dist-info/RECORD,,
+prelude_sdk_beta/models/codes.py,sha256=OXoOHxZMzv5kpi8LkwRHu3LAH75E0rM4hBQLnIsdODE,17173
+prelude_sdk_beta-1430.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
+prelude_sdk_beta-1430.dist-info/METADATA,sha256=uKsHsA9_rCLipEy1QPKXc_XU7gQ1NeOfaQ1lekruNBA,1190
+prelude_sdk_beta-1430.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+prelude_sdk_beta-1430.dist-info/top_level.txt,sha256=pqXTtEd5ElvJKoO6HAz232H9FW5j6X7gW4kEEakfSFM,17
+prelude_sdk_beta-1430.dist-info/RECORD,,