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

prelude_cli_beta/views/shared.py

@@ -0,0 +1,37 @@
+from functools import wraps
+from rich import print_json
+from rich.progress import Progress, TextColumn, SpinnerColumn
+
+
+def pretty_print(func):
+    @wraps(func)
+    def handler(*args, **kwargs):
+        try:
+            res = func(*args, **kwargs)
+            msg = None
+            if isinstance(res, tuple):
+                res, msg = res
+            if not isinstance(res, list):
+                res = [res]
+            return print_json(data=dict(status="complete", results=res, message=msg))
+        except Exception as e:
+            return print_json(
+                data=dict(
+                    status="error",
+                    results=None,
+                    message=" ".join(str(arg) for arg in e.args),
+                )
+            )
+
+    return handler
+
+
+class Spinner(Progress):
+    def __init__(self, description="Loading"):
+        super().__init__(
+            SpinnerColumn(style="green", spinner_name="line"),
+            TextColumn("[green]{task.description}..."),
+            transient=True,
+            refresh_per_second=10,
+        )
+        self.add_task(description)

prelude_cli_beta-1407.dist-info/METADATA

@@ -0,0 +1,38 @@
+Metadata-Version: 2.4
+Name: prelude-cli-beta
+Version: 1407
+Summary: For interacting with the Prelude SDK
+Home-page: https://github.com/preludeorg
+Author: Prelude Research
+Author-email: support@preludesecurity.com
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: prelude-sdk-beta==1407
+Requires-Dist: click>8
+Requires-Dist: rich
+Requires-Dist: python-dateutil
+Requires-Dist: pyyaml
+Dynamic: license-file
+
+# Prelude CLI
+
+Interact with the full range of features in Prelude Detect, organized by:
+
+- IAM: manage your account 
+- Build: write and maintain your collection of security tests
+- Detect: schedule security tests for your endpoints
+
+## Quick start
+```bash
+pip install prelude-cli
+prelude --help
+prelude --interactive
+```
+
+## Documentation
+
+https://docs.preludesecurity.com/docs/prelude-cli

prelude_cli_beta-1407.dist-info/RECORD

@@ -0,0 +1,20 @@
+prelude_cli_beta/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+prelude_cli_beta/cli.py,sha256=H3JbHuFxb46JbhsLZpH2I_5Vt9airCxUVGhqBAW6lIM,1454
+prelude_cli_beta/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+prelude_cli_beta/views/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+prelude_cli_beta/views/auth.py,sha256=W-Bc6p3CXpHiQXJCQ4FoB3SdB9xIoeJlV4WMUezSBZ4,1864
+prelude_cli_beta/views/build.py,sha256=olBRS2zjqxkvXKFmWLbqjKbbP_MDkwyjeAimHGRyfvw,16508
+prelude_cli_beta/views/configure.py,sha256=saj0kR9mQqBp7cCmq-yfEr3UwZCZIV1vL8WDQLDAO7o,991
+prelude_cli_beta/views/detect.py,sha256=jhx38nTRjZwxPAN4QbjyMkmk1mZDqZGh4uKPyG_FZ1s,13012
+prelude_cli_beta/views/generate.py,sha256=hfrlmRkb6aSo4LPaPVLhTRUFm8cxZwDqTEjLEPMlBMU,4679
+prelude_cli_beta/views/iam.py,sha256=J8y6kJGbQkEexcia69q6vLJ3aEhLyUFteCylTptBHBQ,10013
+prelude_cli_beta/views/jobs.py,sha256=2FeiJxHrw4zfgtUJq_bEoG84i_9TqZ5w6CulA80WoNA,1455
+prelude_cli_beta/views/partner.py,sha256=16zXcX5ZhiNZqKSXG9ePPGB9K3A-OgrVIdJGDJhB6f0,6379
+prelude_cli_beta/views/scm.py,sha256=DKIl9DjVNmShphD5LDHRBP_BLZAbm8CBuU8PI3gUxjY,21617
+prelude_cli_beta/views/shared.py,sha256=ZKvY8N1Vi6RtEbJli5PDzJ9R6L_bX2F27n1tm6Knvgs,1101
+prelude_cli_beta-1407.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
+prelude_cli_beta-1407.dist-info/METADATA,sha256=Q_5LBqVGpbz0pFHtUghXGxq9B5GKHeVSbZ7QdKLGjT4,993
+prelude_cli_beta-1407.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+prelude_cli_beta-1407.dist-info/entry_points.txt,sha256=WowrC6fz2D6S8S-5OY0g-bxUGGSZZ_Z6KzSXXd34pC4,88
+prelude_cli_beta-1407.dist-info/top_level.txt,sha256=j50aCGsQamLMiQh9PcolDBCAeUJzi9y08e0i9Gqshkk,17
+prelude_cli_beta-1407.dist-info/RECORD,,

