prelude-cli-beta 1405__py3-none-any.whl → 1407__py3-none-any.whl

prelude_sdk_beta/controllers/detect_controller.py

@@ -1,243 +0,0 @@
-from prelude_sdk.controllers.http_controller import HttpController
-from prelude_sdk.models.account import verify_credentials
-
-
-class DetectController(HttpController):
-
-    def __init__(self, account):
-        super().__init__(account)
-
-    def register_endpoint(self, host, serial_num, reg_string, tags=None):
-        """Register (or re-register) an endpoint to your account"""
-        body = dict(id=f"{host}:{serial_num}")
-        if tags:
-            body["tags"] = tags
-        account, token = reg_string.split("/")
-
-        res = self._session.post(
-            f"{self.account.hq}/detect/endpoint",
-            headers=dict(account=account, token=token, _product="py-sdk"),
-            json=body,
-            timeout=10,
-        )
-        return res.text
-
-    @verify_credentials
-    def update_endpoint(self, endpoint_id, tags=None):
-        """Update an endpoint in your account"""
-        body = dict()
-        if tags is not None:
-            body["tags"] = tags
-
-        res = self.post(
-            f"{self.account.hq}/detect/endpoint/{endpoint_id}",
-            headers=self.account.headers,
-            json=body,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def delete_endpoint(self, ident: str):
-        """Delete an endpoint from your account"""
-        params = dict(id=ident)
-        res = self.delete(
-            f"{self.account.hq}/detect/endpoint",
-            headers=self.account.headers,
-            json=params,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def list_endpoints(self, days: int = 90):
-        """List all endpoints on your account"""
-        params = dict(days=days)
-        res = self.get(
-            f"{self.account.hq}/detect/endpoint",
-            headers=self.account.headers,
-            params=params,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def describe_activity(self, filters: dict, view: str = "protected"):
-        """Get report for an Account"""
-        params = dict(view=view, **filters)
-        res = self.get(
-            f"{self.account.hq}/detect/activity",
-            headers=self.account.headers,
-            params=params,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def threat_hunt_activity(self, threat_hunt_id=None, test_id=None, threat_id=None):
-        """Get threat hunt activity"""
-        filters = dict(
-            threat_hunt_id=threat_hunt_id, test_id=test_id, threat_id=threat_id
-        )
-        res = self.get(
-            f"{self.account.hq}/detect/threat_hunt_activity",
-            headers=self.account.headers,
-            params=filters,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def list_tests(self, filters: dict = None):
-        """List all tests available to an account"""
-        res = self.get(
-            f"{self.account.hq}/detect/tests",
-            headers=self.account.headers,
-            params=filters if filters else {},
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def get_test(self, test_id):
-        """Get properties of an existing test"""
-        res = self.get(
-            f"{self.account.hq}/detect/tests/{test_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def list_techniques(self):
-        """List techniques"""
-        res = self.get(
-            f"{self.account.hq}/detect/techniques",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def list_threats(self):
-        """List threats"""
-        res = self.get(
-            f"{self.account.hq}/detect/threats",
-            headers=self.account.headers,
-            params={},
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def get_threat(self, threat_id):
-        """Get properties of an existing threat"""
-        res = self.get(
-            f"{self.account.hq}/detect/threats/{threat_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def list_detections(self):
-        """List detections"""
-        res = self.get(
-            f"{self.account.hq}/detect/detections",
-            headers=self.account.headers,
-            params={},
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def get_detection(self, detection_id):
-        """Get properties of an existing detection"""
-        res = self.get(
-            f"{self.account.hq}/detect/detections/{detection_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def list_threat_hunts(self, filters: dict = None):
-        """List threat hunts"""
-        res = self.get(
-            f"{self.account.hq}/detect/threat_hunts",
-            headers=self.account.headers,
-            params=filters if filters else {},
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def get_threat_hunt(self, threat_hunt_id):
-        """Get properties of an existing threat hunt"""
-        res = self.get(
-            f"{self.account.hq}/detect/threat_hunts/{threat_hunt_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def do_threat_hunt(self, threat_hunt_id):
-        """Run a threat hunt"""
-        res = self.post(
-            f"{self.account.hq}/detect/threat_hunts/{threat_hunt_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def download(self, test_id, filename):
-        """Clone a test file or attachment"""
-        res = self.get(
-            f"{self.account.hq}/detect/tests/{test_id}/{filename}",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.content
-
-    @verify_credentials
-    def schedule(self, items: list):
-        """Schedule tests and threats so endpoints will start running them
-
-        Example: items=[dict(run_code='DAILY', tags='grp-1,grp2', test_id='123-123-123'),
-                        dict(run_code='DAILY', tags='grp-1', threat_id='abc-def-ghi')]
-        """
-        res = self.post(
-            url=f"{self.account.hq}/detect/queue",
-            headers=self.account.headers,
-            json=dict(items=items),
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def unschedule(self, items: list):
-        """Unschedule tests and threats so endpoints will stop running them
-
-        Example: items=[dict(tags='grp-1,grp2', test_id='123-123-123'),
-                        dict(tags='grp-1', threat_id='abc-def-ghi')]
-        """
-        res = self.delete(
-            f"{self.account.hq}/detect/queue",
-            headers=self.account.headers,
-            json=dict(items=items),
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def accept_terms(self, name, version):
-        """Accept terms and conditions"""
-        res = self.post(
-            f"{self.account.hq}/iam/terms",
-            headers=self.account.headers,
-            json=dict(name=name, version=version),
-            timeout=10,
-        )
-        return res.json()

prelude_sdk_beta/controllers/export_controller.py

@@ -1,31 +0,0 @@
-from prelude_sdk.controllers.http_controller import HttpController
-from prelude_sdk.models.account import verify_credentials
-from prelude_sdk.models.codes import SCMCategory
-
-
-class ExportController(HttpController):
-
-    def __init__(self, account):
-        super().__init__(account)
-
-    @verify_credentials
-    def export_scm(
-        self,
-        export_type: SCMCategory,
-        filter: str = None,
-        orderby: str = None,
-        top: int = None,
-    ):
-        """Download partner data as a CSV"""
-        params = {
-            "$filter": filter,
-            "$orderby": orderby,
-            "$top": top,
-        }
-        res = self.post(
-            f"{self.account.hq}/export/scm/{export_type.name}",
-            params=params,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()

prelude_sdk_beta/controllers/generate_controller.py

@@ -1,40 +0,0 @@
-from prelude_sdk.controllers.http_controller import HttpController
-from prelude_sdk.models.account import verify_credentials
-from prelude_sdk.models.codes import Control
-
-
-class GenerateController(HttpController):
-    def __init__(self, account):
-        super().__init__(account)
-
-    @verify_credentials
-    def upload_threat_intel(self, file: str):
-        with open(file, "rb") as f:
-            body = f.read()
-        res = self.post(
-            f"{self.account.hq}/generate/threat-intel",
-            data=body,
-            headers=self.account.headers | {"Content-Type": "application/pdf"},
-            timeout=30,
-        )
-        return res.json()
-
-    @verify_credentials
-    def get_threat_intel(self, job_id: str):
-        res = self.get(
-            f"{self.account.hq}/generate/threat-intel/{job_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def generate_from_partner_advisory(self, partner: Control, advisory_id: str):
-        params = dict(advisory_id=advisory_id)
-        res = self.post(
-            f"{self.account.hq}/generate/partner-advisories/{partner.name}",
-            headers=self.account.headers,
-            json=params,
-            timeout=30,
-        )
-        return res.json()

prelude_sdk_beta/controllers/http_controller.py

@@ -1,63 +0,0 @@
-import os
-import requests
-
-from requests.adapters import HTTPAdapter, Retry
-
-
-PRELUDE_BACKOFF_FACTOR = int(os.getenv("PRELUDE_BACKOFF_FACTOR", 30))
-PRELUDE_BACKOFF_TOTAL = int(os.getenv("PRELUDE_BACKOFF_TOTAL", 0))
-
-
-class HttpController(object):
-    def __init__(self, account):
-        self._session = requests.Session()
-        self.account = account
-
-        retry = Retry(
-            total=PRELUDE_BACKOFF_TOTAL,
-            backoff_factor=PRELUDE_BACKOFF_FACTOR,
-            status_forcelist=[429],
-        )
-
-        self._session.mount("http://", HTTPAdapter(max_retries=retry))
-        self._session.mount("https://", HTTPAdapter(max_retries=retry))
-
-    def get(self, url, retry=True, **kwargs):
-        res = self._session.get(url, **kwargs)
-        if res.status_code == 200:
-            return res
-        if res.status_code == 401 and retry and self.account.token_location:
-            self.account.refresh_tokens()
-            self.account.update_auth_header()
-            return self.get(url, retry=False, **kwargs)
-        raise Exception(res.text)
-
-    def post(self, url, retry=True, **kwargs):
-        res = self._session.post(url, **kwargs)
-        if res.status_code == 200:
-            return res
-        if res.status_code == 401 and retry and self.account.token_location:
-            self.account.refresh_tokens()
-            self.account.update_auth_header()
-            return self.post(url, retry=False, **kwargs)
-        raise Exception(res.text)
-
-    def delete(self, url, retry=True, **kwargs):
-        res = self._session.delete(url, **kwargs)
-        if res.status_code == 200:
-            return res
-        if res.status_code == 401 and retry and self.account.token_location:
-            self.account.refresh_tokens()
-            self.account.update_auth_header()
-            return self.delete(url, retry=False, **kwargs)
-        raise Exception(res.text)
-
-    def put(self, url, retry=True, **kwargs):
-        res = self._session.put(url, **kwargs)
-        if res.status_code == 200:
-            return res
-        if res.status_code == 401 and retry and self.account.token_location:
-            self.account.refresh_tokens()
-            self.account.update_auth_header()
-            return self.put(url, retry=False, **kwargs)
-        raise Exception(res.text)

prelude_sdk_beta/controllers/iam_controller.py

@@ -1,278 +0,0 @@
-from prelude_sdk.controllers.http_controller import HttpController
-from prelude_sdk.models.account import verify_credentials
-from prelude_sdk.models.codes import Mode, Permission
-
-
-class IAMAccountController(HttpController):
-
-    def __init__(self, account):
-        super().__init__(account)
-
-    @verify_credentials
-    def get_account(self):
-        """Get account properties"""
-        res = self.get(
-            f"{self.account.hq}/iam/account", headers=self.account.headers, timeout=10
-        )
-        return res.json()
-
-    @verify_credentials
-    def purge_account(self):
-        """Delete an account and all things in it"""
-        res = self.delete(
-            f"{self.account.hq}/iam/account", headers=self.account.headers, timeout=10
-        )
-        return res.json()
-
-    @verify_credentials
-    def update_account(self, mode: Mode = None, company: str = None, slug: str = None):
-        """Update properties on an account"""
-        body = dict()
-        if mode is not None:
-            body["mode"] = mode.name
-        if company is not None:
-            body["company"] = company
-        if slug is not None:
-            body["slug"] = slug
-
-        res = self.put(
-            f"{self.account.hq}/iam/account",
-            headers=self.account.headers,
-            json=body,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def attach_oidc(
-        self,
-        client_id: str,
-        client_secret: str,
-        issuer: str,
-        oidc_url: str,
-    ):
-        """Attach OIDC to an account"""
-        email_attr = "email"
-        if issuer == "azure":
-            email_attr = "upn"
-        body = dict(
-            client_id=client_id,
-            client_secret=client_secret,
-            email_attr=email_attr,
-            issuer=issuer,
-            oidc_url=oidc_url,
-        )
-
-        res = self.post(
-            f"{self.account.hq}/iam/account/oidc",
-            headers=self.account.headers,
-            json=body,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def detach_oidc(self):
-        """Detach OIDC to an account"""
-        res = self.delete(
-            f"{self.account.hq}/iam/account/oidc",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def invite_user(
-        self,
-        email: str,
-        oidc: str | None,
-        permission: Permission,
-        name: str | None = None,
-    ):
-        """Invite a new user to the account"""
-        body = dict(permission=permission.name, handle=email, oidc=oidc)
-        if name:
-            body["name"] = name
-
-        res = self.post(
-            url=f"{self.account.hq}/iam/account/user",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def create_service_user(self, name: str):
-        """Create a service user"""
-        body = dict(name=name)
-
-        res = self.post(
-            f"{self.account.hq}/iam/account/service_user",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def delete_service_user(self, handle: str):
-        """Delete service user"""
-        body = dict(handle=handle)
-
-        res = self.delete(
-            f"{self.account.hq}/iam/account/service_user",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def update_account_user(
-        self,
-        email: str,
-        oidc: str | None,
-        permission: Permission = None,
-    ):
-        """Update properties on an account user"""
-        body = dict(handle=email, oidc=oidc)
-        if permission is not None:
-            body["permission"] = permission.name
-
-        res = self.put(
-            f"{self.account.hq}/iam/account/user",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def remove_user(self, email: str, oidc: str | None):
-        """Remove user from the account"""
-        params = dict(handle=email, oidc=oidc)
-
-        res = self.delete(
-            f"{self.account.hq}/iam/account/user",
-            params=params,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def audit_logs(self, days: int = 7, limit: int = 1000):
-        """Get audit logs from the last X days"""
-        params = dict(days=days, limit=limit)
-        res = self.get(
-            f"{self.account.hq}/iam/audit",
-            headers=self.account.headers,
-            params=params,
-            timeout=30,
-        )
-        return res.json()
-
-
-    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)
-
-        res = self._session.post(
-            f"{self.account.hq}/iam/new_user_and_account",
-            headers=self.account.headers,
-            json=body,
-            timeout=20,
-        )
-        data = res.json()
-        if self.account.profile:
-            self.account.keychain.configure_keychain(
-                account=data["account_id"],
-                handle=data["user_id"],
-                hq=self.account.hq,
-                profile=self.account.profile,
-            )
-        return data
-
-
-class IAMUserController(HttpController):
-
-    def __init__(self, account):
-        super().__init__(account)
-
-    @verify_credentials
-    def list_accounts(self):
-        """List all accounts for your user"""
-        res = self.get(
-            f"{self.account.hq}/iam/user/account",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def purge_user(self):
-        """Delete your user"""
-        res = self.delete(
-            f"{self.account.hq}/iam/user", headers=self.account.headers, timeout=10
-        )
-        return res.json()
-
-    @verify_credentials
-    def update_user(
-        self,
-        name: str = None,
-    ):
-        """Update properties on a user"""
-        body = dict()
-        if name is not None:
-            body["name"] = name
-
-        res = self.put(
-            f"{self.account.hq}/iam/user",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    def forgot_password(self):
-        """Send a forgot password email"""
-        body = dict(handle=self.account.handle)
-
-        res = self.post(
-            f"{self.account.hq}/iam/user/forgot_password",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    def confirm_forgot_password(self, confirmation_code: str, new_password: str):
-        """Change a password using confirmation code"""
-        body = dict(
-            handle=self.account.handle,
-            confirmation_code=confirmation_code,
-            password=new_password,
-        )
-
-        res = self.post(
-            f"{self.account.hq}/iam/user/forgot_password",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def change_password(self, current_password: str, new_password: str):
-        """Change your password"""
-        body = dict(current_password=current_password, new_password=new_password)
-
-        res = self.post(
-            f"{self.account.hq}/iam/user/change_password",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()

prelude_sdk_beta/controllers/jobs_controller.py

@@ -1,26 +0,0 @@
-from prelude_sdk.controllers.http_controller import HttpController
-from prelude_sdk.models.account import verify_credentials
-
-
-class JobsController(HttpController):
-
-    def __init__(self, account):
-        super().__init__(account)
-
-    @verify_credentials
-    def job_statuses(self):
-        """Get job statuses"""
-        res = self.get(
-            f"{self.account.hq}/jobs/statuses", headers=self.account.headers, timeout=30
-        )
-        return res.json()
-
-    @verify_credentials
-    def job_status(self, job_id: str):
-        """Get job status given job ID"""
-        res = self.get(
-            f"{self.account.hq}/jobs/statuses/{job_id}",
-            headers=self.account.headers,
-            timeout=30,
-        )
-        return res.json()