prelude-cli-beta 1406__py3-none-any.whl → 1408__py3-none-any.whl

prelude_cli_beta/cli.py

@@ -26,8 +26,13 @@ def complete_profile(ctx, param, incomplete):
     show_default=True,
     shell_complete=complete_profile,
 )
-def cli(ctx, profile):
-    ctx.obj = Account.from_keychain(profile=profile)
+@click.option(
+    "--resolve_enums",
+    is_flag=True,
+    help="Resolve enum values to their string representation",
+)
+def cli(ctx, profile, resolve_enums):
+    ctx.obj = Account.from_keychain(profile=profile, resolve_enums=resolve_enums)
     if ctx.invoked_subcommand is None:
         click.echo(ctx.get_help())
 

prelude_cli_beta/views/generate.py

@@ -48,6 +48,11 @@ def _process_results(result: dict, output_dir: str, job_id: str) -> dict:
                         f,
                         indent=4,
                     )
+                if content["readme"]:
+                    with open(
+                        f"{output_dir}/{technique_directory}/README.md", "w"
+                    ) as f:
+                        f.write(content["readme"])
         return dict(
             output_dir=output_dir,
             successfully_generated=[

prelude_cli_beta/views/scm.py

@@ -205,7 +205,17 @@ def export(controller, type, output_file, limit, odata_filter, odata_orderby):
         return result, f"Exported data to {output_file}"
 
 
-@scm.command("groups")
+@click.group()
+@click.pass_context
+def group(ctx):
+    """SCM group commands"""
+    ctx.obj = ScmController(account=ctx.obj.account)
+
+
+scm.add_command(group)
+
+
+@group.command("list")
 @click.option("--odata_filter", help="OData filter string", default=None)
 @click.option("--odata_orderby", help="OData orderby string", default=None)
 @click.pass_obj
@@ -213,10 +223,12 @@ def export(controller, type, output_file, limit, odata_filter, odata_orderby):
 def list_partner_groups(controller, odata_filter, odata_orderby):
     """List all partner groups"""
     with Spinner(description="Fetching partner groups"):
-        return controller.list_partner_groups(filter=odata_filter, orderby=odata_orderby)
+        return controller.list_partner_groups(
+            filter=odata_filter, orderby=odata_orderby
+        )
 
 
-@scm.command("sync-groups")
+@group.command("sync")
 @click.argument(
     "partner",
     type=click.Choice(
@@ -242,7 +254,17 @@ def sync_groups(controller, partner, instance_id, group_ids):
         return result
 
 
-@scm.command("create-threat")
+@click.group()
+@click.pass_context
+def threat(ctx):
+    """SCM threat commands"""
+    ctx.obj = ScmController(account=ctx.obj.account)
+
+
+scm.add_command(threat)
+
+
+@threat.command("create")
 @click.argument("name")
 @click.option(
     "-d", "--description", help="description of the threat", default=None, type=str
@@ -298,7 +320,7 @@ def create_threat(
         )
 
 
-@scm.command("delete-threat")
+@threat.command("delete")
 @click.argument("threat_id")
 @click.confirmation_option(prompt="Are you sure?")
 @click.pass_obj
@@ -309,7 +331,7 @@ def delete_threat(controller, threat_id):
         return controller.delete_threat(id=threat_id)
 
 
-@scm.command("threats")
+@threat.command("list")
 @click.pass_obj
 @pretty_print
 def list_threats(controller):
@@ -318,7 +340,7 @@ def list_threats(controller):
         return controller.list_threats()
 
 
-@scm.command("threat")
+@threat.command("get")
 @click.argument("threat_id")
 @click.pass_obj
 @pretty_print
@@ -356,7 +378,237 @@ def parse_from_partner_advisory(controller, partner, advisory_id):
         )
 
 
-@scm.command("list-notifications")
+@click.group()
+@click.pass_context
+def exception(ctx):
+    """SCM exception commands"""
+    ctx.obj = ScmController(account=ctx.obj.account)
+
+
+@click.group()
+@click.pass_context
+def object(ctx):
+    """SCM object exception commands"""
+    ctx.obj = ScmController(account=ctx.obj.account)
+
+
+@click.group()
+@click.pass_context
+def policy(ctx):
+    """SCM policy exception commands"""
+    ctx.obj = ScmController(account=ctx.obj.account)
+
+
+exception.add_command(object)
+exception.add_command(policy)
+scm.add_command(exception)
+
+
+@object.command("list")
+@click.pass_obj
+@pretty_print
+def list_object_exceptions(controller):
+    """List all object exceptions"""
+    with Spinner(description="Fetching object exceptions"):
+        return controller.list_object_exceptions()
+
+
+@object.command("create")
+@click.argument(
+    "category",
+    type=click.Choice(
+        [
+            c.name
+            for c in ControlCategory
+            if c
+            not in [
+                ControlCategory.NONE,
+                ControlCategory.INVALID,
+                ControlCategory.PRIVATE_REPO,
+            ]
+        ],
+        case_sensitive=False,
+    ),
+)
+@click.option(
+    "-f", "--filter", help="OData filter string", default=None, required=True, type=str
+)
+@click.option(
+    "-e",
+    "--expires",
+    help="expiry date (YYYY-MM-DD hh:mm:ss ([+-]hh:mm))",
+    default=None,
+    type=str,
+)
+@click.option("-n", "--name", help="exception name", default=None, type=str)
+@click.pass_obj
+@pretty_print
+def create_object_exception(controller, category, filter, expires, name):
+    """Create object exception"""
+    with Spinner(description=f"Creating object exception"):
+        return controller.create_object_exception(
+            category=ControlCategory[category],
+            filter=filter,
+            name=name,
+            expires=expires,
+        )
+
+
+@object.command("update")
+@click.argument("exception_id", type=str)
+@click.option(
+    "-e",
+    "--expires",
+    help="Expiry Date (YYYY-MM-DD hh:mm:ss ([+-]hh:mm))",
+    default=ScmController.default,
+)
+@click.option("-f", "--filter", help="OData filter string", default=None, type=str)
+@click.option("-n", "--name", help="Exception Name", default=None, type=str)
+@click.pass_obj
+@pretty_print
+def update_object_exception(controller, exception_id, expires, filter, name):
+    """Update object exception"""
+    with Spinner(description=f"Updating object exception"):
+        return controller.update_object_exception(
+            exception_id=exception_id, filter=filter, name=name, expires=expires
+        )
+
+
+@object.command("delete")
+@click.argument("exception_id", type=str)
+@click.confirmation_option(prompt="Are you sure?")
+@click.pass_obj
+@pretty_print
+def delete_object_exception(controller, exception_id):
+    """Delete object exception"""
+    with Spinner(description=f"Delete object exception"):
+        return controller.delete_object_exception(exception_id=exception_id)
+
+
+@policy.command("list")
+@click.pass_obj
+@pretty_print
+def list_policy_exceptions(controller):
+    """List all policy exceptions"""
+    with Spinner(description="Fetching policy exceptions"):
+        return controller.list_policy_exceptions()
+
+
+@policy.command("create")
+@click.argument(
+    "partner",
+    type=click.Choice(
+        [c.name for c in Control if c != Control.INVALID], case_sensitive=False
+    ),
+)
+@click.option("-i", "--instance_id", required=True, help="instance ID of the partner")
+@click.option("-p", "--policy_id", required=True, help="ID of the policy to create")
+@click.option(
+    "-s",
+    "--settings",
+    required=True,
+    help="Comma separated list of all setting names to be excluded",
+)
+@click.option(
+    "-e",
+    "--expires",
+    help="Expiry Date (YYYY-MM-DD hh:mm:ss ([+-]hh:mm))",
+    default=None,
+    type=str,
+)
+@click.pass_obj
+@pretty_print
+def create_policy_exception(
+    controller, partner, instance_id, policy_id, settings, expires
+):
+    """Create policy exception"""
+    with Spinner(description=f"Creating policy exception"):
+        return controller.create_policy_exceptions(
+            partner=Control[partner],
+            expires=expires,
+            instance_id=instance_id,
+            policy_id=policy_id,
+            setting_names=settings.split(",") if settings else None,
+        )
+
+
+@policy.command("update")
+@click.argument(
+    "partner",
+    type=click.Choice(
+        [c.name for c in Control if c != Control.INVALID], case_sensitive=False
+    ),
+)
+@click.option("-i", "--instance_id", required=True, help="instance ID of the partner")
+@click.option("-p", "--policy_id", required=True, help="ID of the policy to update")
+@click.option(
+    "-e",
+    "--expires",
+    help="Expiry Date (YYYY-MM-DD hh:mm:ss ([+-]hh:mm))",
+    default=ScmController.default,
+)
+@click.option(
+    "-s",
+    "--settings",
+    help="Comma separated list of all setting names to be excluded",
+)
+@click.pass_obj
+@pretty_print
+def update_policy_exception(
+    controller, partner, instance_id, policy_id, expires, settings
+):
+    """Update policy exception"""
+    with Spinner(description=f"Updating Policy exception"):
+        return controller.update_policy_exception(
+            partner=Control[partner],
+            expires=expires,
+            instance_id=instance_id,
+            policy_id=policy_id,
+            setting_names=settings.split(",") if settings else None,
+        )
+
+
+@policy.command("delete")
+@click.argument(
+    "partner",
+    type=click.Choice(
+        [c.name for c in Control if c != Control.INVALID], case_sensitive=False
+    ),
+)
+@click.option("-i", "--instance_id", required=True, help="instance ID of the partner")
+@click.option("-p", "--policy_id", required=True, help="ID of the policy to be deleted")
+@click.confirmation_option(prompt="Are you sure?")
+@click.pass_obj
+@pretty_print
+def delete_policy_exception(controller, partner, instance_id, policy_id):
+    """Delete policy exception"""
+    with Spinner(description=f"Deleting Policy exception"):
+        return controller.delete_policy_exception(
+            instance_id=instance_id, policy_id=policy_id
+        )
+
+    """Delete policy exception removes all exceptions in a policy"""
+    with Spinner(description=f"Deleting policy exception"):
+        return controller.put_policy_exceptions(
+            partner=Control[partner],
+            expires=None,
+            instance_id=instance_id,
+            policy_id=policy_id,
+            setting_names=[],
+        )
+
+
+@click.group()
+@click.pass_context
+def notification(ctx):
+    """SCM notification commands"""
+    ctx.obj = ScmController(account=ctx.obj.account)
+
+
+scm.add_command(notification)
+
+
+@notification.command("list")
 @click.pass_obj
 @pretty_print
 def list_notifications(controller):
@@ -364,7 +616,7 @@ def list_notifications(controller):
         return controller.list_notifications()
 
 
-@scm.command("delete-notification")
+@notification.command("delete")
 @click.argument("notification_id", type=str)
 @click.confirmation_option(prompt="Are you sure?")
 @click.pass_obj
@@ -374,7 +626,7 @@ def delete_notification(controller, notification_id):
         return controller.delete_notification(notification_id)
 
 
-@scm.command("upsert-notification")
+@notification.command("upsert")
 @click.argument(
     "control_category",
     type=click.Choice([c.name for c in ControlCategory], case_sensitive=False),
@@ -469,3 +721,24 @@ def upsert_notification(
             teams_urls=teams_urls.split(",") if teams_urls else None,
             title=title,
         )
+
+
+@scm.command("notations")
+@click.pass_obj
+@pretty_print
+def list_notations(controller):
+    """List all notations"""
+    with Spinner("Fetching notations"):
+        return controller.list_notations()
+
+
+@scm.command("history")
+@click.option("--odata_filter", help="OData filter string", default=None)
+@click.option("--start", type=str, default=None, help="start date")
+@click.option("--end", type=str, default=None, help="end date")
+@click.pass_obj
+@pretty_print
+def list_history(controller, odata_filter, start, end):
+    """List history"""
+    with Spinner("Fetching SCM history"):
+        return controller.list_history(start, end, filter=odata_filter)

{prelude_cli_beta-1406.dist-info → prelude_cli_beta-1408.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prelude-cli-beta
-Version: 1406
+Version: 1408
 Summary: For interacting with the Prelude SDK
 Home-page: https://github.com/preludeorg
 Author: Prelude Research
@@ -11,7 +11,7 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: prelude-sdk-beta==1406
+Requires-Dist: prelude-sdk-beta==1408
 Requires-Dist: click>8
 Requires-Dist: rich
 Requires-Dist: python-dateutil

{prelude_cli_beta-1406.dist-info → prelude_cli_beta-1408.dist-info}/RECORD

@@ -1,20 +1,20 @@
 prelude_cli_beta/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prelude_cli_beta/cli.py,sha256=FiBYXVT0_Ft6sTAlB00tMfGsRdSHhnNnm0dHWZ06VMU,1289
+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/generate.py,sha256=zkdzBaBO7K5tXarnKfzZjo9-WaG5i80y1ZbQk0398sA,4904
 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=_SkwYXIR-fSIBnXZfID0aHkH-quK6ePdhkxx18wczl0,14025
+prelude_cli_beta/views/scm.py,sha256=DKIl9DjVNmShphD5LDHRBP_BLZAbm8CBuU8PI3gUxjY,21617
 prelude_cli_beta/views/shared.py,sha256=ZKvY8N1Vi6RtEbJli5PDzJ9R6L_bX2F27n1tm6Knvgs,1101
-prelude_cli_beta-1406.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
-prelude_cli_beta-1406.dist-info/METADATA,sha256=TDssBOmg9KcpnuYki6XYu_-tlsMMGHO3TPTkW5vzZy8,993
-prelude_cli_beta-1406.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-prelude_cli_beta-1406.dist-info/entry_points.txt,sha256=WowrC6fz2D6S8S-5OY0g-bxUGGSZZ_Z6KzSXXd34pC4,88
-prelude_cli_beta-1406.dist-info/top_level.txt,sha256=j50aCGsQamLMiQh9PcolDBCAeUJzi9y08e0i9Gqshkk,17
-prelude_cli_beta-1406.dist-info/RECORD,,
+prelude_cli_beta-1408.dist-info/licenses/LICENSE,sha256=ttdT5omfN6LNmtQoIjUhkkFhz6i44SDMRNwKrbfyTf8,1069
+prelude_cli_beta-1408.dist-info/METADATA,sha256=jHue7P7KlUR23pJtGbQR9-sru5skrYOjZQcxWGnOVJM,993
+prelude_cli_beta-1408.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+prelude_cli_beta-1408.dist-info/entry_points.txt,sha256=WowrC6fz2D6S8S-5OY0g-bxUGGSZZ_Z6KzSXXd34pC4,88
+prelude_cli_beta-1408.dist-info/top_level.txt,sha256=j50aCGsQamLMiQh9PcolDBCAeUJzi9y08e0i9Gqshkk,17
+prelude_cli_beta-1408.dist-info/RECORD,,