prelude_cli_beta-1407.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+prelude = prelude_cli.cli:cli
+prelude-beta = prelude_cli_beta.cli:cli

prelude_cli_beta-1407.dist-info/top_level.txt

@@ -0,0 +1 @@
+prelude_cli_beta

prelude_cli_beta-1405.dist-info/METADATA

@@ -1,46 +0,0 @@
-Metadata-Version: 2.4
-Name: prelude-cli-beta
-Version: 1405
-Summary: For interacting with the Prelude API
-Home-page: https://github.com/preludeorg
-Author: Prelude Research
-Author-email: support@preludesecurity.com
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: requests
-Dynamic: license-file
-
-# Prelude SDK
-
-Interact with the Prelude Service API via Python. 
-
-> The prelude-cli utility wraps around this SDK to provide a rich command line experience.
-
-Install this package to write your own tooling that works with Build or Detect functionality.
-
-- IAM: manage your account
-- Build: write and maintain your collection of security tests
-- Detect: schedule security tests for your endpoints
-
-## Quick start
-
-```bash
-pip install prelude-sdk
-```
-
-## Documentation 
-
-TBD
-
-## Testing
-
-To test the Python SDK and Probes, run the following commands from the python/sdk/ directory:
-
-```bash
-pip install -r tests/requirements.txt
-pytest tests --api https://api.preludesecurity.com --email <EMAIL>
-```

prelude_cli_beta-1405.dist-info/RECORD

@@ -1,20 +0,0 @@
-prelude_cli_beta-1405.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
-prelude_sdk_beta/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prelude_sdk_beta/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prelude_sdk_beta/controllers/build_controller.py,sha256=tMWMMp7SttZl9rnja_mMkkRRApLOSAjBBZKFrsXPR7c,8802
-prelude_sdk_beta/controllers/detect_controller.py,sha256=zB30PESzSqds8bNG2FKxM9miwO0PponTk9XWfZ4WvPA,7499
-prelude_sdk_beta/controllers/export_controller.py,sha256=lIRmO7b2sMLyA8WAsGFqtXuCp79r25zgopJGM5KnxvA,847
-prelude_sdk_beta/controllers/generate_controller.py,sha256=JUYlBdMh8alSMvin32qb3hB1JNGbT-BarEgxhMhg9PI,1304
-prelude_sdk_beta/controllers/http_controller.py,sha256=cj1SxmAX049ioAhTOQR4atuM82nJgd11S60-ISCg738,2300
-prelude_sdk_beta/controllers/iam_controller.py,sha256=eDT0kw1IfA_enQF_1kC_vSWS7-LFL9PO24EoxxM8Ax4,7856
-prelude_sdk_beta/controllers/jobs_controller.py,sha256=Ekf41oD72gBJeEfSfGYUfxpsQI-SUqY8EdtBmzqEIM0,755
-prelude_sdk_beta/controllers/partner_controller.py,sha256=T4D3Nl1SF6OLIiGPR81cPGAxUd4GSnKZ9Jl-rbAHZFE,5113
-prelude_sdk_beta/controllers/probe_controller.py,sha256=rwBnqkFJa1KuVwZMluM-pfo8bhQFB6JKSRSeK7s-kew,406
-prelude_sdk_beta/controllers/scm_controller.py,sha256=l30RZa4X19nBm6E411kAT36LM1CDl15Bd0yC3_jUb80,12778
-prelude_sdk_beta/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prelude_sdk_beta/models/account.py,sha256=ksyTZDOZpbT8XDu-Ygs_51ZG25oBd8qiwjO6-L6HH9Y,8734
-prelude_sdk_beta/models/codes.py,sha256=-gxBflkV-RdE3LwGSDDSO48eK-J8skZ5CivMD2MYc5w,11425
-prelude_cli_beta-1405.dist-info/METADATA,sha256=fBpqVU1UPe3eZAb9WL7FPXfB8A4FA4f2iQZZ2pNblwo,1190
-prelude_cli_beta-1405.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-prelude_cli_beta-1405.dist-info/top_level.txt,sha256=pqXTtEd5ElvJKoO6HAz232H9FW5j6X7gW4kEEakfSFM,17
-prelude_cli_beta-1405.dist-info/RECORD,,

