Skip to content

Commit f767fbb

Browse files
author
ci.datadog-api-spec
committed
Regenerate client from commit c211405 of spec repo
1 parent 523ae5e commit f767fbb

58 files changed

Lines changed: 4418 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: 883 additions & 0 deletions
Large diffs are not rendered by default.
Lines changed: 118 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,118 @@
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 import (
11+
DatabricksCloudCostMetricsIntegrationDataflowSettings,
12+
)
13+
from datadog_api_client.v2.model.databricks_cluster_logs_integration_dataflow_request import (
14+
DatabricksClusterLogsIntegrationDataflowRequest,
15+
)
16+
from datadog_api_client.v2.model.databricks_data_observability_integration_dataflow_request import (
17+
DatabricksDataObservabilityIntegrationDataflowRequest,
18+
)
19+
from datadog_api_client.v2.model.databricks_data_observability_integration_dataflow_settings import (
20+
DatabricksDataObservabilityIntegrationDataflowSettings,
21+
)
22+
from datadog_api_client.v2.model.databricks_djm_metrics_integration_dataflow_request import (
23+
DatabricksDjmMetricsIntegrationDataflowRequest,
24+
)
25+
from datadog_api_client.v2.model.databricks_djm_metrics_integration_dataflow_settings_request import (
26+
DatabricksDjmMetricsIntegrationDataflowSettingsRequest,
27+
)
28+
from datadog_api_client.v2.model.databricks_gpu_metrics_integration_dataflow_request import (
29+
DatabricksGpuMetricsIntegrationDataflowRequest,
30+
)
31+
from datadog_api_client.v2.model.databricks_integration_account_create_attributes import (
32+
DatabricksIntegrationAccountCreateAttributes,
33+
)
34+
from datadog_api_client.v2.model.databricks_integration_account_create_data import (
35+
DatabricksIntegrationAccountCreateData,
36+
)
37+
from datadog_api_client.v2.model.databricks_integration_account_create_request import (
38+
DatabricksIntegrationAccountCreateRequest,
39+
)
40+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_request import (
41+
DatabricksIntegrationAccountOAuthAuthRequest,
42+
)
43+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_type import (
44+
DatabricksIntegrationAccountOAuthAuthType,
45+
)
46+
from datadog_api_client.v2.model.databricks_integration_account_settings_request import (
47+
DatabricksIntegrationAccountSettingsRequest,
48+
)
49+
from datadog_api_client.v2.model.databricks_integration_dataflows_request import DatabricksIntegrationDataflowsRequest
50+
from datadog_api_client.v2.model.databricks_model_serving_metrics_integration_dataflow_request import (
51+
DatabricksModelServingMetricsIntegrationDataflowRequest,
52+
)
53+
from datadog_api_client.v2.model.databricks_serverless_jobs_integration_dataflow_request import (
54+
DatabricksServerlessJobsIntegrationDataflowRequest,
55+
)
56+
from datadog_api_client.v2.model.integration_account_type import IntegrationAccountType
57+
58+
body = DatabricksIntegrationAccountCreateRequest(
59+
data=DatabricksIntegrationAccountCreateData(
60+
attributes=DatabricksIntegrationAccountCreateAttributes(
61+
authentication=DatabricksIntegrationAccountOAuthAuthRequest(
62+
auth_type=DatabricksIntegrationAccountOAuthAuthType.DATABRICKS_OAUTH,
63+
azure_tenant_id="4d3bac44-0230-4732-9e70-cc00736f0a97",
64+
client_id="5c10654a-b3a3-4840-b37f-f477590c70a0",
65+
client_secret="your-client-secret",
66+
),
67+
dataflows=DatabricksIntegrationDataflowsRequest(
68+
databricks_cloud_cost_metrics=DatabricksCloudCostMetricsIntegrationDataflowRequest(
69+
enabled=True,
70+
settings=DatabricksCloudCostMetricsIntegrationDataflowSettings(
71+
ccm_collect_all_workspaces=True,
72+
),
73+
),
74+
databricks_cluster_logs=DatabricksClusterLogsIntegrationDataflowRequest(
75+
enabled=True,
76+
),
77+
databricks_data_observability=DatabricksDataObservabilityIntegrationDataflowRequest(
78+
enabled=True,
79+
settings=DatabricksDataObservabilityIntegrationDataflowSettings(
80+
do_crawlers_cron="0 * * * *",
81+
sync_system_catalog=True,
82+
),
83+
),
84+
databricks_djm_metrics=DatabricksDjmMetricsIntegrationDataflowRequest(
85+
enabled=True,
86+
settings=DatabricksDjmMetricsIntegrationDataflowSettingsRequest(
87+
dd_api_key_id="fe383f4e-09fc-46bf-8e10-4efdd453a646",
88+
dd_api_key_secret="your-datadog-api-key",
89+
djm_global_init_script_enabled=True,
90+
),
91+
),
92+
databricks_gpu_metrics=DatabricksGpuMetricsIntegrationDataflowRequest(
93+
enabled=True,
94+
),
95+
databricks_model_serving_metrics=DatabricksModelServingMetricsIntegrationDataflowRequest(
96+
enabled=True,
97+
),
98+
databricks_serverless_jobs=DatabricksServerlessJobsIntegrationDataflowRequest(
99+
enabled=True,
100+
),
101+
),
102+
name="My Databricks Workspace",
103+
settings=DatabricksIntegrationAccountSettingsRequest(
104+
system_tables_sql_warehouse_id="aba7c023d4172910",
105+
workspace_url="https://dbc-1234abcd.cloud.databricks.com",
106+
),
107+
),
108+
type=IntegrationAccountType.INTEGRATION_ACCOUNT,
109+
),
110+
)
111+
112+
configuration = Configuration()
113+
configuration.unstable_operations["create_databricks_integration_account"] = True
114+
with ApiClient(configuration) as api_client:
115+
api_instance = DatabricksIntegrationAccountsApi(api_client)
116+
response = api_instance.create_databricks_integration_account(body=body)
117+
118+
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: 119 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,119 @@
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 import (
11+
DatabricksCloudCostMetricsIntegrationDataflowSettings,
12+
)
13+
from datadog_api_client.v2.model.databricks_cluster_logs_integration_dataflow_request import (
14+
DatabricksClusterLogsIntegrationDataflowRequest,
15+
)
16+
from datadog_api_client.v2.model.databricks_data_observability_integration_dataflow_request import (
17+
DatabricksDataObservabilityIntegrationDataflowRequest,
18+
)
19+
from datadog_api_client.v2.model.databricks_data_observability_integration_dataflow_settings import (
20+
DatabricksDataObservabilityIntegrationDataflowSettings,
21+
)
22+
from datadog_api_client.v2.model.databricks_djm_metrics_integration_dataflow_request import (
23+
DatabricksDjmMetricsIntegrationDataflowRequest,
24+
)
25+
from datadog_api_client.v2.model.databricks_djm_metrics_integration_dataflow_settings_request import (
26+
DatabricksDjmMetricsIntegrationDataflowSettingsRequest,
27+
)
28+
from datadog_api_client.v2.model.databricks_gpu_metrics_integration_dataflow_request import (
29+
DatabricksGpuMetricsIntegrationDataflowRequest,
30+
)
31+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_type import (
32+
DatabricksIntegrationAccountOAuthAuthType,
33+
)
34+
from datadog_api_client.v2.model.databricks_integration_account_o_auth_auth_update import (
35+
DatabricksIntegrationAccountOAuthAuthUpdate,
36+
)
37+
from datadog_api_client.v2.model.databricks_integration_account_settings_update import (
38+
DatabricksIntegrationAccountSettingsUpdate,
39+
)
40+
from datadog_api_client.v2.model.databricks_integration_account_update_attributes import (
41+
DatabricksIntegrationAccountUpdateAttributes,
42+
)
43+
from datadog_api_client.v2.model.databricks_integration_account_update_data import (
44+
DatabricksIntegrationAccountUpdateData,
45+
)
46+
from datadog_api_client.v2.model.databricks_integration_account_update_request import (
47+
DatabricksIntegrationAccountUpdateRequest,
48+
)
49+
from datadog_api_client.v2.model.databricks_integration_dataflows_request import DatabricksIntegrationDataflowsRequest
50+
from datadog_api_client.v2.model.databricks_model_serving_metrics_integration_dataflow_request import (
51+
DatabricksModelServingMetricsIntegrationDataflowRequest,
52+
)
53+
from datadog_api_client.v2.model.databricks_serverless_jobs_integration_dataflow_request import (
54+
DatabricksServerlessJobsIntegrationDataflowRequest,
55+
)
56+
from datadog_api_client.v2.model.integration_account_type import IntegrationAccountType
57+
58+
body = DatabricksIntegrationAccountUpdateRequest(
59+
data=DatabricksIntegrationAccountUpdateData(
60+
attributes=DatabricksIntegrationAccountUpdateAttributes(
61+
authentication=DatabricksIntegrationAccountOAuthAuthUpdate(
62+
auth_type=DatabricksIntegrationAccountOAuthAuthType.DATABRICKS_OAUTH,
63+
azure_tenant_id="4d3bac44-0230-4732-9e70-cc00736f0a97",
64+
client_id="5c10654a-b3a3-4840-b37f-f477590c70a0",
65+
client_secret="your-client-secret",
66+
),
67+
dataflows=DatabricksIntegrationDataflowsRequest(
68+
databricks_cloud_cost_metrics=DatabricksCloudCostMetricsIntegrationDataflowRequest(
69+
enabled=True,
70+
settings=DatabricksCloudCostMetricsIntegrationDataflowSettings(
71+
ccm_collect_all_workspaces=True,
72+
),
73+
),
74+
databricks_cluster_logs=DatabricksClusterLogsIntegrationDataflowRequest(
75+
enabled=True,
76+
),
77+
databricks_data_observability=DatabricksDataObservabilityIntegrationDataflowRequest(
78+
enabled=True,
79+
settings=DatabricksDataObservabilityIntegrationDataflowSettings(
80+
do_crawlers_cron="0 * * * *",
81+
sync_system_catalog=True,
82+
),
83+
),
84+
databricks_djm_metrics=DatabricksDjmMetricsIntegrationDataflowRequest(
85+
enabled=True,
86+
settings=DatabricksDjmMetricsIntegrationDataflowSettingsRequest(
87+
dd_api_key_id="fe383f4e-09fc-46bf-8e10-4efdd453a646",
88+
dd_api_key_secret="your-datadog-api-key",
89+
djm_global_init_script_enabled=True,
90+
),
91+
),
92+
databricks_gpu_metrics=DatabricksGpuMetricsIntegrationDataflowRequest(
93+
enabled=True,
94+
),
95+
databricks_model_serving_metrics=DatabricksModelServingMetricsIntegrationDataflowRequest(
96+
enabled=True,
97+
),
98+
databricks_serverless_jobs=DatabricksServerlessJobsIntegrationDataflowRequest(
99+
enabled=True,
100+
),
101+
),
102+
name="My Databricks Workspace",
103+
settings=DatabricksIntegrationAccountSettingsUpdate(
104+
system_tables_sql_warehouse_id="aba7c023d4172910",
105+
workspace_url="https://dbc-1234abcd.cloud.databricks.com",
106+
),
107+
),
108+
id="a9a69c2e-4f8d-4e42-9c1a-2a7a2d3b7c6f",
109+
type=IntegrationAccountType.INTEGRATION_ACCOUNT,
110+
),
111+
)
112+
113+
configuration = Configuration()
114+
configuration.unstable_operations["update_databricks_integration_account"] = True
115+
with ApiClient(configuration) as api_client:
116+
api_instance = DatabricksIntegrationAccountsApi(api_client)
117+
response = api_instance.update_databricks_integration_account(account_id="account_id", body=body)
118+
119+
print(response)

src/datadog_api_client/configuration.py

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

0 commit comments

Comments
 (0)