Skip to content

Commit

Permalink
[Compute] BREAKING CHANGE: Migrate commands using Code Gen V2 (Azure#…
Browse files Browse the repository at this point in the history
  • Loading branch information
yanzhudd authored Dec 4, 2024
1 parent 464a79c commit 2c037ba
Show file tree
Hide file tree
Showing 76 changed files with 20,737 additions and 8,532 deletions.
36 changes: 0 additions & 36 deletions src/azure-cli/azure/cli/command_modules/vm/_help.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,32 +77,6 @@
az disk create -g MyResourceGroup -n MyDisk --upload-size-bytes 20972032 --upload-type UploadWithSecurityData --security-type TrustedLaunch --hyper-v-generation V2
"""

helps['disk grant-access'] = """
type: command
short-summary: Grant a resource access to a managed disk.
examples:
- name: Grant a resource read access to a managed disk. (autogenerated)
text: |
az disk grant-access --access-level Read --duration-in-seconds 3600 --name MyManagedDisk --resource-group MyResourceGroup
crafted: true
- name: Grant a resource read access to a disk to generate access SAS and security data access SAS
text: |
az disk grant-access --access-level Read --duration-in-seconds 3600 --name MyDisk --resource-group MyResourceGroup --secure-vm-guest-state-sas
"""

helps['disk update'] = """
type: command
short-summary: Update a managed disk.
examples:
- name: Update a managed disk and associate it with a disk access resource.
text: |
az disk update --name MyManagedDisk --resource-group MyResourceGroup --network-access-policy AllowPrivate --disk-access MyDiskAccessID
- name: Update a managed disk. (autogenerated)
text: |
az disk update --name MyManagedDisk --resource-group MyResourceGroup --size-gb 20
crafted: true
"""

helps['disk wait'] = """
type: command
short-summary: Place the CLI in a waiting state until a condition of a managed disk is met.
Expand Down Expand Up @@ -696,16 +670,6 @@
az image builder show -n mytemplate -g my-group
"""

helps['image update'] = """
type: command
short-summary: Update custom VM images.
examples:
- name: Add or update tags.
text: az image update -n ImageName -g ResourceGroup --tags tag1=val1 tag2=val2
- name: Remove all tags.
text: az image update -n ImageName -g resourceGroup --tags
"""

helps['ppg create'] = """
type: command
short-summary: Create a proximity placement group
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,11 @@
# flake8: noqa

from .__cmd_group import *
from ._create import *
from ._delete import *
from ._grant_access import *
from ._list import *
from ._revoke_access import *
from ._show import *
from ._update import *
from ._wait import *
1,003 changes: 1,003 additions & 0 deletions src/azure-cli/azure/cli/command_modules/vm/aaz/latest/disk/_create.py

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
@@ -0,0 +1,231 @@
# --------------------------------------------------------------------------------------------
# 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(
"disk grant-access",
)
class GrantAccess(AAZCommand):
"""Grant a resource access to a managed disk.
:example: Grant a resource read access to a managed disk.
az disk grant-access --access-level Read --duration-in-seconds 3600 --name MyManagedDisk --resource-group MyResourceGroup
:example: Grant a resource read access to a disk to generate access SAS and security data access SAS
az disk grant-access --access-level Read --duration-in-seconds 3600 --name MyDisk --resource-group MyResourceGroup --secure-vm-guest-state-sas
"""

_aaz_info = {
"version": "2023-04-02",
"resources": [
["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.compute/disks/{}/begingetaccess", "2023-04-02"],
]
}

AZ_SUPPORT_NO_WAIT = True

def _handler(self, command_args):
super()._handler(command_args)
return self.build_lro_poller(self._execute_operations, 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.disk_name = AAZStrArg(
options=["-n", "--name", "--disk-name"],
help="The name of the managed disk that is being created. The name can't be changed after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters.",
required=True,
id_part="name",
)
_args_schema.resource_group = AAZResourceGroupNameArg(
required=True,
)

# define Arg Group "GrantAccessData"

_args_schema = cls._args_schema
_args_schema.access_level = AAZStrArg(
options=["--access", "--access-level"],
arg_group="GrantAccessData",
help="Access level.",
required=True,
default="Read",
enum={"None": "None", "Read": "Read", "Write": "Write"},
)
_args_schema.duration_in_seconds = AAZIntArg(
options=["--duration-in-seconds"],
arg_group="GrantAccessData",
help="Time duration in seconds until the SAS access expires.",
required=True,
)
_args_schema.secure_vm_guest_state_sas = AAZBoolArg(
options=["-s", "--secure-vm-guest-state-sas"],
arg_group="GrantAccessData",
help="Get SAS on managed disk with VM guest state. It will be used by default when the create option of disk is 'secureOSUpload'",
)
return cls._args_schema

def _execute_operations(self):
self.pre_operations()
yield self.DisksGrantAccess(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 DisksGrantAccess(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 [202]:
return self.client.build_lro_polling(
self.ctx.args.no_wait,
session,
self.on_200,
self.on_error,
lro_options={"final-state-via": "location"},
path_format_arguments=self.url_parameters,
)
if session.http_response.status_code in [200]:
return self.client.build_lro_polling(
self.ctx.args.no_wait,
session,
self.on_200,
self.on_error,
lro_options={"final-state-via": "location"},
path_format_arguments=self.url_parameters,
)

return self.on_error(session.http_response)

@property
def url(self):
return self.client.format_url(
"/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess",
**self.url_parameters
)

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

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

@property
def url_parameters(self):
parameters = {
**self.serialize_url_param(
"diskName", self.ctx.args.disk_name,
required=True,
),
**self.serialize_url_param(
"resourceGroupName", self.ctx.args.resource_group,
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-04-02",
required=True,
),
}
return parameters

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

@property
def content(self):
_content_value, _builder = self.new_content_builder(
self.ctx.args,
typ=AAZObjectType,
typ_kwargs={"flags": {"required": True, "client_flatten": True}}
)
_builder.set_prop("access", AAZStrType, ".access_level", typ_kwargs={"flags": {"required": True}})
_builder.set_prop("durationInSeconds", AAZIntType, ".duration_in_seconds", typ_kwargs={"flags": {"required": True}})
_builder.set_prop("getSecureVMGuestStateSAS", AAZBoolType, ".secure_vm_guest_state_sas")

return self.serialize_content(_content_value)

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.access_sas = AAZStrType(
serialized_name="accessSAS",
flags={"read_only": True},
)
_schema_on_200.security_data_access_sas = AAZStrType(
serialized_name="securityDataAccessSAS",
flags={"read_only": True},
)

return cls._schema_on_200


class _GrantAccessHelper:
"""Helper class for GrantAccess"""


__all__ = ["GrantAccess"]
Original file line number Diff line number Diff line change
Expand Up @@ -256,6 +256,7 @@ def _build_schema_on_200(cls):
)
properties.property_updates_in_progress = AAZObjectType(
serialized_name="propertyUpdatesInProgress",
flags={"read_only": True},
)
properties.provisioning_state = AAZStrType(
serialized_name="provisioningState",
Expand Down
Loading

0 comments on commit 2c037ba

Please sign in to comment.