prelude_cli_beta-1405.dist-info/top_level.txt

@@ -1 +0,0 @@
-prelude_sdk_beta

prelude_sdk_beta/controllers/build_controller.py

@@ -1,309 +0,0 @@
-import urllib
-
-from prelude_sdk.controllers.http_controller import HttpController
-from prelude_sdk.models.account import verify_credentials
-from prelude_sdk.models.codes import Control, EDRResponse
-
-
-class BuildController(HttpController):
-
-    def __init__(self, account):
-        super().__init__(account)
-
-    @verify_credentials
-    def clone_test(self, source_test_id):
-        """Clone a test"""
-        res = self.post(
-            f"{self.account.hq}/build/tests",
-            json=dict(source_test_id=source_test_id),
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def create_test(self, name, unit, technique=None, test_id=None):
-        """Create or update a test"""
-        body = dict(name=name, unit=unit)
-        if technique:
-            body["technique"] = technique
-        if test_id:
-            body["id"] = test_id
-
-        res = self.post(
-            f"{self.account.hq}/build/tests",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def update_test(
-        self,
-        test_id,
-        name=None,
-        unit=None,
-        technique=None,
-        crowdstrike_expected_outcome: EDRResponse = None,
-    ):
-        """Update a test"""
-        body = dict()
-        if crowdstrike_expected_outcome:
-            body["expected"] = dict(crowdstrike=crowdstrike_expected_outcome.value)
-        if name:
-            body["name"] = name
-        if unit:
-            body["unit"] = unit
-        if technique is not None:
-            body["technique"] = technique
-
-        res = self.post(
-            f"{self.account.hq}/build/tests/{test_id}",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def delete_test(self, test_id, purge):
-        """Delete an existing test"""
-        res = self.delete(
-            f"{self.account.hq}/build/tests/{test_id}",
-            json=dict(purge=purge),
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def undelete_test(self, test_id):
-        """Undelete a tombstoned test"""
-        res = self.post(
-            f"{self.account.hq}/build/tests/{test_id}/undelete",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def upload(self, test_id, filename, data, skip_compile=False):
-        """Upload a test or attachment"""
-        if len(data) > 1000000:
-            raise ValueError(f"File size must be under 1MB ({filename})")
-
-        h = self.account.headers | {"Content-Type": "application/octet-stream"}
-        query_params = ""
-        if skip_compile:
-            query_params = "?" + urllib.parse.urlencode(dict(skip_compile=True))
-        res = self.post(
-            f"{self.account.hq}/build/tests/{test_id}/{filename}{query_params}",
-            data=data,
-            headers=h,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def compile_code_string(self, code: str, source_test_id: str = None):
-        """Compile a code string"""
-        res = self.post(
-            f"{self.account.hq}/build/compile",
-            json=dict(code=code, source_test_id=source_test_id),
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def get_compile_status(self, job_id):
-        res = self.get(
-            f"{self.account.hq}/build/compile/{job_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def create_threat(
-        self, name, published, threat_id=None, source_id=None, source=None, tests=None
-    ):
-        """Create a threat"""
-        body = dict(name=name, published=published)
-        if threat_id:
-            body["id"] = threat_id
-        if source_id:
-            body["source_id"] = source_id
-        if source:
-            body["source"] = source
-        if tests:
-            body["tests"] = tests
-
-        res = self.post(
-            f"{self.account.hq}/build/threats",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def update_threat(
-        self,
-        threat_id,
-        name=None,
-        source_id=None,
-        source=None,
-        published=None,
-        tests=None,
-    ):
-        """Update a threat"""
-        body = dict()
-        if name:
-            body["name"] = name
-        if source_id is not None:
-            body["source_id"] = source_id
-        if source is not None:
-            body["source"] = source
-        if published is not None:
-            body["published"] = published
-        if tests is not None:
-            body["tests"] = tests
-
-        res = self.post(
-            f"{self.account.hq}/build/threats/{threat_id}",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def delete_threat(self, threat_id, purge):
-        """Delete an existing threat"""
-        res = self.delete(
-            f"{self.account.hq}/build/threats/{threat_id}",
-            json=dict(purge=purge),
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def undelete_threat(self, threat_id):
-        """Undelete a tombstoned threat"""
-        res = self.post(
-            f"{self.account.hq}/build/threats/{threat_id}/undelete",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def create_detection(
-        self, rule: str, test_id: str, detection_id=None, rule_id=None
-    ):
-        """Create a detection"""
-        body = dict(rule=rule, test_id=test_id)
-        if detection_id:
-            body["detection_id"] = detection_id
-        if rule_id:
-            body["rule_id"] = rule_id
-
-        res = self.post(
-            f"{self.account.hq}/build/detections",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def update_detection(self, detection_id: str, rule=None, test_id=None):
-        """Update a detection"""
-        body = dict()
-        if rule:
-            body["rule"] = rule
-        if test_id:
-            body["test_id"] = test_id
-
-        res = self.post(
-            f"{self.account.hq}/build/detections/{detection_id}",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def delete_detection(self, detection_id: str):
-        """Delete an existing detection"""
-        res = self.delete(
-            f"{self.account.hq}/build/detections/{detection_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def create_threat_hunt(
-        self,
-        control: Control,
-        test_id: str,
-        name: str,
-        query: str,
-        threat_hunt_id: str = None,
-    ):
-        """Create a threat hunt"""
-        body = dict(
-            control=control.name,
-            name=name,
-            query=query,
-            test_id=test_id,
-        )
-        if threat_hunt_id:
-            body["id"] = threat_hunt_id
-
-        res = self.post(
-            f"{self.account.hq}/build/threat_hunts",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def update_threat_hunt(
-        self,
-        threat_hunt_id: str,
-        name: str = None,
-        query: str = None,
-        test_id: str = None,
-    ):
-        """Update a threat hunt"""
-        body = dict()
-        if name:
-            body["name"] = name
-        if query:
-            body["query"] = query
-        if test_id:
-            body["test_id"] = test_id
-
-        res = self.post(
-            f"{self.account.hq}/build/threat_hunts/{threat_hunt_id}",
-            json=body,
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()
-
-    @verify_credentials
-    def delete_threat_hunt(self, threat_hunt_id: str):
-        """Delete an existing threat hunt"""
-        res = self.delete(
-            f"{self.account.hq}/build/threat_hunts/{threat_hunt_id}",
-            headers=self.account.headers,
-            timeout=10,
-        )
-        return res.json()