1010from datadog_api_client .v2 .model .dataset_response_multi import DatasetResponseMulti
1111from datadog_api_client .v2 .model .dataset_response_single import DatasetResponseSingle
1212from datadog_api_client .v2 .model .dataset_create_request import DatasetCreateRequest
13+ from datadog_api_client .v2 .model .dataset_edit_request import DatasetEditRequest
1314
1415
1516class DatasetsApi :
@@ -67,6 +68,32 @@ def __init__(self, api_client=None):
6768 api_client = api_client ,
6869 )
6970
71+ self ._edit_dataset_endpoint = _Endpoint (
72+ settings = {
73+ "response_type" : (DatasetResponseSingle ,),
74+ "auth" : ["apiKeyAuth" , "appKeyAuth" , "AuthZ" ],
75+ "endpoint_path" : "/api/v2/datasets/{dataset_id}" ,
76+ "operation_id" : "edit_dataset" ,
77+ "http_method" : "PUT" ,
78+ "version" : "v2" ,
79+ },
80+ params_map = {
81+ "dataset_id" : {
82+ "required" : True ,
83+ "openapi_types" : (str ,),
84+ "attribute" : "dataset_id" ,
85+ "location" : "path" ,
86+ },
87+ "body" : {
88+ "required" : True ,
89+ "openapi_types" : (DatasetEditRequest ,),
90+ "location" : "body" ,
91+ },
92+ },
93+ headers_map = {"accept" : ["application/json" ], "content_type" : ["application/json" ]},
94+ api_client = api_client ,
95+ )
96+
7097 self ._get_all_datasets_endpoint = _Endpoint (
7198 settings = {
7299 "response_type" : (DatasetResponseMulti ,),
@@ -114,6 +141,25 @@ def create_dataset(
114141
115142 Create a dataset with the configurations in the request.
116143
144+ **Datasets Constraints**
145+
146+ *
147+ **Tag Limit per Dataset** :
148+
149+ * Each Restricted Dataset supports a maximum of 10 key:value pairs.
150+
151+ *
152+ **Tag Key Rules per Telemetry Type** :
153+
154+ * Only one tag key or attribute may be used to define access within a single telemetry type.
155+ * The same or different tag key may be used across different telemetry types.
156+
157+ *
158+ **Tag Value Uniqueness** :
159+
160+ * Tag values must be unique within a single dataset.
161+ * A tag value used in one dataset cannot be reused in another dataset of the same telemetry type.
162+
117163 :param body: Dataset payload
118164 :type body: DatasetCreateRequest
119165 :rtype: DatasetResponseSingle
@@ -140,6 +186,47 @@ def delete_dataset(
140186
141187 return self ._delete_dataset_endpoint .call_with_http_info (** kwargs )
142188
189+ def edit_dataset (
190+ self ,
191+ dataset_id : str ,
192+ body : DatasetEditRequest ,
193+ ) -> DatasetResponseSingle :
194+ """Edits a dataset.
195+
196+ Edits the dataset associated with the ID.
197+
198+ **Datasets Constraints**
199+
200+ *
201+ **Tag Limit per Dataset** :
202+
203+ * Each Restricted Dataset supports a maximum of 10 key:value pairs.
204+
205+ *
206+ **Tag Key Rules per Telemetry Type** :
207+
208+ * Only one tag key or attribute may be used to define access within a single telemetry type.
209+ * The same or different tag key may be used across different telemetry types.
210+
211+ *
212+ **Tag Value Uniqueness** :
213+
214+ * Tag values must be unique within a single dataset.
215+ * A tag value used in one dataset cannot be reused in another dataset of the same telemetry type.
216+
217+ :param dataset_id: The ID of a defined dataset.
218+ :type dataset_id: str
219+ :param body: Dataset payload
220+ :type body: DatasetEditRequest
221+ :rtype: DatasetResponseSingle
222+ """
223+ kwargs : Dict [str , Any ] = {}
224+ kwargs ["dataset_id" ] = dataset_id
225+
226+ kwargs ["body" ] = body
227+
228+ return self ._edit_dataset_endpoint .call_with_http_info (** kwargs )
229+
143230 def get_all_datasets (
144231 self ,
145232 ) -> DatasetResponseMulti :
0 commit comments