Add public API for security findings automation rules

examples/v2/security-monitoring/CreateSecurityFindingsAutomationDueDateRule.py

@@ -0,0 +1,49 @@
+"""
+Create a due date rule returns "Successfully created the due date rule" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+from datadog_api_client.v2.model.automation_rule_scope import AutomationRuleScope
+from datadog_api_client.v2.model.due_date_from import DueDateFrom
+from datadog_api_client.v2.model.due_date_per_severity_item import DueDatePerSeverityItem
+from datadog_api_client.v2.model.due_date_rule_action import DueDateRuleAction
+from datadog_api_client.v2.model.due_date_rule_attributes_create import DueDateRuleAttributesCreate
+from datadog_api_client.v2.model.due_date_rule_create_request import DueDateRuleCreateRequest
+from datadog_api_client.v2.model.due_date_rule_data_create import DueDateRuleDataCreate
+from datadog_api_client.v2.model.due_date_rule_type import DueDateRuleType
+from datadog_api_client.v2.model.due_date_severity import DueDateSeverity
+from datadog_api_client.v2.model.security_finding_type import SecurityFindingType
+
+body = DueDateRuleCreateRequest(
+    data=DueDateRuleDataCreate(
+        attributes=DueDateRuleAttributesCreate(
+            action=DueDateRuleAction(
+                due_days_per_severity=[
+                    DueDatePerSeverityItem(
+                        due_in_days=7,
+                        severity=DueDateSeverity.CRITICAL,
+                    ),
+                ],
+                due_from=DueDateFrom.FIRST_SEEN,
+            ),
+            enabled=True,
+            name="Example-Security-Monitoring",
+            rule=AutomationRuleScope(
+                finding_types=[
+                    SecurityFindingType.MISCONFIGURATION,
+                ],
+                query="env:staging",
+            ),
+        ),
+        type=DueDateRuleType.DUE_DATE_RULES,
+    ),
+)
+
+configuration = Configuration()
+configuration.unstable_operations["create_security_findings_automation_due_date_rule"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.create_security_findings_automation_due_date_rule(body=body)
+
+    print(response)

examples/v2/security-monitoring/CreateSecurityFindingsAutomationMuteRule.py

@@ -0,0 +1,41 @@
+"""
+Create a mute rule returns "Successfully created the mute rule" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+from datadog_api_client.v2.model.automation_rule_scope import AutomationRuleScope
+from datadog_api_client.v2.model.mute_reason import MuteReason
+from datadog_api_client.v2.model.mute_rule_action import MuteRuleAction
+from datadog_api_client.v2.model.mute_rule_attributes_create import MuteRuleAttributesCreate
+from datadog_api_client.v2.model.mute_rule_create_request import MuteRuleCreateRequest
+from datadog_api_client.v2.model.mute_rule_data_create import MuteRuleDataCreate
+from datadog_api_client.v2.model.mute_rule_type import MuteRuleType
+from datadog_api_client.v2.model.security_finding_type import SecurityFindingType
+
+body = MuteRuleCreateRequest(
+    data=MuteRuleDataCreate(
+        attributes=MuteRuleAttributesCreate(
+            action=MuteRuleAction(
+                reason=MuteReason.RISK_ACCEPTED,
+            ),
+            enabled=True,
+            name="Example-Security-Monitoring",
+            rule=AutomationRuleScope(
+                finding_types=[
+                    SecurityFindingType.MISCONFIGURATION,
+                ],
+                query="env:staging",
+            ),
+        ),
+        type=MuteRuleType.MUTE_RULES,
+    ),
+)
+
+configuration = Configuration()
+configuration.unstable_operations["create_security_findings_automation_mute_rule"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.create_security_findings_automation_mute_rule(body=body)
+
+    print(response)

examples/v2/security-monitoring/CreateSecurityFindingsAutomationTicketCreationRule.py

@@ -0,0 +1,44 @@
+"""
+Create a ticket creation rule returns "Successfully created the ticket creation rule" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+from datadog_api_client.v2.model.automation_rule_scope import AutomationRuleScope
+from datadog_api_client.v2.model.security_finding_type import SecurityFindingType
+from datadog_api_client.v2.model.ticket_creation_rule_action import TicketCreationRuleAction
+from datadog_api_client.v2.model.ticket_creation_rule_attributes_create import TicketCreationRuleAttributesCreate
+from datadog_api_client.v2.model.ticket_creation_rule_create_request import TicketCreationRuleCreateRequest
+from datadog_api_client.v2.model.ticket_creation_rule_data_create import TicketCreationRuleDataCreate
+from datadog_api_client.v2.model.ticket_creation_rule_type import TicketCreationRuleType
+from datadog_api_client.v2.model.ticket_creation_target import TicketCreationTarget
+from uuid import UUID
+
+body = TicketCreationRuleCreateRequest(
+    data=TicketCreationRuleDataCreate(
+        attributes=TicketCreationRuleAttributesCreate(
+            action=TicketCreationRuleAction(
+                max_tickets_per_day=10,
+                project_id=UUID("11111111-1111-1111-1111-111111111111"),
+                target=TicketCreationTarget.JIRA,
+            ),
+            enabled=True,
+            name="Example-Security-Monitoring",
+            rule=AutomationRuleScope(
+                finding_types=[
+                    SecurityFindingType.MISCONFIGURATION,
+                ],
+                query="env:staging",
+            ),
+        ),
+        type=TicketCreationRuleType.TICKET_CREATION_RULES,
+    ),
+)
+
+configuration = Configuration()
+configuration.unstable_operations["create_security_findings_automation_ticket_creation_rule"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.create_security_findings_automation_ticket_creation_rule(body=body)
+
+    print(response)

examples/v2/security-monitoring/DeleteSecurityFindingsAutomationDueDateRule.py

@@ -0,0 +1,18 @@
+"""
+Delete a due date rule returns "Rule successfully deleted." response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+
+# there is a valid "valid_due_date_rule" in the system
+VALID_DUE_DATE_RULE_DATA_ID = environ["VALID_DUE_DATE_RULE_DATA_ID"]
+
+configuration = Configuration()
+configuration.unstable_operations["delete_security_findings_automation_due_date_rule"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    api_instance.delete_security_findings_automation_due_date_rule(
+        rule_id=VALID_DUE_DATE_RULE_DATA_ID,
+    )

examples/v2/security-monitoring/DeleteSecurityFindingsAutomationMuteRule.py

@@ -0,0 +1,18 @@
+"""
+Delete a mute rule returns "Rule successfully deleted." response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+
+# there is a valid "valid_mute_rule" in the system
+VALID_MUTE_RULE_DATA_ID = environ["VALID_MUTE_RULE_DATA_ID"]
+
+configuration = Configuration()
+configuration.unstable_operations["delete_security_findings_automation_mute_rule"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    api_instance.delete_security_findings_automation_mute_rule(
+        rule_id=VALID_MUTE_RULE_DATA_ID,
+    )

examples/v2/security-monitoring/DeleteSecurityFindingsAutomationTicketCreationRule.py

@@ -0,0 +1,18 @@
+"""
+Delete a ticket creation rule returns "Rule successfully deleted." response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+
+# there is a valid "valid_ticket_creation_rule" in the system
+VALID_TICKET_CREATION_RULE_DATA_ID = environ["VALID_TICKET_CREATION_RULE_DATA_ID"]
+
+configuration = Configuration()
+configuration.unstable_operations["delete_security_findings_automation_ticket_creation_rule"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    api_instance.delete_security_findings_automation_ticket_creation_rule(
+        rule_id=VALID_TICKET_CREATION_RULE_DATA_ID,
+    )

examples/v2/security-monitoring/GetSecurityFindingsAutomationDueDateRule.py

@@ -0,0 +1,20 @@
+"""
+Get a due date rule returns "Successfully retrieved the due date rule" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+
+# there is a valid "valid_due_date_rule" in the system
+VALID_DUE_DATE_RULE_DATA_ID = environ["VALID_DUE_DATE_RULE_DATA_ID"]
+
+configuration = Configuration()
+configuration.unstable_operations["get_security_findings_automation_due_date_rule"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.get_security_findings_automation_due_date_rule(
+        rule_id=VALID_DUE_DATE_RULE_DATA_ID,
+    )
+
+    print(response)

examples/v2/security-monitoring/GetSecurityFindingsAutomationMuteRule.py

@@ -0,0 +1,20 @@
+"""
+Get a mute rule returns "Successfully retrieved the mute rule" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+
+# there is a valid "valid_mute_rule" in the system
+VALID_MUTE_RULE_DATA_ID = environ["VALID_MUTE_RULE_DATA_ID"]
+
+configuration = Configuration()
+configuration.unstable_operations["get_security_findings_automation_mute_rule"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.get_security_findings_automation_mute_rule(
+        rule_id=VALID_MUTE_RULE_DATA_ID,
+    )
+
+    print(response)

examples/v2/security-monitoring/GetSecurityFindingsAutomationTicketCreationRule.py

@@ -0,0 +1,20 @@
+"""
+Get a ticket creation rule returns "Successfully retrieved the ticket creation rule" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+
+# there is a valid "valid_ticket_creation_rule" in the system
+VALID_TICKET_CREATION_RULE_DATA_ID = environ["VALID_TICKET_CREATION_RULE_DATA_ID"]
+
+configuration = Configuration()
+configuration.unstable_operations["get_security_findings_automation_ticket_creation_rule"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.get_security_findings_automation_ticket_creation_rule(
+        rule_id=VALID_TICKET_CREATION_RULE_DATA_ID,
+    )
+
+    print(response)

examples/v2/security-monitoring/ListSecurityFindingsAutomationDueDateRules.py

@@ -0,0 +1,14 @@
+"""
+Get all due date rules returns "Successfully retrieved the list of due date rules" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+
+configuration = Configuration()
+configuration.unstable_operations["list_security_findings_automation_due_date_rules"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.list_security_findings_automation_due_date_rules()
+
+    print(response)

examples/v2/security-monitoring/ListSecurityFindingsAutomationMuteRules.py

@@ -0,0 +1,14 @@
+"""
+Get all mute rules returns "Successfully retrieved the list of mute rules" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+
+configuration = Configuration()
+configuration.unstable_operations["list_security_findings_automation_mute_rules"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.list_security_findings_automation_mute_rules()
+
+    print(response)

examples/v2/security-monitoring/ListSecurityFindingsAutomationTicketCreationRules.py

@@ -0,0 +1,14 @@
+"""
+Get all ticket creation rules returns "Successfully retrieved the list of ticket creation rules" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+
+configuration = Configuration()
+configuration.unstable_operations["list_security_findings_automation_ticket_creation_rules"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.list_security_findings_automation_ticket_creation_rules()
+
+    print(response)

examples/v2/security-monitoring/ReorderSecurityFindingsAutomationDueDateRules.py

@@ -0,0 +1,30 @@
+"""
+Reorder due date rules returns "Successfully reordered the due date rules" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+from datadog_api_client.v2.model.due_date_rule_reorder_item import DueDateRuleReorderItem
+from datadog_api_client.v2.model.due_date_rule_reorder_request import DueDateRuleReorderRequest
+from datadog_api_client.v2.model.due_date_rule_type import DueDateRuleType
+
+# there is a valid "valid_due_date_rule" in the system
+VALID_DUE_DATE_RULE_DATA_ID = environ["VALID_DUE_DATE_RULE_DATA_ID"]
+
+body = DueDateRuleReorderRequest(
+    data=[
+        DueDateRuleReorderItem(
+            id=VALID_DUE_DATE_RULE_DATA_ID,
+            type=DueDateRuleType.DUE_DATE_RULES,
+        ),
+    ],
+)
+
+configuration = Configuration()
+configuration.unstable_operations["reorder_security_findings_automation_due_date_rules"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.reorder_security_findings_automation_due_date_rules(body=body)
+
+    print(response)

examples/v2/security-monitoring/ReorderSecurityFindingsAutomationMuteRules.py

@@ -0,0 +1,30 @@
+"""
+Reorder mute rules returns "Successfully reordered the mute rules" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+from datadog_api_client.v2.model.mute_rule_reorder_item import MuteRuleReorderItem
+from datadog_api_client.v2.model.mute_rule_reorder_request import MuteRuleReorderRequest
+from datadog_api_client.v2.model.mute_rule_type import MuteRuleType
+
+# there is a valid "valid_mute_rule" in the system
+VALID_MUTE_RULE_DATA_ID = environ["VALID_MUTE_RULE_DATA_ID"]
+
+body = MuteRuleReorderRequest(
+    data=[
+        MuteRuleReorderItem(
+            id=VALID_MUTE_RULE_DATA_ID,
+            type=MuteRuleType.MUTE_RULES,
+        ),
+    ],
+)
+
+configuration = Configuration()
+configuration.unstable_operations["reorder_security_findings_automation_mute_rules"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.reorder_security_findings_automation_mute_rules(body=body)
+
+    print(response)

examples/v2/security-monitoring/ReorderSecurityFindingsAutomationTicketCreationRules.py

@@ -0,0 +1,30 @@
+"""
+Reorder ticket creation rules returns "Successfully reordered the ticket creation rules" response
+"""
+
+from os import environ
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.security_monitoring_api import SecurityMonitoringApi
+from datadog_api_client.v2.model.ticket_creation_rule_reorder_item import TicketCreationRuleReorderItem
+from datadog_api_client.v2.model.ticket_creation_rule_reorder_request import TicketCreationRuleReorderRequest
+from datadog_api_client.v2.model.ticket_creation_rule_type import TicketCreationRuleType
+
+# there is a valid "valid_ticket_creation_rule" in the system
+VALID_TICKET_CREATION_RULE_DATA_ID = environ["VALID_TICKET_CREATION_RULE_DATA_ID"]
+
+body = TicketCreationRuleReorderRequest(
+    data=[
+        TicketCreationRuleReorderItem(
+            id=VALID_TICKET_CREATION_RULE_DATA_ID,
+            type=TicketCreationRuleType.TICKET_CREATION_RULES,
+        ),
+    ],
+)
+
+configuration = Configuration()
+configuration.unstable_operations["reorder_security_findings_automation_ticket_creation_rules"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = SecurityMonitoringApi(api_client)
+    response = api_instance.reorder_security_findings_automation_ticket_creation_rules(body=body)
+
+    print(response)