chore: minor refactoring and formatting

Author: mihirvala08

src/secops/chronicle/client.py

@@ -288,7 +288,7 @@
     list_retrohunts as _list_retrohunts,
 )
 from secops.chronicle.rule_set import (
-    batch_update_curated_rule_set_deployments as _batch_update_curated_rule_set_deployments, #pylint: disable=line-too-long
+    batch_update_curated_rule_set_deployments as _batch_update_curated_rule_set_deployments,  # pylint: disable=line-too-long
 )
 from secops.chronicle.rule_set import get_curated_rule as _get_curated_rule
 from secops.chronicle.rule_set import (
@@ -304,7 +304,7 @@
     get_curated_rule_set_deployment as _get_curated_rule_set_deployment,
 )
 from secops.chronicle.rule_set import (
-    get_curated_rule_set_deployment_by_name as _get_curated_rule_set_deployment_by_name, #pylint: disable=line-too-long
+    get_curated_rule_set_deployment_by_name as _get_curated_rule_set_deployment_by_name,  # pylint: disable=line-too-long
 )
 from secops.chronicle.rule_set import (
     list_curated_rule_set_categories as _list_curated_rule_set_categories,

src/secops/chronicle/models.py

@@ -105,6 +105,7 @@ class IntegrationType(str, Enum):
     RESPONSE = "RESPONSE"
     EXTENSION = "EXTENSION"
 
+
 class IntegrationParamType(str, Enum):
     """Parameter types for Chronicle SOAR integration functions."""
 
@@ -127,6 +128,7 @@ class IntegrationParamType(str, Enum):
     FILTER_LIST = "FILTER_LIST"
     NUMERICAL_VALUES = "NUMERICAL_VALUES"
 
+
 @dataclass
 class IntegrationParam:
     """A parameter definition for a Chronicle SOAR integration.

src/secops/chronicle/soar/integration/integrations.py

@@ -599,14 +599,14 @@ def update_integration(
     """
     body, params = build_patch_body(
         field_map=[
-            ("displayName", "display_name", display_name),
+            ("display_name", "displayName", display_name),
             ("description", "description", description),
-            ("imageBase64", "image_base64", image_base64),
-            ("svgIcon", "svg_icon", svg_icon),
-            ("pythonVersion", "python_version", python_version),
+            ("image_base64", "imageBase64", image_base64),
+            ("svg_icon", "svgIcon", svg_icon),
+            ("python_version", "pythonVersion", python_version),
             ("parameters", "parameters", parameters),
             ("categories", "categories", categories),
-            ("integrationType", "integration_type", integration_type),
+            ("integration_type", "integrationType", integration_type),
             ("staging", "staging", staging),
         ],
         update_mask=update_mask,

src/secops/cli/commands/integration/integration.py

@@ -14,6 +14,7 @@
 #
 """Google SecOps CLI integration commands"""
 
+import json
 import sys
 
 from secops.chronicle.models import (
@@ -127,6 +128,18 @@ def setup_integrations_command(subparsers):
         help="Integration type",
         dest="integration_type",
     )
+    create_parser.add_argument(
+        "--parameters",
+        type=str,
+        help="JSON string representing a list of integration parameters",
+        dest="parameters",
+    )
+    create_parser.add_argument(
+        "--categories",
+        type=str,
+        help="Comma-separated list of categories for the integration",
+        dest="categories",
+    )
     create_parser.set_defaults(func=handle_integration_create_command)
 
     # download command
@@ -412,6 +425,18 @@ def setup_integrations_command(subparsers):
         help="Integration type",
         dest="integration_type",
     )
+    update_parser.add_argument(
+        "--parameters",
+        type=str,
+        help="JSON string representing a list of integration parameters",
+        dest="parameters",
+    )
+    update_parser.add_argument(
+        "--categories",
+        type=str,
+        help="Comma-separated list of categories for the integration",
+        dest="categories",
+    )
     update_parser.add_argument(
         "--staging",
         action="store_true",
@@ -489,6 +514,18 @@ def setup_integrations_command(subparsers):
         help="Integration type",
         dest="integration_type",
     )
+    update_custom_parser.add_argument(
+        "--parameters",
+        type=str,
+        help="JSON string representing a list of integration parameters",
+        dest="parameters",
+    )
+    update_custom_parser.add_argument(
+        "--categories",
+        type=str,
+        help="Comma-separated list of categories for the integration",
+        dest="categories",
+    )
     update_custom_parser.add_argument(
         "--staging",
         action="store_true",
@@ -580,6 +617,16 @@ def handle_integration_create_command(args, chronicle):
                 if args.integration_type
                 else None
             ),
+            parameters=(
+                json.loads(args.parameters)
+                if getattr(args, "parameters", None)
+                else None
+            ),
+            categories=(
+                [c.strip() for c in args.categories.split(",")]
+                if getattr(args, "categories", None)
+                else None
+            ),
         )
         output_formatter(out, getattr(args, "output", "json"))
     except Exception as e:  # pylint: disable=broad-exception-caught
@@ -590,7 +637,7 @@ def handle_integration_create_command(args, chronicle):
 def handle_integration_download_command(args, chronicle):
     """Handle download integration command"""
     try:
-        zip_bytes = chronicle.download_integration(
+        zip_bytes = chronicle.soar.download_integration(
             integration_name=args.integration_id,
         )
         with open(args.output_file, "wb") as f:
@@ -622,7 +669,7 @@ def handle_download_integration_dependency_command(args, chronicle):
 def handle_export_integration_items_command(args, chronicle):
     """Handle export integration items command"""
     try:
-        zip_bytes = chronicle.export_integration_items(
+        zip_bytes = chronicle.soar.export_integration_items(
             integration_name=args.integration_id,
             actions=args.actions,
             jobs=args.jobs,
@@ -736,7 +783,17 @@ def handle_update_integration_command(args, chronicle):
                 if args.integration_type
                 else None
             ),
-            staging=args.staging or None,
+            parameters=(
+                json.loads(args.parameters)
+                if getattr(args, "parameters", None)
+                else None
+            ),
+            categories=(
+                [c.strip() for c in args.categories.split(",")]
+                if getattr(args, "categories", None)
+                else None
+            ),
+            staging=args.staging if "staging" in vars(args) else None,
             dependencies_to_remove=args.dependencies_to_remove,
             update_mask=args.update_mask,
         )
@@ -765,6 +822,16 @@ def handle_updated_custom_integration_command(args, chronicle):
                 if args.integration_type
                 else None
             ),
+            parameters=(
+                json.loads(args.parameters)
+                if getattr(args, "parameters", None)
+                else None
+            ),
+            categories=(
+                [c.strip() for c in args.categories.split(",")]
+                if getattr(args, "categories", None)
+                else None
+            ),
             staging=args.staging or None,
             dependencies_to_remove=args.dependencies_to_remove,
             update_mask=args.update_mask,