prelude-sdk 2.6.42__py3-none-any.whl → 2.6.43__py3-none-any.whl

prelude_sdk/controllers/iam_controller.py

@@ -1,3 +1,7 @@
+import hashlib
+import json
+import os
+
 from prelude_sdk.controllers.http_controller import HttpController
 from prelude_sdk.models.account import verify_credentials
 from prelude_sdk.models.codes import Control, Mode, Permission
@@ -142,10 +146,22 @@ class IAMAccountController(HttpController):
     def sign_up(self, company, email, name):
         """(NOT AVAIABLE IN PRODUCTION) Create a new user and account"""
         body = dict(company=company, email=email, name=name)
+        headers = dict(_product="py-sdk")
+        if token := os.getenv("SDKTOKEN"):
+            sig = hashlib.sha256(
+                f"{token}{json.dumps(body)}".encode("utf-8")
+            ).hexdigest()
+            headers["x-sdk-sig"] = sig
 
         res = self._session.post(
-            f"{self.account.hq}/iam/new_user_and_account", json=body, timeout=20
+            f"{self.account.hq}/iam/new_user_and_account",
+            headers=headers,
+            json=body,
+            timeout=20,
         )
+        if not res.ok:
+            raise Exception(res.text)
+
         data = res.json()
         if self.account.profile:
             self.account.keychain.configure_keychain(

prelude_sdk/controllers/probe_controller.py

@@ -8,5 +8,8 @@ class ProbeController(HttpController):
 
     def download(self, name: str, dos: str):
         """Download a probe executable"""
-        res = self.get(f"{self.account.hq}/download/{name}", headers=dict(dos=dos))
+        res = self.get(
+            f"{self.account.hq}/download/{name}",
+            headers=dict(dos=dos, _product="py-sdk"),
+        )
         return res.text

prelude_sdk/models/codes.py

@@ -191,6 +191,7 @@ class Control(Enum, metaclass=MissingItem):
     CROWDSTRIKE_VULN = 31
     DEFENDER_VULN = 32
     NETSKOPE = 33
+    CUSTOM = 34
 
     @classmethod
     def _missing_(cls, value):
@@ -329,6 +330,13 @@ class ControlCategory(Enum, metaclass=MissingItem):
     def _missing_(cls, value):
         return ControlCategory.INVALID
 
+    @property
+    def scm_category(self):
+        for k, v in SCMCategory.category_mapping().items():
+            if self in v:
+                return k
+        return SCMCategory.NONE
+
     @classmethod
     def mapping(cls):
         return {

{prelude_sdk-2.6.42.dist-info → prelude_sdk-2.6.43.dist-info}/METADATA

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

{prelude_sdk-2.6.42.dist-info → prelude_sdk-2.6.43.dist-info}/RECORD

@@ -5,16 +5,16 @@ prelude_sdk/controllers/detect_controller.py,sha256=SQO2cHHQFeqnFpdk68x_ctazKWYj
 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/iam_controller.py,sha256=mlvv-Yhp4yKPLp9GDsdz5Wu-xfWHkmk1wqZIIpUulzg,7072
 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/probe_controller.py,sha256=jlNpXOki4gXuBIq5qWFp55f4RZ98743HJn-fAZRJ_zA,426
 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=zdusWseiIRupooHWboIFigNASYSwSKxw5Y-YI7n8tds,19777
-prelude_sdk-2.6.42.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
-prelude_sdk-2.6.42.dist-info/METADATA,sha256=W9ormB9l9YOckcM6Vmh2weJ8K1ZGPzbvLou4zQKJ9W8,1187
-prelude_sdk-2.6.42.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-prelude_sdk-2.6.42.dist-info/top_level.txt,sha256=6O7C8nl-yK7FsVpsPaka_GV8PYy2uvAJtus8Tlzw4dE,12
-prelude_sdk-2.6.42.dist-info/RECORD,,
+prelude_sdk/models/codes.py,sha256=XTYBnp9OJIv6JQbj8ydzuHEZeTKvPCGr8w_96GIQr8w,19979
+prelude_sdk-2.6.43.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
+prelude_sdk-2.6.43.dist-info/METADATA,sha256=wXPOzyWVUR_FMYF9PD6Yb_5IfF7MDXXQm5WO10-dvYo,1187
+prelude_sdk-2.6.43.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+prelude_sdk-2.6.43.dist-info/top_level.txt,sha256=6O7C8nl-yK7FsVpsPaka_GV8PYy2uvAJtus8Tlzw4dE,12
+prelude_sdk-2.6.43.dist-info/RECORD,,

{prelude_sdk-2.6.42.dist-info → prelude_sdk-2.6.43.dist-info}/WHEEL

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