migrate latest profile

Author: cxznmhdcxz

src/azure-cli/azure/cli/command_modules/vm/_client_factory.py

@@ -95,10 +95,6 @@ def cf_gallery_image_versions(cli_ctx, _):
     return _compute_client_factory(cli_ctx).gallery_image_versions
 
 
-def cf_gallery_application_version(cli_ctx, *_):
-    return _compute_client_factory(cli_ctx).gallery_application_versions
-
-
 def cf_proximity_placement_groups(cli_ctx, _):
     return _compute_client_factory(cli_ctx).proximity_placement_groups
 

src/azure-cli/azure/cli/command_modules/vm/_help.py

@@ -1110,29 +1110,6 @@
 short-summary: Manage gallery sharing profile
 """
 
-helps['sig gallery-application version'] = """
-    type: group
-    short-summary: Manage gallery application version
-"""
-
-helps['sig gallery-application version create'] = """
-    type: command
-    short-summary: "Create a gallery Application Version."
-    examples:
-      - name: Create a simple gallery Application Version.
-        text: |-
-               az sig gallery-application version create --gallery-name myGalleryName --application-name myGalleryApplicationName -n 1.0.0 -g myResourceGroup --package-file-link https://{myStorageAccount}.blob.core.windows.net/{myStorageContainer}/{myStorageBlob}  --install-command installCommand  --remove-command removeCommand
-"""
-
-helps['sig gallery-application version update'] = """
-    type: command
-    short-summary: "Update a gallery Application Version."
-    examples:
-      - name: Update a simple gallery Application Version.
-        text: |-
-               az sig gallery-application version update --gallery-name myGalleryName --application-name myGalleryApplicationName -n 1.0.0 -g myResourceGroup --package-file-link https://{myStorageAccount}.blob.core.windows.net/{myStorageContainer}/{myStorageBlob} --end-of-life-date "2050-07-01T07:00:00Z"
-"""
-
 helps['snapshot create'] = """
 type: command
 short-summary: Create a snapshot.

src/azure-cli/azure/cli/command_modules/vm/_params.py

@@ -1408,41 +1408,6 @@ def load_arguments(self, _):
 
     # endregion
 
-    # region Gallery applications
-    with self.argument_context('sig gallery-application version') as c:
-        c.argument('gallery_application_name', options_list=['--application-name'],
-                   help='The name of the gallery Application')
-        c.argument('gallery_application_version_name', options_list=['--name', '-n', '--version-name'],
-                   help='The name of the gallery Application Version')
-
-    with self.argument_context('sig gallery-application version create') as c:
-        c.argument('package_file_name', help='The name to assign the downloaded package file on the VM. This is limited to 4096 characters.'
-                                             'If not specified, the package file will be named the same as the Gallery Application name.')
-        c.argument('config_file_name', help='The name to assign the downloaded config file on the VM. This is limited to 4096 characters. '
-                                            'If not specified, the config file will be named the Gallery Application name appended with "_config"')
-
-    for scope in ['create', 'update']:
-        with self.argument_context('sig gallery-application version {}'.format(scope)) as c:
-            c.argument('location', arg_type=get_location_type(self.cli_ctx), required=False,
-                       validator=get_default_location_from_resource_group)
-            c.argument('tags', tags_type)
-            c.argument('package_file_link', help='The mediaLink of the artifact, must be a readable storage page blob.')
-            c.argument('install_command', help='The path and arguments to install the gallery application.')
-            c.argument('remove_command', help='The path and arguments to remove the gallery application.')
-            c.argument('update_command', help='The path and arguments to update the gallery application. If not present,'
-                                              ' then update operation will invoke remove command on the previous version'
-                                              ' and install command on the current version of the gallery application.')
-            c.argument('target_regions', type=validate_file_or_dict, help='The target regions where the Image Version is'
-                       'going to be replicated to. This property is updatable. Expected value: '
-                       'json-string/json-file/@json-file.')
-            c.argument('default_file_link', help='The default configuration link of the artifact, must be a readable storage page blob.')
-            c.argument('exclude_from', arg_type=get_three_state_flag(), help='If set to true, Virtual Machines '
-                       'deployed from the latest version of the Image Definition won\'t use this Image Version.',
-                       arg_group='Publishing Profile')
-            c.argument('end_of_life_date', help='The end of life date of the gallery image version. This property can be '
-                       'used for decommissioning purposes. This property is updatable.', arg_group='Publishing Profile')
-    # endregion
-
     # region Proximity Placement Group
     with self.argument_context('ppg', min_api='2018-04-01') as c:
         c.argument('proximity_placement_group_name', arg_type=name_arg_type, help="The name of the proximity placement group.")

