Add tag pipeline, custom allocation rule and get cloud account by id for AWS/Azure/GCP configs APIs (#2831)

Co-authored-by: ci.datadog-api-spec <[email protected]>

Author: api-clients-generation-pipeline[bot]

.generator/schemas/v2/openapi.yaml

@@ -0,0 +1,86 @@
+"""
+Create arbitrary cost rule returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.cloud_cost_management_api import CloudCostManagementApi
+from datadog_api_client.v2.model.arbitrary_cost_upsert_request import ArbitraryCostUpsertRequest
+from datadog_api_client.v2.model.arbitrary_cost_upsert_request_data import ArbitraryCostUpsertRequestData
+from datadog_api_client.v2.model.arbitrary_cost_upsert_request_data_attributes import (
+    ArbitraryCostUpsertRequestDataAttributes,
+)
+from datadog_api_client.v2.model.arbitrary_cost_upsert_request_data_attributes_costs_to_allocate_items import (
+    ArbitraryCostUpsertRequestDataAttributesCostsToAllocateItems,
+)
+from datadog_api_client.v2.model.arbitrary_cost_upsert_request_data_attributes_strategy import (
+    ArbitraryCostUpsertRequestDataAttributesStrategy,
+)
+from datadog_api_client.v2.model.arbitrary_cost_upsert_request_data_attributes_strategy_based_on_costs_items import (
+    ArbitraryCostUpsertRequestDataAttributesStrategyBasedOnCostsItems,
+)
+from datadog_api_client.v2.model.arbitrary_cost_upsert_request_data_type import ArbitraryCostUpsertRequestDataType
+
+body = ArbitraryCostUpsertRequest(
+    data=ArbitraryCostUpsertRequestData(
+        attributes=ArbitraryCostUpsertRequestDataAttributes(
+            costs_to_allocate=[
+                ArbitraryCostUpsertRequestDataAttributesCostsToAllocateItems(
+                    condition="is",
+                    tag="account_id",
+                    value="123456789",
+                    values=[],
+                ),
+                ArbitraryCostUpsertRequestDataAttributesCostsToAllocateItems(
+                    condition="in",
+                    tag="environment",
+                    value="",
+                    values=[
+                        "production",
+                        "staging",
+                    ],
+                ),
+            ],
+            enabled=True,
+            order_id=1,
+            provider=[
+                "aws",
+                "gcp",
+            ],
+            rule_name="example-arbitrary-cost-rule",
+            strategy=ArbitraryCostUpsertRequestDataAttributesStrategy(
+                allocated_by_tag_keys=[
+                    "team",
+                    "environment",
+                ],
+                based_on_costs=[
+                    ArbitraryCostUpsertRequestDataAttributesStrategyBasedOnCostsItems(
+                        condition="is",
+                        tag="service",
+                        value="web-api",
+                        values=[],
+                    ),
+                    ArbitraryCostUpsertRequestDataAttributesStrategyBasedOnCostsItems(
+                        condition="not in",
+                        tag="team",
+                        value="",
+                        values=[
+                            "legacy",
+                            "deprecated",
+                        ],
+                    ),
+                ],
+                granularity="daily",
+                method="proportional",
+            ),
+            type="shared",
+        ),
+        type=ArbitraryCostUpsertRequestDataType.UPSERT_ARBITRARY_RULE,
+    ),
+)
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = CloudCostManagementApi(api_client)
+    response = api_instance.create_arbitrary_cost_rule(body=body)
+
+    print(response)

docs/datadog_api_client.v2.model.rst

@@ -27,7 +27,6 @@
                 storage_container="dd-storage-container",
             ),
             client_id="1234abcd-1234-abcd-1234-1234abcd1234",
-            is_enabled=True,
             scope="subscriptions/1234abcd-1234-abcd-1234-1234abcd1234",
         ),
         type=AzureUCConfigPostRequestType.AZURE_UC_CONFIG_POST_REQUEST,

examples/v2/cloud-cost-management/CreateArbitraryCostRule.py

@@ -1,5 +1,5 @@
 """
-Create Cloud Cost Management GCP Usage Cost config returns "OK" response
+Create Google Cloud Usage Cost config returns "OK" response
 """
 
 from datadog_api_client import ApiClient, Configuration

examples/v2/cloud-cost-management/CreateCostAzureUCConfigs.py

@@ -0,0 +1,53 @@
+"""
+Create ruleset returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.cloud_cost_management_api import CloudCostManagementApi
+from datadog_api_client.v2.model.create_ruleset_request import CreateRulesetRequest
+from datadog_api_client.v2.model.create_ruleset_request_data import CreateRulesetRequestData
+from datadog_api_client.v2.model.create_ruleset_request_data_attributes import CreateRulesetRequestDataAttributes
+from datadog_api_client.v2.model.create_ruleset_request_data_attributes_rules_items import (
+    CreateRulesetRequestDataAttributesRulesItems,
+)
+from datadog_api_client.v2.model.create_ruleset_request_data_attributes_rules_items_query import (
+    CreateRulesetRequestDataAttributesRulesItemsQuery,
+)
+from datadog_api_client.v2.model.create_ruleset_request_data_attributes_rules_items_query_addition import (
+    CreateRulesetRequestDataAttributesRulesItemsQueryAddition,
+)
+from datadog_api_client.v2.model.create_ruleset_request_data_type import CreateRulesetRequestDataType
+
+body = CreateRulesetRequest(
+    data=CreateRulesetRequestData(
+        attributes=CreateRulesetRequestDataAttributes(
+            enabled=True,
+            rules=[
+                CreateRulesetRequestDataAttributesRulesItems(
+                    enabled=True,
+                    mapping=None,
+                    name="Add Cost Center Tag",
+                    query=CreateRulesetRequestDataAttributesRulesItemsQuery(
+                        addition=CreateRulesetRequestDataAttributesRulesItemsQueryAddition(
+                            key="cost_center",
+                            value="engineering",
+                        ),
+                        case_insensitivity=False,
+                        if_not_exists=True,
+                        query='account_id:"123456789" AND service:"web-api"',
+                    ),
+                    reference_table=None,
+                ),
+            ],
+        ),
+        id="New Ruleset",
+        type=CreateRulesetRequestDataType.CREATE_RULESET,
+    ),
+)
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = CloudCostManagementApi(api_client)
+    response = api_instance.create_ruleset(body=body)
+
+    print(response)

examples/v2/cloud-cost-management/CreateCostGCPUsageCostConfig.py

@@ -0,0 +1,13 @@
+"""
+Delete arbitrary cost rule returns "No Content" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.cloud_cost_management_api import CloudCostManagementApi
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = CloudCostManagementApi(api_client)
+    api_instance.delete_arbitrary_cost_rule(
+        rule_id=123456,
+    )

examples/v2/cloud-cost-management/CreateRuleset.py

@@ -1,5 +1,5 @@
 """
-Delete Cloud Cost Management GCP Usage Cost config returns "No Content" response
+Delete Google Cloud Usage Cost config returns "No Content" response
 """
 
 from datadog_api_client import ApiClient, Configuration

examples/v2/cloud-cost-management/DeleteArbitraryCostRule.py

@@ -0,0 +1,13 @@
+"""
+Delete ruleset returns "No Content" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.cloud_cost_management_api import CloudCostManagementApi
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = CloudCostManagementApi(api_client)
+    api_instance.delete_ruleset(
+        ruleset_id="1c5dae14-237d-4b9a-a515-aa55b3939142",
+    )

examples/v2/cloud-cost-management/DeleteCostGCPUsageCostConfig.py

@@ -0,0 +1,15 @@
+"""
+Get arbitrary cost rule returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.cloud_cost_management_api import CloudCostManagementApi
+
+configuration = Configuration()
+with ApiClient(configuration) as api_client:
+    api_instance = CloudCostManagementApi(api_client)
+    response = api_instance.get_arbitrary_cost_rule(
+        rule_id=123456,
+    )
+
+    print(response)