Support for In-App WAF policies operations

docs/datadog_api_client.v2.model.rst

@@ -669,6 +669,104 @@ datadog\_api\_client.v2.model.application\_keys\_type module
    :members:
    :show-inheritance:
 
+datadog\_api\_client.v2.model.application\_security\_policy\_attributes module
+------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_attributes
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_create\_attributes module
+--------------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_create_attributes
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_create\_data module
+--------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_create_data
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_create\_request module
+-----------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_create_request
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_data module
+------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_data
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_list\_response module
+----------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_list_response
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_metadata module
+----------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_metadata
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_response module
+----------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_response
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_rule\_override module
+----------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_rule_override
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_scope module
+-------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_scope
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_type module
+------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_type
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_update\_attributes module
+--------------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_update_attributes
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_update\_data module
+--------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_update_data
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.application\_security\_policy\_update\_request module
+-----------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.application_security_policy_update_request
+   :members:
+   :show-inheritance:
+
 datadog\_api\_client.v2.model.application\_security\_waf\_custom\_rule\_action module
 -------------------------------------------------------------------------------------
 

examples/v2/application-security/CreateApplicationSecurityWafPolicy.py

@@ -0,0 +1,52 @@
+"""
+Create a WAF Policy returns "Created" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.application_security_api import ApplicationSecurityApi
+from datadog_api_client.v2.model.application_security_policy_create_attributes import (
+    ApplicationSecurityPolicyCreateAttributes,
+)
+from datadog_api_client.v2.model.application_security_policy_create_data import ApplicationSecurityPolicyCreateData
+from datadog_api_client.v2.model.application_security_policy_create_request import (
+    ApplicationSecurityPolicyCreateRequest,
+)
+from datadog_api_client.v2.model.application_security_policy_rule_override import ApplicationSecurityPolicyRuleOverride
+from datadog_api_client.v2.model.application_security_policy_scope import ApplicationSecurityPolicyScope
+from datadog_api_client.v2.model.application_security_policy_type import ApplicationSecurityPolicyType
+
+body = ApplicationSecurityPolicyCreateRequest(
+    data=ApplicationSecurityPolicyCreateData(
+        attributes=ApplicationSecurityPolicyCreateAttributes(
+            based_on="recommended",
+            description="Policy applied to internal web applications.",
+            is_default=False,
+            name="Internal Network Policy",
+            protection_presets=[
+                "attack-tools",
+            ],
+            rules=[
+                ApplicationSecurityPolicyRuleOverride(
+                    blocking=False,
+                    enabled=True,
+                    id="rasp-001-002",
+                ),
+            ],
+            scope=[
+                ApplicationSecurityPolicyScope(
+                    env="prod",
+                    service="billing-service",
+                ),
+            ],
+            version=0,
+        ),
+        type=ApplicationSecurityPolicyType.POLICY,
+    ),
+)
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = ApplicationSecurityApi(api_client)
+    response = api_instance.create_application_security_waf_policy(body=body)
+
+    print(response)

examples/v2/application-security/DeleteApplicationSecurityWafPolicy.py

@@ -0,0 +1,13 @@
+"""
+Delete a WAF Policy returns "No Content" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.application_security_api import ApplicationSecurityApi
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = ApplicationSecurityApi(api_client)
+    api_instance.delete_application_security_waf_policy(
+        policy_id="policy_id",
+    )

examples/v2/application-security/GetApplicationSecurityWafPolicy.py

@@ -0,0 +1,15 @@
+"""
+Get a WAF Policy returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.application_security_api import ApplicationSecurityApi
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = ApplicationSecurityApi(api_client)
+    response = api_instance.get_application_security_waf_policy(
+        policy_id="policy_id",
+    )
+
+    print(response)

examples/v2/application-security/ListApplicationSecurityWAFPolicies.py

@@ -0,0 +1,13 @@
+"""
+List all WAF policies returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.application_security_api import ApplicationSecurityApi
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = ApplicationSecurityApi(api_client)
+    response = api_instance.list_application_security_waf_policies()
+
+    print(response)

examples/v2/application-security/UpdateApplicationSecurityWafPolicy.py

@@ -0,0 +1,51 @@
+"""
+Update a WAF Policy returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.application_security_api import ApplicationSecurityApi
+from datadog_api_client.v2.model.application_security_policy_rule_override import ApplicationSecurityPolicyRuleOverride
+from datadog_api_client.v2.model.application_security_policy_scope import ApplicationSecurityPolicyScope
+from datadog_api_client.v2.model.application_security_policy_type import ApplicationSecurityPolicyType
+from datadog_api_client.v2.model.application_security_policy_update_attributes import (
+    ApplicationSecurityPolicyUpdateAttributes,
+)
+from datadog_api_client.v2.model.application_security_policy_update_data import ApplicationSecurityPolicyUpdateData
+from datadog_api_client.v2.model.application_security_policy_update_request import (
+    ApplicationSecurityPolicyUpdateRequest,
+)
+
+body = ApplicationSecurityPolicyUpdateRequest(
+    data=ApplicationSecurityPolicyUpdateData(
+        attributes=ApplicationSecurityPolicyUpdateAttributes(
+            description="Policy applied to internal web applications.",
+            is_default=False,
+            name="Internal Network Policy",
+            protection_presets=[
+                "attack-tools",
+            ],
+            rules=[
+                ApplicationSecurityPolicyRuleOverride(
+                    blocking=False,
+                    enabled=True,
+                    id="rasp-001-002",
+                ),
+            ],
+            scope=[
+                ApplicationSecurityPolicyScope(
+                    env="prod",
+                    service="billing-service",
+                ),
+            ],
+            version=0,
+        ),
+        type=ApplicationSecurityPolicyType.POLICY,
+    ),
+)
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = ApplicationSecurityApi(api_client)
+    response = api_instance.update_application_security_waf_policy(policy_id="policy_id", body=body)
+
+    print(response)