Update a Databricks integration account

"""
Update a Databricks integration account returns "OK" response
"""

from os import environ
from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.databricks_integration_api import DatabricksIntegrationApi
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_request import (
    DatabricksCloudCostMetricsIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_settings_request import (
    DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_request import (
    DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_settings_request import (
    DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_request import (
    DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_settings_request import (
    DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest,
)
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_type import (
    DatabricksIntegrationAccountOAuthAuthType,
)
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_update import (
    DatabricksIntegrationAccountOAuthAuthUpdate,
)
from datadog_api_client.v2.model.databricks_integration_account_settings_update import (
    DatabricksIntegrationAccountSettingsUpdate,
)
from datadog_api_client.v2.model.databricks_integration_account_update_attributes import (
    DatabricksIntegrationAccountUpdateAttributes,
)
from datadog_api_client.v2.model.databricks_integration_account_update_data import (
    DatabricksIntegrationAccountUpdateData,
)
from datadog_api_client.v2.model.databricks_integration_account_update_request import (
    DatabricksIntegrationAccountUpdateRequest,
)
from datadog_api_client.v2.model.databricks_integration_dataflows_request import DatabricksIntegrationDataflowsRequest
from datadog_api_client.v2.model.databricks_model_serving_metrics_integration_dataflow_request import (
    DatabricksModelServingMetricsIntegrationDataflowRequest,
)
from datadog_api_client.v2.model.integration_account_type import IntegrationAccountType

body = DatabricksIntegrationAccountUpdateRequest(
    data=DatabricksIntegrationAccountUpdateData(
        attributes=DatabricksIntegrationAccountUpdateAttributes(
            authentication=DatabricksIntegrationAccountOAuthAuthUpdate(
                auth_type=DatabricksIntegrationAccountOAuthAuthType.DATABRICKS_OAUTH,
                azure_tenant_id="4d3bac44-0230-4732-9e70-cc00736f0a97",
                client_id="5c10654a-b3a3-4840-b37f-f477590c70a0",
                client_secret="your-client-secret",
            ),
            dataflows=DatabricksIntegrationDataflowsRequest(
                databricks_cloud_cost_metrics=DatabricksCloudCostMetricsIntegrationDataflowRequest(
                    enabled=True,
                    settings=DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest(
                        ccm_collect_all_workspaces=True,
                    ),
                ),
                databricks_data_observability_jobs_monitoring=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest(
                    enabled=True,
                    settings=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest(
                        dd_api_key_id="fe383f4e-09fc-46bf-8e10-4efdd453a646",
                        dd_api_key_secret="your-datadog-api-key",
                        djm_global_init_script_enabled=True,
                        script_gpum_enabled=True,
                        script_logs_enabled=True,
                        serverless_jobs_enabled=True,
                    ),
                ),
                databricks_data_observability_quality_monitoring=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest(
                    enabled=True,
                    settings=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest(
                        do_crawlers_cron="0 * * * *",
                        sync_system_catalog=True,
                    ),
                ),
                databricks_model_serving_metrics=DatabricksModelServingMetricsIntegrationDataflowRequest(
                    enabled=True,
                ),
            ),
            name="My Databricks Workspace",
            settings=DatabricksIntegrationAccountSettingsUpdate(
                system_tables_sql_warehouse_id="aba7c023d4172910",
                workspace_url="https://dbc-1234abcd.cloud.databricks.com",
            ),
        ),
        id="a9a69c2e-4f8d-4e42-9c1a-2a7a2d3b7c6f",
        type=IntegrationAccountType.INTEGRATION_ACCOUNT,
    ),
)

configuration = Configuration()
configuration.access_token = environ["DD_BEARER_TOKEN"]
configuration.unstable_operations["update_databricks_integration_account"] = True
with ApiClient(configuration) as api_client:
    api_instance = DatabricksIntegrationApi(api_client)
    response = api_instance.update_databricks_integration_account(account_id="account_id", body=body)

    print(response)

# Update a Databricks integration account returns "OK" response

require "datadog_api_client"
DatadogAPIClient.configure do |config|
  config.access_token = ENV["DD_BEARER_TOKEN"]
  config.unstable_operations["v2.update_databricks_integration_account".to_sym] = true
end
api_instance = DatadogAPIClient::V2::DatabricksIntegrationAPI.new

body = DatadogAPIClient::V2::DatabricksIntegrationAccountUpdateRequest.new({
  data: DatadogAPIClient::V2::DatabricksIntegrationAccountUpdateData.new({
    attributes: DatadogAPIClient::V2::DatabricksIntegrationAccountUpdateAttributes.new({
      authentication: DatadogAPIClient::V2::DatabricksIntegrationAccountOAuthAuthUpdate.new({
        auth_type: DatadogAPIClient::V2::DatabricksIntegrationAccountOAuthAuthType::DATABRICKS_OAUTH,
        azure_tenant_id: "4d3bac44-0230-4732-9e70-cc00736f0a97",
        client_id: "5c10654a-b3a3-4840-b37f-f477590c70a0",
        client_secret: "your-client-secret",
      }),
      dataflows: DatadogAPIClient::V2::DatabricksIntegrationDataflowsRequest.new({
        databricks_cloud_cost_metrics: DatadogAPIClient::V2::DatabricksCloudCostMetricsIntegrationDataflowRequest.new({
          enabled: true,
          settings: DatadogAPIClient::V2::DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest.new({
            ccm_collect_all_workspaces: true,
          }),
        }),
        databricks_data_observability_jobs_monitoring: DatadogAPIClient::V2::DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest.new({
          enabled: true,
          settings: DatadogAPIClient::V2::DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest.new({
            dd_api_key_id: "fe383f4e-09fc-46bf-8e10-4efdd453a646",
            dd_api_key_secret: "your-datadog-api-key",
            djm_global_init_script_enabled: true,
            script_gpum_enabled: true,
            script_logs_enabled: true,
            serverless_jobs_enabled: true,
          }),
        }),
        databricks_data_observability_quality_monitoring: DatadogAPIClient::V2::DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest.new({
          enabled: true,
          settings: DatadogAPIClient::V2::DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest.new({
            do_crawlers_cron: "0 * * * *",
            sync_system_catalog: true,
          }),
        }),
        databricks_model_serving_metrics: DatadogAPIClient::V2::DatabricksModelServingMetricsIntegrationDataflowRequest.new({
          enabled: true,
        }),
      }),
      name: "My Databricks Workspace",
      settings: DatadogAPIClient::V2::DatabricksIntegrationAccountSettingsUpdate.new({
        system_tables_sql_warehouse_id: "aba7c023d4172910",
        workspace_url: "https://dbc-1234abcd.cloud.databricks.com",
      }),
    }),
    id: "a9a69c2e-4f8d-4e42-9c1a-2a7a2d3b7c6f",
    type: DatadogAPIClient::V2::IntegrationAccountType::INTEGRATION_ACCOUNT,
  }),
})
p api_instance.update_databricks_integration_account("account_id", body)

// Update a Databricks integration account returns "OK" response

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"os"

	"github.com/DataDog/datadog-api-client-go/v2/api/datadog"
	"github.com/DataDog/datadog-api-client-go/v2/api/datadogV2"
)

func main() {
	body := datadogV2.DatabricksIntegrationAccountUpdateRequest{
		Data: datadogV2.DatabricksIntegrationAccountUpdateData{
			Attributes: datadogV2.DatabricksIntegrationAccountUpdateAttributes{
				Authentication: &datadogV2.DatabricksIntegrationAccountAuthenticationUpdate{
					DatabricksIntegrationAccountOAuthAuthUpdate: &datadogV2.DatabricksIntegrationAccountOAuthAuthUpdate{
						AuthType:      datadogV2.DATABRICKSINTEGRATIONACCOUNTOAUTHAUTHTYPE_DATABRICKS_OAUTH,
						AzureTenantId: *datadog.NewNullableString(datadog.PtrString("4d3bac44-0230-4732-9e70-cc00736f0a97")),
						ClientId:      datadog.PtrString("5c10654a-b3a3-4840-b37f-f477590c70a0"),
						ClientSecret:  datadog.PtrString("your-client-secret"),
					}},
				Dataflows: &datadogV2.DatabricksIntegrationDataflowsRequest{
					DatabricksCloudCostMetrics: &datadogV2.DatabricksCloudCostMetricsIntegrationDataflowRequest{
						Enabled: datadog.PtrBool(true),
						Settings: &datadogV2.DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest{
							CcmCollectAllWorkspaces: datadog.PtrBool(true),
						},
					},
					DatabricksDataObservabilityJobsMonitoring: &datadogV2.DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest{
						Enabled: datadog.PtrBool(true),
						Settings: &datadogV2.DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest{
							DdApiKeyId:                 datadog.PtrString("fe383f4e-09fc-46bf-8e10-4efdd453a646"),
							DdApiKeySecret:             datadog.PtrString("your-datadog-api-key"),
							DjmGlobalInitScriptEnabled: datadog.PtrBool(true),
							ScriptGpumEnabled:          datadog.PtrBool(true),
							ScriptLogsEnabled:          datadog.PtrBool(true),
							ServerlessJobsEnabled:      datadog.PtrBool(true),
						},
					},
					DatabricksDataObservabilityQualityMonitoring: &datadogV2.DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest{
						Enabled: datadog.PtrBool(true),
						Settings: &datadogV2.DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest{
							DoCrawlersCron:    datadog.PtrString("0 * * * *"),
							SyncSystemCatalog: datadog.PtrBool(true),
						},
					},
					DatabricksModelServingMetrics: &datadogV2.DatabricksModelServingMetricsIntegrationDataflowRequest{
						Enabled: datadog.PtrBool(true),
					},
				},
				Name: datadog.PtrString("My Databricks Workspace"),
				Settings: &datadogV2.DatabricksIntegrationAccountSettingsUpdate{
					SystemTablesSqlWarehouseId: datadog.PtrString("aba7c023d4172910"),
					WorkspaceUrl:               datadog.PtrString("https://dbc-1234abcd.cloud.databricks.com"),
				},
			},
			Id:   "a9a69c2e-4f8d-4e42-9c1a-2a7a2d3b7c6f",
			Type: datadogV2.INTEGRATIONACCOUNTTYPE_INTEGRATION_ACCOUNT,
		},
	}
	ctx := datadog.NewDefaultContext(context.Background())
	ctx = context.WithValue(ctx, datadog.ContextAccessToken, os.Getenv("DD_BEARER_TOKEN"))
	configuration := datadog.NewConfiguration()
	configuration.SetUnstableOperationEnabled("v2.UpdateDatabricksIntegrationAccount", true)
	apiClient := datadog.NewAPIClient(configuration)
	api := datadogV2.NewDatabricksIntegrationApi(apiClient)
	resp, r, err := api.UpdateDatabricksIntegrationAccount(ctx, "account_id", body)

	if err != nil {
		fmt.Fprintf(os.Stderr, "Error when calling `DatabricksIntegrationApi.UpdateDatabricksIntegrationAccount`: %v\n", err)
		fmt.Fprintf(os.Stderr, "Full HTTP response: %v\n", r)
	}

	responseContent, _ := json.MarshalIndent(resp, "", "  ")
	fmt.Fprintf(os.Stdout, "Response from `DatabricksIntegrationApi.UpdateDatabricksIntegrationAccount`:\n%s\n", responseContent)
}