Skip to content

Commit

Permalink
Add new subgroup for scheduled events (#7469)
Browse files Browse the repository at this point in the history
* add new extension

* uncomment tests

* new recording

* uncomment tests

* update version of RG

* Revert "update version of RG"

This reverts commit 240ce2a.

* Revert "uncomment tests"

This reverts commit d563b9f.

* import live only

* add history
  • Loading branch information
rbandi124 committed Apr 9, 2024
1 parent 4b6912b commit daf267a
Show file tree
Hide file tree
Showing 10 changed files with 4,649 additions and 3,009 deletions.
4 changes: 4 additions & 0 deletions src/maintenance/HISTORY.rst
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,10 @@
Release History
===============
1.6.0
++++++
* Added support for approving scheduled events

1.5.0
++++++
* Added support for maintenance configuration canncellation.
Expand Down
4 changes: 4 additions & 0 deletions src/maintenance/azext_maintenance/aaz/latest/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,3 +4,7 @@
#
# Code generated by aaz-dev-tools
# --------------------------------------------------------------------------------------------

# pylint: skip-file
# flake8: noqa

Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
#
# Code generated by aaz-dev-tools
# --------------------------------------------------------------------------------------------

# pylint: skip-file
# flake8: noqa

from azure.cli.core.aaz import *


@register_command_group(
"maintenance scheduledevent",
is_preview=True,
)
class __CMDGroup(AAZCommandGroup):
"""Scheduled event actions
"""
pass


__all__ = ["__CMDGroup"]
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
#
# Code generated by aaz-dev-tools
# --------------------------------------------------------------------------------------------

# pylint: skip-file
# flake8: noqa

from .__cmd_group import *
from ._acknowledge import *
Original file line number Diff line number Diff line change
@@ -0,0 +1,192 @@
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
#
# Code generated by aaz-dev-tools
# --------------------------------------------------------------------------------------------

# pylint: skip-file
# flake8: noqa

from azure.cli.core.aaz import *


@register_command(
"maintenance scheduledevent acknowledge",
is_preview=True,
)
class Acknowledge(AAZCommand):
"""Post Scheduled Event Acknowledgement
:example: Acknowledge scheduled event of a VM
az --resource-group {resourceGroup} --resource-name {VMname} --scheduled-event-id {GuidEventId} --subscription {subscriptionId}
:example: Acknowledge scheduled event of a VMSS
az --resource-group {resourceGroup} --resource-name {VMSSname} --scheduled-event-id {GuidEventId} --subscription {subscriptionId}
:example: Acknowledge scheduled event of a AvailabilitySet
az --resource-group {resourceGroup} --resource-name {AVSetname} --scheduled-event-id {GuidEventId} --subscription {subscriptionId}
"""

_aaz_info = {
"version": "2023-10-01-preview",
"resources": [
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.compute/{}/{}/providers/microsoft.maintenance/scheduledevents/{}/acknowledge", "2023-10-01-preview"],
]
}

def _handler(self, command_args):
super()._handler(command_args)
self._execute_operations()
return self._output()

_args_schema = None

@classmethod
def _build_arguments_schema(cls, *args, **kwargs):
if cls._args_schema is not None:
return cls._args_schema
cls._args_schema = super()._build_arguments_schema(*args, **kwargs)

# define Arg Group ""

_args_schema = cls._args_schema
_args_schema.resource_group = AAZResourceGroupNameArg(
required=True,
)
_args_schema.resource_name = AAZStrArg(
options=["--resource-name"],
help="Resource Name",
required=True,
id_part="name",
)
_args_schema.resource_type = AAZStrArg(
options=["--resource-type"],
help="Resource type",
required=True,
id_part="type",
)
_args_schema.scheduled_event_id = AAZStrArg(
options=["--scheduled-event-id"],
help="Scheduled Event Id. This is a GUID-formatted string (e.g. 00000000-0000-0000-0000-000000000000)",
required=True,
id_part="child_name_1",
)
return cls._args_schema

def _execute_operations(self):
self.pre_operations()
self.ScheduledEventAcknowledge(ctx=self.ctx)()
self.post_operations()

@register_callback
def pre_operations(self):
pass

@register_callback
def post_operations(self):
pass

def _output(self, *args, **kwargs):
result = self.deserialize_output(self.ctx.vars.instance, client_flatten=True)
return result

class ScheduledEventAcknowledge(AAZHttpOperation):
CLIENT_TYPE = "MgmtClient"

def __call__(self, *args, **kwargs):
request = self.make_request()
session = self.client.send_request(request=request, stream=False, **kwargs)
if session.http_response.status_code in [200]:
return self.on_200(session)

return self.on_error(session.http_response)

@property
def url(self):
return self.client.format_url(
"/subscriptions/{subscriptionId}/resourcegroups/{resourceGroupName}/providers/Microsoft.Compute/{resourceType}/{resourceName}/providers/Microsoft.Maintenance/scheduledevents/{scheduledEventId}/acknowledge",
**self.url_parameters
)

@property
def method(self):
return "POST"

@property
def error_format(self):
return "ODataV4Format"

@property
def url_parameters(self):
parameters = {
**self.serialize_url_param(
"resourceGroupName", self.ctx.args.resource_group,
required=True,
),
**self.serialize_url_param(
"resourceName", self.ctx.args.resource_name,
required=True,
),
**self.serialize_url_param(
"resourceType", self.ctx.args.resource_type,
required=True,
),
**self.serialize_url_param(
"scheduledEventId", self.ctx.args.scheduled_event_id,
required=True,
),
**self.serialize_url_param(
"subscriptionId", self.ctx.subscription_id,
required=True,
),
}
return parameters

@property
def query_parameters(self):
parameters = {
**self.serialize_query_param(
"api-version", "2023-10-01-preview",
required=True,
),
}
return parameters

@property
def header_parameters(self):
parameters = {
**self.serialize_header_param(
"Accept", "application/json",
),
}
return parameters

def on_200(self, session):
data = self.deserialize_http_content(session)
self.ctx.set_var(
"instance",
data,
schema_builder=self._build_schema_on_200
)

_schema_on_200 = None

@classmethod
def _build_schema_on_200(cls):
if cls._schema_on_200 is not None:
return cls._schema_on_200

cls._schema_on_200 = AAZObjectType()

_schema_on_200 = cls._schema_on_200
_schema_on_200.value = AAZStrType()

return cls._schema_on_200


class _AcknowledgeHelper:
"""Helper class for Acknowledge"""


__all__ = ["Acknowledge"]
2 changes: 1 addition & 1 deletion src/maintenance/azext_maintenance/azext_metadata.json
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
{
"azext.isExperimental": true,
"azext.minCliCoreVersion": "2.51.0"
"azext.minCliCoreVersion": "2.56.0"
}
17 changes: 0 additions & 17 deletions src/maintenance/azext_maintenance/manual/commands.py

This file was deleted.

Loading

0 comments on commit daf267a

Please sign in to comment.