Skip to content

Commit 672f558

Browse files
author
ci.datadog-api-spec
committed
Regenerate client from commit 0e22451 of spec repo
1 parent 975dca7 commit 672f558

54 files changed

Lines changed: 4241 additions & 0 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎.generator/schemas/v2/openapi.yaml‎

Lines changed: 873 additions & 0 deletions
Large diffs are not rendered by default.
Lines changed: 103 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,103 @@
1+
"""
2+
Create a Databricks integration account returns "Created" response
3+
"""
4+
5+
from datadog_api_client import ApiClient, Configuration
6+
from datadog_api_client.v2.api.databricks_integration_accounts_api import DatabricksIntegrationAccountsApi
7+
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_request import (
8+
DatabricksCloudCostMetricsIntegrationDataflowRequest,
9+
)
10+
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_settings_request import (
11+
DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest,
12+
)
13+
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_request import (
14+
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest,
15+
)
16+
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_settings_request import (
17+
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest,
18+
)
19+
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_request import (
20+
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest,
21+
)
22+
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_settings_request import (
23+
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest,
24+
)
25+
from datadog_api_client.v2.model.databricks_integration_account_create_attributes import (
26+
DatabricksIntegrationAccountCreateAttributes,
27+
)
28+
from datadog_api_client.v2.model.databricks_integration_account_create_data import (
29+
DatabricksIntegrationAccountCreateData,
30+
)
31+
from datadog_api_client.v2.model.databricks_integration_account_create_request import (
32+
DatabricksIntegrationAccountCreateRequest,
33+
)
34+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_request import (
35+
DatabricksIntegrationAccountOAuthAuthRequest,
36+
)
37+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_type import (
38+
DatabricksIntegrationAccountOAuthAuthType,
39+
)
40+
from datadog_api_client.v2.model.databricks_integration_account_settings_request import (
41+
DatabricksIntegrationAccountSettingsRequest,
42+
)
43+
from datadog_api_client.v2.model.databricks_integration_dataflows_request import DatabricksIntegrationDataflowsRequest
44+
from datadog_api_client.v2.model.databricks_model_serving_metrics_integration_dataflow_request import (
45+
DatabricksModelServingMetricsIntegrationDataflowRequest,
46+
)
47+
from datadog_api_client.v2.model.integration_account_type import IntegrationAccountType
48+
49+
body = DatabricksIntegrationAccountCreateRequest(
50+
data=DatabricksIntegrationAccountCreateData(
51+
attributes=DatabricksIntegrationAccountCreateAttributes(
52+
authentication=DatabricksIntegrationAccountOAuthAuthRequest(
53+
auth_type=DatabricksIntegrationAccountOAuthAuthType.DATABRICKS_OAUTH,
54+
azure_tenant_id="4d3bac44-0230-4732-9e70-cc00736f0a97",
55+
client_id="5c10654a-b3a3-4840-b37f-f477590c70a0",
56+
client_secret="your-client-secret",
57+
),
58+
dataflows=DatabricksIntegrationDataflowsRequest(
59+
databricks_cloud_cost_metrics=DatabricksCloudCostMetricsIntegrationDataflowRequest(
60+
enabled=True,
61+
settings=DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest(
62+
ccm_collect_all_workspaces=True,
63+
),
64+
),
65+
databricks_data_observability_jobs_monitoring=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest(
66+
enabled=True,
67+
settings=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest(
68+
dd_api_key_id="fe383f4e-09fc-46bf-8e10-4efdd453a646",
69+
dd_api_key_secret="your-datadog-api-key",
70+
djm_global_init_script_enabled=True,
71+
script_gpum_enabled=True,
72+
script_logs_enabled=True,
73+
serverless_jobs_enabled=True,
74+
),
75+
),
76+
databricks_data_observability_quality_monitoring=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest(
77+
enabled=True,
78+
settings=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest(
79+
do_crawlers_cron="0 * * * *",
80+
sync_system_catalog=True,
81+
),
82+
),
83+
databricks_model_serving_metrics=DatabricksModelServingMetricsIntegrationDataflowRequest(
84+
enabled=True,
85+
),
86+
),
87+
name="My Databricks Workspace",
88+
settings=DatabricksIntegrationAccountSettingsRequest(
89+
system_tables_sql_warehouse_id="aba7c023d4172910",
90+
workspace_url="https://dbc-1234abcd.cloud.databricks.com",
91+
),
92+
),
93+
type=IntegrationAccountType.INTEGRATION_ACCOUNT,
94+
),
95+
)
96+
97+
configuration = Configuration()
98+
configuration.unstable_operations["create_databricks_integration_account"] = True
99+
with ApiClient(configuration) as api_client:
100+
api_instance = DatabricksIntegrationAccountsApi(api_client)
101+
response = api_instance.create_databricks_integration_account(body=body)
102+
103+
print(response)
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
"""
2+
Delete a Databricks integration account returns "OK" response
3+
"""
4+
5+
from datadog_api_client import ApiClient, Configuration
6+
from datadog_api_client.v2.api.databricks_integration_accounts_api import DatabricksIntegrationAccountsApi
7+
8+
configuration = Configuration()
9+
configuration.unstable_operations["delete_databricks_integration_account"] = True
10+
with ApiClient(configuration) as api_client:
11+
api_instance = DatabricksIntegrationAccountsApi(api_client)
12+
api_instance.delete_databricks_integration_account(
13+
account_id="account_id",
14+
)
Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
"""
2+
Get a Databricks integration account returns "OK" response
3+
"""
4+
5+
from datadog_api_client import ApiClient, Configuration
6+
from datadog_api_client.v2.api.databricks_integration_accounts_api import DatabricksIntegrationAccountsApi
7+
8+
configuration = Configuration()
9+
configuration.unstable_operations["get_databricks_integration_account"] = True
10+
with ApiClient(configuration) as api_client:
11+
api_instance = DatabricksIntegrationAccountsApi(api_client)
12+
response = api_instance.get_databricks_integration_account(
13+
account_id="account_id",
14+
)
15+
16+
print(response)
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
"""
2+
List Databricks integration accounts returns "OK" response
3+
"""
4+
5+
from datadog_api_client import ApiClient, Configuration
6+
from datadog_api_client.v2.api.databricks_integration_accounts_api import DatabricksIntegrationAccountsApi
7+
8+
configuration = Configuration()
9+
configuration.unstable_operations["list_databricks_integration_accounts"] = True
10+
with ApiClient(configuration) as api_client:
11+
api_instance = DatabricksIntegrationAccountsApi(api_client)
12+
response = api_instance.list_databricks_integration_accounts()
13+
14+
print(response)
Lines changed: 104 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,104 @@
1+
"""
2+
Update a Databricks integration account returns "OK" response
3+
"""
4+
5+
from datadog_api_client import ApiClient, Configuration
6+
from datadog_api_client.v2.api.databricks_integration_accounts_api import DatabricksIntegrationAccountsApi
7+
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_request import (
8+
DatabricksCloudCostMetricsIntegrationDataflowRequest,
9+
)
10+
from datadog_api_client.v2.model.databricks_cloud_cost_metrics_integration_dataflow_settings_request import (
11+
DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest,
12+
)
13+
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_request import (
14+
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest,
15+
)
16+
from datadog_api_client.v2.model.databricks_data_observability_jobs_monitoring_integration_dataflow_settings_request import (
17+
DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest,
18+
)
19+
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_request import (
20+
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest,
21+
)
22+
from datadog_api_client.v2.model.databricks_data_observability_quality_monitoring_integration_dataflow_settings_request import (
23+
DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest,
24+
)
25+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_type import (
26+
DatabricksIntegrationAccountOAuthAuthType,
27+
)
28+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_update import (
29+
DatabricksIntegrationAccountOAuthAuthUpdate,
30+
)
31+
from datadog_api_client.v2.model.databricks_integration_account_settings_update import (
32+
DatabricksIntegrationAccountSettingsUpdate,
33+
)
34+
from datadog_api_client.v2.model.databricks_integration_account_update_attributes import (
35+
DatabricksIntegrationAccountUpdateAttributes,
36+
)
37+
from datadog_api_client.v2.model.databricks_integration_account_update_data import (
38+
DatabricksIntegrationAccountUpdateData,
39+
)
40+
from datadog_api_client.v2.model.databricks_integration_account_update_request import (
41+
DatabricksIntegrationAccountUpdateRequest,
42+
)
43+
from datadog_api_client.v2.model.databricks_integration_dataflows_request import DatabricksIntegrationDataflowsRequest
44+
from datadog_api_client.v2.model.databricks_model_serving_metrics_integration_dataflow_request import (
45+
DatabricksModelServingMetricsIntegrationDataflowRequest,
46+
)
47+
from datadog_api_client.v2.model.integration_account_type import IntegrationAccountType
48+
49+
body = DatabricksIntegrationAccountUpdateRequest(
50+
data=DatabricksIntegrationAccountUpdateData(
51+
attributes=DatabricksIntegrationAccountUpdateAttributes(
52+
authentication=DatabricksIntegrationAccountOAuthAuthUpdate(
53+
auth_type=DatabricksIntegrationAccountOAuthAuthType.DATABRICKS_OAUTH,
54+
azure_tenant_id="4d3bac44-0230-4732-9e70-cc00736f0a97",
55+
client_id="5c10654a-b3a3-4840-b37f-f477590c70a0",
56+
client_secret="your-client-secret",
57+
),
58+
dataflows=DatabricksIntegrationDataflowsRequest(
59+
databricks_cloud_cost_metrics=DatabricksCloudCostMetricsIntegrationDataflowRequest(
60+
enabled=True,
61+
settings=DatabricksCloudCostMetricsIntegrationDataflowSettingsRequest(
62+
ccm_collect_all_workspaces=True,
63+
),
64+
),
65+
databricks_data_observability_jobs_monitoring=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowRequest(
66+
enabled=True,
67+
settings=DatabricksDataObservabilityJobsMonitoringIntegrationDataflowSettingsRequest(
68+
dd_api_key_id="fe383f4e-09fc-46bf-8e10-4efdd453a646",
69+
dd_api_key_secret="your-datadog-api-key",
70+
djm_global_init_script_enabled=True,
71+
script_gpum_enabled=True,
72+
script_logs_enabled=True,
73+
serverless_jobs_enabled=True,
74+
),
75+
),
76+
databricks_data_observability_quality_monitoring=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowRequest(
77+
enabled=True,
78+
settings=DatabricksDataObservabilityQualityMonitoringIntegrationDataflowSettingsRequest(
79+
do_crawlers_cron="0 * * * *",
80+
sync_system_catalog=True,
81+
),
82+
),
83+
databricks_model_serving_metrics=DatabricksModelServingMetricsIntegrationDataflowRequest(
84+
enabled=True,
85+
),
86+
),
87+
name="My Databricks Workspace",
88+
settings=DatabricksIntegrationAccountSettingsUpdate(
89+
system_tables_sql_warehouse_id="aba7c023d4172910",
90+
workspace_url="https://dbc-1234abcd.cloud.databricks.com",
91+
),
92+
),
93+
id="a9a69c2e-4f8d-4e42-9c1a-2a7a2d3b7c6f",
94+
type=IntegrationAccountType.INTEGRATION_ACCOUNT,
95+
),
96+
)
97+
98+
configuration = Configuration()
99+
configuration.unstable_operations["update_databricks_integration_account"] = True
100+
with ApiClient(configuration) as api_client:
101+
api_instance = DatabricksIntegrationAccountsApi(api_client)
102+
response = api_instance.update_databricks_integration_account(account_id="account_id", body=body)
103+
104+
print(response)

‎src/datadog_api_client/configuration.py‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -641,6 +641,11 @@ def __init__(
641641
"v2.update_incident_user_defined_field": False,
642642
"v2.update_incident_user_defined_role": False,
643643
"v2.update_timestamp_override": False,
644+
"v2.create_databricks_integration_account": False,
645+
"v2.delete_databricks_integration_account": False,
646+
"v2.get_databricks_integration_account": False,
647+
"v2.list_databricks_integration_accounts": False,
648+
"v2.update_databricks_integration_account": False,
644649
"v2.create_elastic_cloud_integration_account": False,
645650
"v2.delete_elastic_cloud_integration_account": False,
646651
"v2.get_elastic_cloud_integration_account": False,

0 commit comments

Comments
 (0)