src/azure-cli/azure/cli/command_modules/vm/commands.py

@@ -17,7 +17,7 @@
                                                           cf_shared_gallery_image_version,
                                                           cf_capacity_reservation_groups, cf_capacity_reservations,
                                                           cf_vmss_run_commands,
-                                                          cf_gallery_application_version, cf_restore_point,
+                                                          cf_restore_point,
                                                           cf_restore_point_collection, cf_community_gallery,
                                                           cf_community_gallery_image,
                                                           cf_community_gallery_image_version)
@@ -158,9 +158,8 @@ def load_command_table(self, _):
         operations_tmpl='azure.mgmt.compute.operations#GalleryApplicationsOperations.{}',
     )
 
-    compute_gallery_application_version_sdk = CliCommandType(
+    compute_gallery_application_version_profile = CliCommandType(
         operations_tmpl='azure.mgmt.compute.operations#GalleryApplicationVersionsOperations.{}',
-        client_factory=cf_gallery_application_version,
     )
 
     compute_proximity_placement_groups_sdk = CliCommandType(
@@ -561,9 +560,10 @@ def load_command_table(self, _):
         from .operations.sig_gallery_application import SigGalleryApplicationCreate
         self.command_table['sig gallery-application create'] = SigGalleryApplicationCreate(loader=self)
 
-    with self.command_group('sig gallery-application version', compute_gallery_application_version_sdk, client_factory=cf_gallery_application_version, min_api='2021-07-01', operation_group='gallery_application_versions') as g:
-        g.custom_command('create', 'gallery_application_version_create', supports_no_wait=True)
-        g.custom_command('update', 'gallery_application_version_update', supports_no_wait=True)
+    with self.command_group('sig gallery-application version', compute_gallery_application_version_profile, operation_group='gallery_application_versions'):
+        from .operations.sig_gallery_application_version import SigGalleryApplicationVersionCreate, SiggalleryApplicationversionUpdate
+        self.command_table['sig gallery-application version create'] = SigGalleryApplicationVersionCreate(loader=self)
+        self.command_table['sig gallery-application version update'] = SiggalleryApplicationversionUpdate(loader=self)
 
     with self.command_group('ppg', compute_proximity_placement_groups_sdk, min_api='2018-04-01', client_factory=cf_proximity_placement_groups) as g:
         g.custom_command('create', 'create_proximity_placement_group', validator=process_ppg_create_namespace)

src/azure-cli/azure/cli/command_modules/vm/operations/sig_gallery_application_version.py

@@ -0,0 +1,28 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+# pylint: disable=no-self-use, line-too-long, protected-access, too-few-public-methods, unused-argument
+from knack.log import get_logger
+
+from ..aaz.latest.sig.gallery_application.version import Create as _SigGalleryApplicationVersionCreate, Update as _SigGalleryApplicationVersionUpdate
+
+logger = get_logger(__name__)
+
+
+class SigGalleryApplicationVersionCreate(_SigGalleryApplicationVersionCreate):
+    @classmethod
+    def _build_arguments_schema(cls, *args, **kwargs):
+        args_schema = super()._build_arguments_schema(*args, **kwargs)
+        args_schema.install_command._required = True
+        args_schema.package_file_link._required = True
+        args_schema.remove_command._required = True
+        return args_schema
+
+
+class SiggalleryApplicationversionUpdate(_SigGalleryApplicationVersionUpdate):
+    @classmethod
+    def _build_arguments_schema(cls, *args, **kwargs):
+        args_schema = super()._build_arguments_schema(*args, **kwargs)
+        args_schema.package_file_link._required = True
+        return args_schema