[Scheduled-Query] update api version (Azure#3399)

* Update venderSDK

* support new parameters

* support new parameters

* fix linter issue

* Update HISTORY.rst

* use auto-mitigate instead of disable-auto-mitigate

* update tests

* update tests

* update tests

Author: kairu-ms

src/scheduled-query/HISTORY.rst

@@ -6,6 +6,9 @@ Release History
 ++++++
 * Support query placeholder for `--condition` parameter.
 * Add `--condition-query` parameter to support query placeholder.
+* Add `--skip-query-validation` parameter
+* Add `--check-ws-alerts-storage` parameter
+* [Breaking Change] the default value of `--mute-actions-duration` is changed to None, as alert `--auto-mitigate` is supported
 
 0.2.2
 ++++++

src/scheduled-query/azext_scheduled_query/_client_factory.py

@@ -6,5 +6,5 @@
 
 def cf_scheduled_query(cli_ctx, *_):
     from azure.cli.core.commands.client_factory import get_mgmt_service_client
-    from .vendored_sdks.azure_mgmt_scheduled_query._monitor_client import MonitorClient
-    return get_mgmt_service_client(cli_ctx, MonitorClient).scheduled_query_rules
+    from .vendored_sdks.azure_mgmt_scheduled_query._monitor_management_client import MonitorManagementClient
+    return get_mgmt_service_client(cli_ctx, MonitorManagementClient).scheduled_query_rules

src/scheduled-query/azext_scheduled_query/_params.py

@@ -4,7 +4,7 @@
 # --------------------------------------------------------------------------------------------
 # pylint: disable=line-too-long
 
-from azure.cli.core.commands.parameters import tags_type, get_three_state_flag
+from azure.cli.core.commands.parameters import tags_type, get_three_state_flag, get_enum_type
 from azure.cli.command_modules.monitor.actions import get_period_type
 from azure.cli.command_modules.monitor.validators import get_action_group_validator
 from knack.arguments import CLIArgumentType
@@ -23,6 +23,7 @@ def load_arguments(self, _):
         c.argument('severity', type=int, help='Severity of the alert from 0 (critical) to 4 (verbose).')
         c.argument('window_size', type=get_period_type(), help='Time over which to aggregate metrics in "##h##m##s" format.')
         c.argument('evaluation_frequency', type=get_period_type(), help='Frequency with which to evaluate the rule in "##h##m##s" format.')
+        c.argument('display_name', help='The display name of the alert rule')
         c.argument('condition', options_list=['--condition'], action=ScheduleQueryConditionAction, nargs='+')
         c.argument('condition_query', options_list=['--condition-query'], nargs='+', action=ScheduleQueryConditionQueryAction, help='Query deteils to replace the placeholders in `--condition` argument.')
         c.argument('description', help='Free-text description of the rule.')
@@ -36,3 +37,6 @@ def load_arguments(self, _):
                    options_list=['--mute-actions-duration', '--mad'],
                    help='Mute actions for the chosen period of time (in ISO 8601 duration format) after the alert is fired.')
         c.argument('actions', options_list=['--action', '-a'], action=ScheduleQueryAddAction, nargs='+', validator=get_action_group_validator('actions'))
+        c.argument('auto_mitigate', arg_type=get_three_state_flag(), help='The flag that indicates whether the alert should be automatically resolved or not. The default is true.')
+        c.argument('skip_query_validation', arg_type=get_three_state_flag(), help='The flag which indicates whether the provided query should be validated or not.')
+        c.argument('check_workspace_alerts_storage', options_list=['--check-ws-alerts-storage', '--cwas'], arg_type=get_three_state_flag(), help="The flag which indicates whether this scheduled query rule should be stored in the customer's storage.")

src/scheduled-query/azext_scheduled_query/custom.py

@@ -26,12 +26,14 @@ def _build_criteria(condition, condition_query):
 
 
 def create_scheduled_query(client, resource_group_name, rule_name, scopes, condition, condition_query=None,
-                           disabled=False, description=None, tags=None, location=None,
+                           disabled=False, description=None, tags=None, location=None, display_name=None,
+                           auto_mitigate=True, skip_query_validation=False, check_workspace_alerts_storage=False,
                            actions=None, severity=2, window_size='5m', evaluation_frequency='5m',
-                           target_resource_type=None, mute_actions_duration='PT30M'):
+                           target_resource_type=None, mute_actions_duration=None):
     from .vendored_sdks.azure_mgmt_scheduled_query.models import ScheduledQueryRuleResource
     criteria = _build_criteria(condition, condition_query)
     kwargs = {
+        'display_name': display_name,
         'description': description,
         'severity': severity,
         'enabled': not disabled,
@@ -43,7 +45,10 @@ def create_scheduled_query(client, resource_group_name, rule_name, scopes, condi
         'actions': actions,
         'tags': tags,
         'location': location,
-        'mute_actions_duration': mute_actions_duration
+        'mute_actions_duration': mute_actions_duration,
+        'check_workspace_alerts_storage_configured': check_workspace_alerts_storage,
+        'skip_query_validation': skip_query_validation,
+        'auto_mitigate': auto_mitigate,
     }
     return client.create_or_update(resource_group_name, rule_name, ScheduledQueryRuleResource(**kwargs))
 
@@ -54,12 +59,19 @@ def list_scheduled_query(client, resource_group_name=None):
     return client.list_by_subscription()
 
 
-def update_scheduled_query(cmd, instance, tags=None, disabled=False, condition=None, condition_query=None,
-                           description=None, actions=None, severity=None, window_size=None,
-                           evaluation_frequency=None, mute_actions_duration=None):
+def update_scheduled_query(cmd, instance, tags=None, disabled=None, condition=None, condition_query=None,
+                           description=None, actions=None, severity=None, window_size=None, display_name=None,
+                           auto_mitigate=None, evaluation_frequency=None, mute_actions_duration=None,
+                           skip_query_validation=None, check_workspace_alerts_storage=None):
     with cmd.update_context(instance) as c:
         c.set_param('tags', tags)
-        c.set_param('enabled', not disabled)
+        c.set_param('display_name', display_name)
+        if disabled is not None:
+            c.set_param('enabled', not disabled)
+        if auto_mitigate is not None:
+            c.set_param('auto_mitigate', auto_mitigate)
+        c.set_param('skip_query_validation', skip_query_validation)
+        c.set_param('check_workspace_alerts_storage_configured', check_workspace_alerts_storage)
         c.set_param('description', description)
         c.set_param('actions', actions)
         c.set_param('severity', severity)