Add group_id action_parameter handler.

Relates to #647.
CrowdStrike · May 9, 2022 · 7864107 · 7864107
1 parent 67e2f57
commit 7864107
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 0 deletions.
diff --git a/src/falconpy/device_control_policies.py b/src/falconpy/device_control_policies.py
@@ -173,6 +173,12 @@ def perform_action(self: object, body: dict = None, parameters: dict = None, **k
         if action_name.lower() in _allowed_actions:
             if not body:
                 body = generic_payload_list(submitted_keywords=kwargs, payload_value="ids")
+                if kwargs.get("group_id", None):
+                    body["action_parameters"] = [{
+                        "name": "group_id",
+                        "value": kwargs.get("group_id", None)
+                    }]
+                # Passing an action_parameters list will override the group_id keyword
                 if kwargs.get("action_parameters", None):
                     body["action_parameters"] = kwargs.get("action_parameters", None)
 

diff --git a/src/falconpy/prevention_policy.py b/src/falconpy/prevention_policy.py
@@ -168,6 +168,12 @@ def perform_policies_action(self: object,
         """
         if not body:
             body = generic_payload_list(submitted_keywords=kwargs, payload_value="ids")
+            if kwargs.get("group_id", None):
+                body["action_parameters"] = [{
+                    "name": "group_id",
+                    "value": kwargs.get("group_id", None)
+                }]
+            # Passing an action_parameters list will override the group_id keyword
             if kwargs.get("action_parameters", None):
                 body["action_parameters"] = kwargs.get("action_parameters", None)
 

diff --git a/src/falconpy/response_policies.py b/src/falconpy/response_policies.py
@@ -161,6 +161,12 @@ def perform_policies_action(self: object,
         """
         if not body:
             body = generic_payload_list(submitted_keywords=kwargs, payload_value="ids")
+            if kwargs.get("group_id", None):
+                body["action_parameters"] = [{
+                    "name": "group_id",
+                    "value": kwargs.get("group_id", None)
+                }]
+            # Passing an action_parameters list will override the group_id keyword
             if kwargs.get("action_parameters", None):
                 body["action_parameters"] = kwargs.get("action_parameters", None)
 

diff --git a/src/falconpy/sensor_update_policy.py b/src/falconpy/sensor_update_policy.py
@@ -308,6 +308,12 @@ def perform_policies_action(self: object,
         if action_name.lower() in _allowed_actions:
             if not body:
                 body = generic_payload_list(submitted_keywords=kwargs, payload_value="ids")
+                if kwargs.get("group_id", None):
+                    body["action_parameters"] = [{
+                        "name": "group_id",
+                        "value": kwargs.get("group_id", None)
+                    }]
+                # Passing an action_parameters list will override the group_id keyword
                 if kwargs.get("action_parameters", None):
                     body["action_parameters"] = kwargs.get("action_parameters", None)