66
77# pylint: disable=protected-access
88
9- from typing import (
10- Union ,
11- TYPE_CHECKING
12- )
9+ from typing import Union , TYPE_CHECKING
1310import datetime
1411import six
1512from msrest import Serializer
2825 DatasourceSqlConnectionString ,
2926 DatasourceDataLakeGen2SharedKey ,
3027 DatasourceServicePrincipal ,
31- DatasourceServicePrincipalInKeyVault
28+ DatasourceServicePrincipalInKeyVault ,
3229)
3330from ._metrics_advisor_key_credential import MetricsAdvisorKeyCredential
3431from ._metrics_advisor_key_credential_policy import MetricsAdvisorKeyCredentialPolicy
32+
3533if TYPE_CHECKING :
3634 from ._generated .models import MetricFeedback
3735
3836
3937def construct_alert_config_dict (update_kwargs ):
4038
4139 if "metricAlertingConfigurations" in update_kwargs :
42- update_kwargs ["metricAlertingConfigurations" ] = [
43- config ._to_generated () for config in
44- update_kwargs ["metricAlertingConfigurations" ]
45- ] if update_kwargs ["metricAlertingConfigurations" ] else None
40+ update_kwargs ["metricAlertingConfigurations" ] = (
41+ [
42+ config ._to_generated ()
43+ for config in update_kwargs ["metricAlertingConfigurations" ]
44+ ]
45+ if update_kwargs ["metricAlertingConfigurations" ]
46+ else None
47+ )
4648
4749 return update_kwargs
4850
4951
5052def construct_detection_config_dict (update_kwargs ):
5153
5254 if "wholeMetricConfiguration" in update_kwargs :
53- update_kwargs ["wholeMetricConfiguration" ] = update_kwargs ["wholeMetricConfiguration" ]._to_generated_patch () \
54- if update_kwargs ["wholeMetricConfiguration" ] else None
55+ update_kwargs ["wholeMetricConfiguration" ] = (
56+ update_kwargs ["wholeMetricConfiguration" ]._to_generated_patch ()
57+ if update_kwargs ["wholeMetricConfiguration" ]
58+ else None
59+ )
5560 if "dimensionGroupOverrideConfigurations" in update_kwargs :
56- update_kwargs ["dimensionGroupOverrideConfigurations" ] = [
57- group ._to_generated () for group in update_kwargs ["dimensionGroupOverrideConfigurations" ]
58- ] if update_kwargs ["dimensionGroupOverrideConfigurations" ] else None
61+ update_kwargs ["dimensionGroupOverrideConfigurations" ] = (
62+ [
63+ group ._to_generated ()
64+ for group in update_kwargs ["dimensionGroupOverrideConfigurations" ]
65+ ]
66+ if update_kwargs ["dimensionGroupOverrideConfigurations" ]
67+ else None
68+ )
5969 if "seriesOverrideConfigurations" in update_kwargs :
60- update_kwargs ["seriesOverrideConfigurations" ] = [
61- series ._to_generated () for series in update_kwargs ["seriesOverrideConfigurations" ]
62- ] if update_kwargs ["seriesOverrideConfigurations" ] else None
70+ update_kwargs ["seriesOverrideConfigurations" ] = (
71+ [
72+ series ._to_generated ()
73+ for series in update_kwargs ["seriesOverrideConfigurations" ]
74+ ]
75+ if update_kwargs ["seriesOverrideConfigurations" ]
76+ else None
77+ )
6378
6479 return update_kwargs
6580
@@ -71,9 +86,16 @@ def construct_hook_dict(update_kwargs, hook_type):
7186 update_kwargs ["hookParameter" ] = {}
7287 update_kwargs ["hookParameter" ]["toList" ] = update_kwargs ["toList" ]
7388 update_kwargs .pop ("toList" )
74- elif hook_type .lower () == "web" \
75- and any (key in update_kwargs for key in
76- ["endpoint" , "username" , "password" , "certificateKey" , "certificatePassword" ]):
89+ elif hook_type .lower () == "web" and any (
90+ key in update_kwargs
91+ for key in [
92+ "endpoint" ,
93+ "username" ,
94+ "password" ,
95+ "certificateKey" ,
96+ "certificatePassword" ,
97+ ]
98+ ):
7799 update_kwargs ["hookType" ] = "Webhook"
78100 update_kwargs ["hookParameter" ] = {}
79101 if "endpoint" in update_kwargs :
@@ -83,38 +105,50 @@ def construct_hook_dict(update_kwargs, hook_type):
83105 if "password" in update_kwargs :
84106 update_kwargs ["hookParameter" ]["password" ] = update_kwargs .pop ("password" )
85107 if "certificateKey" in update_kwargs :
86- update_kwargs ["hookParameter" ]["certificateKey" ] = update_kwargs .pop ("certificateKey" )
108+ update_kwargs ["hookParameter" ]["certificateKey" ] = update_kwargs .pop (
109+ "certificateKey"
110+ )
87111 if "certificatePassword" in update_kwargs :
88- update_kwargs ["hookParameter" ]["certificatePassword" ] = update_kwargs .pop ("certificatePassword" )
112+ update_kwargs ["hookParameter" ]["certificatePassword" ] = update_kwargs .pop (
113+ "certificatePassword"
114+ )
89115
90116 return update_kwargs
91117
92118
93119def construct_data_feed_dict (update_kwargs ):
94120 if "dataStartFrom" in update_kwargs :
95- update_kwargs ["dataStartFrom" ] = Serializer .serialize_iso (update_kwargs ["dataStartFrom" ])
121+ update_kwargs ["dataStartFrom" ] = Serializer .serialize_iso (
122+ update_kwargs ["dataStartFrom" ]
123+ )
96124
97125 if "dataSourceParameter" in update_kwargs :
98- update_kwargs ["authenticationType" ] = update_kwargs ["dataSourceParameter" ].authentication_type
99- update_kwargs ["credentialId" ] = update_kwargs ["dataSourceParameter" ].credential_id
100- update_kwargs ["dataSourceParameter" ] = update_kwargs ["dataSourceParameter" ]._to_generated_patch ()
126+ update_kwargs ["authenticationType" ] = update_kwargs [
127+ "dataSourceParameter"
128+ ].authentication_type
129+ update_kwargs ["credentialId" ] = update_kwargs [
130+ "dataSourceParameter"
131+ ].credential_id
132+ update_kwargs ["dataSourceParameter" ] = update_kwargs [
133+ "dataSourceParameter"
134+ ]._to_generated_patch ()
101135 return update_kwargs
102136
103137
104138def convert_to_generated_data_feed_type (
105- generated_feed_type ,
106- name ,
107- source ,
108- granularity ,
109- schema ,
110- ingestion_settings ,
111- admins = None ,
112- data_feed_description = None ,
113- missing_data_point_fill_settings = None ,
114- rollup_settings = None ,
115- viewers = None ,
116- access_mode = None ,
117- action_link_template = None
139+ generated_feed_type ,
140+ name ,
141+ source ,
142+ granularity ,
143+ schema ,
144+ ingestion_settings ,
145+ admins = None ,
146+ data_feed_description = None ,
147+ missing_data_point_fill_settings = None ,
148+ rollup_settings = None ,
149+ viewers = None ,
150+ access_mode = None ,
151+ action_link_template = None ,
118152):
119153 """Convert input to data feed generated model type
120154
@@ -178,7 +212,9 @@ def convert_to_generated_data_feed_type(
178212 granularity_name = granularity .granularity_type ,
179213 granularity_amount = granularity .custom_granularity_value ,
180214 metrics = [metric ._to_generated () for metric in schema .metrics ],
181- dimension = [dimension ._to_generated () for dimension in schema .dimensions ] if schema .dimensions else None ,
215+ dimension = [dimension ._to_generated () for dimension in schema .dimensions ]
216+ if schema .dimensions
217+ else None ,
182218 timestamp_column = schema .timestamp_column ,
183219 data_start_from = ingestion_settings .ingestion_begin_time ,
184220 max_concurrency = ingestion_settings .data_source_request_concurrency ,
@@ -187,22 +223,28 @@ def convert_to_generated_data_feed_type(
187223 stop_retry_after_in_seconds = ingestion_settings .stop_retry_after ,
188224 data_feed_description = data_feed_description ,
189225 need_rollup = DataFeedRollupType ._to_generated (rollup_settings .rollup_type )
190- if rollup_settings else None ,
226+ if rollup_settings
227+ else None ,
191228 roll_up_method = rollup_settings .rollup_method if rollup_settings else None ,
192229 roll_up_columns = rollup_settings .auto_rollup_group_by_column_names
193- if rollup_settings else None ,
230+ if rollup_settings
231+ else None ,
194232 all_up_identification = rollup_settings .rollup_identification_value
195- if rollup_settings else None ,
233+ if rollup_settings
234+ else None ,
196235 fill_missing_point_type = missing_data_point_fill_settings .fill_type
197- if missing_data_point_fill_settings else None ,
236+ if missing_data_point_fill_settings
237+ else None ,
198238 fill_missing_point_value = missing_data_point_fill_settings .custom_fill_value
199- if missing_data_point_fill_settings else None ,
239+ if missing_data_point_fill_settings
240+ else None ,
200241 viewers = viewers ,
201242 view_mode = access_mode ,
202243 admins = admins ,
203- action_link_template = action_link_template
244+ action_link_template = action_link_template ,
204245 )
205246
247+
206248def convert_to_sub_feedback (feedback ):
207249 # type: (MetricFeedback) -> Union[AnomalyFeedback, ChangePointFeedback, CommentFeedback, PeriodFeedback]
208250 if feedback .feedback_type == "Anomaly" :
@@ -215,6 +257,7 @@ def convert_to_sub_feedback(feedback):
215257 return PeriodFeedback ._from_generated (feedback ) # type: ignore
216258 raise HttpResponseError ("Invalid feedback type returned in the response." )
217259
260+
218261def convert_datetime (date_time ):
219262 # type: (Union[str, datetime.datetime]) -> datetime.datetime
220263 if isinstance (date_time , datetime .datetime ):
@@ -229,6 +272,7 @@ def convert_datetime(date_time):
229272 return datetime .datetime .strptime (date_time , "%Y-%m-%d %H:%M:%S" )
230273 raise TypeError ("Bad datetime type" )
231274
275+
232276def get_authentication_policy (credential ):
233277 authentication_policy = None
234278 if credential is None :
@@ -243,6 +287,7 @@ def get_authentication_policy(credential):
243287
244288 return authentication_policy
245289
290+
246291def convert_to_datasource_credential (datasource_credential ):
247292 if datasource_credential .data_source_credential_type == "AzureSQLConnectionString" :
248293 return DatasourceSqlConnectionString ._from_generated (datasource_credential )
0 commit comments