Skip to content

Commit 190770a

Browse files
authored
run black on search & ma (Azure#20213)
* run black on core, search & ma * update * update
1 parent 48db587 commit 190770a

Some content is hidden

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

47 files changed

+4066
-3019
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__path__ = __import__('pkgutil').extend_path(__path__, __name__) # type: ignore
1+
__path__ = __import__("pkgutil").extend_path(__path__, __name__) # type: ignore
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__path__ = __import__('pkgutil').extend_path(__path__, __name__) # type: ignore
1+
__path__ = __import__("pkgutil").extend_path(__path__, __name__) # type: ignore

sdk/metricsadvisor/azure-ai-metricsadvisor/azure/ai/metricsadvisor/_helpers.py

Lines changed: 91 additions & 46 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,7 @@
66

77
# pylint: disable=protected-access
88

9-
from typing import (
10-
Union,
11-
TYPE_CHECKING
12-
)
9+
from typing import Union, TYPE_CHECKING
1310
import datetime
1411
import six
1512
from msrest import Serializer
@@ -28,38 +25,56 @@
2825
DatasourceSqlConnectionString,
2926
DatasourceDataLakeGen2SharedKey,
3027
DatasourceServicePrincipal,
31-
DatasourceServicePrincipalInKeyVault
28+
DatasourceServicePrincipalInKeyVault,
3229
)
3330
from ._metrics_advisor_key_credential import MetricsAdvisorKeyCredential
3431
from ._metrics_advisor_key_credential_policy import MetricsAdvisorKeyCredentialPolicy
32+
3533
if TYPE_CHECKING:
3634
from ._generated.models import MetricFeedback
3735

3836

3937
def 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

5052
def 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

93119
def 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

104138
def 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+
206248
def 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+
218261
def 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+
232276
def 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+
246291
def 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

Comments
 (0)