app auto upgrade

Author: Jing-song

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

@@ -1053,6 +1053,7 @@ def load_arguments(self, _):
                        'It should have the same number of items as the application version ids. Null is available for a application '
                        'which does not have a configuration override.')
             c.argument('treat_deployment_as_failure', nargs='*', help="Space-separated list of true or false corresponding to the application version ids. If set to true, failure to install or update gallery application version operation will fail this operation")
+            c.argument('enable_automatic_upgrade', options_list=['--enable-automatic-upgrade', '--enable-auto-upgrade'], arg_type=get_three_state_flag(), help='If set to true, when a new Gallery Application version is available in PIR/SIG, it will be automatically updated for the VM/VMSS')
 
     for scope in ['vm application list', 'vmss application list']:
         with self.argument_context(scope) as c:

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

@@ -5734,7 +5734,7 @@ def show_capacity_reservation_group(client, resource_group_name, capacity_reserv
                       expand=expand)
 
 
-def set_vm_applications(cmd, vm_name, resource_group_name, application_version_ids, order_applications=False, application_configuration_overrides=None, treat_deployment_as_failure=None, no_wait=False):
+def set_vm_applications(cmd, vm_name, resource_group_name, application_version_ids, order_applications=False, application_configuration_overrides=None, treat_deployment_as_failure=None, enable_automatic_upgrade=None, no_wait=False):
     from .aaz.latest.vm import Update as _VMUpdate
 
     class SetVMApplications(_VMUpdate):
@@ -5743,7 +5743,7 @@ def pre_operations(self):
             args.no_wait = no_wait
 
         def pre_instance_update(self, instance):
-            instance.properties.application_profile.gallery_applications = [{"package_reference_id": avid} for avid in application_version_ids]
+            instance.properties.application_profile.gallery_applications = [{"package_reference_id": avid, 'enable_automatic_upgrade': enable_automatic_upgrade} for avid in application_version_ids]
 
             if order_applications:
                 index = 1
@@ -5796,15 +5796,15 @@ def list_vm_applications(cmd, vm_name, resource_group_name):
     return vm.get("applicationProfile", {})
 
 
-def set_vmss_applications(cmd, vmss_name, resource_group_name, application_version_ids, order_applications=False, application_configuration_overrides=None, treat_deployment_as_failure=None, no_wait=False):
+def set_vmss_applications(cmd, vmss_name, resource_group_name, application_version_ids, order_applications=False, application_configuration_overrides=None, treat_deployment_as_failure=None, enable_automatic_upgrade=None, no_wait=False):
     client = _compute_client_factory(cmd.cli_ctx)
     ApplicationProfile, VMGalleryApplication = cmd.get_models('ApplicationProfile', 'VMGalleryApplication')
     try:
         vmss = client.virtual_machine_scale_sets.get(resource_group_name, vmss_name)
     except ResourceNotFoundError:
         raise ResourceNotFoundError('Could not find vmss {}.'.format(vmss_name))
 
-    vmss.virtual_machine_profile.application_profile = ApplicationProfile(gallery_applications=[VMGalleryApplication(package_reference_id=avid) for avid in application_version_ids])
+    vmss.virtual_machine_profile.application_profile = ApplicationProfile(gallery_applications=[VMGalleryApplication(package_reference_id=avid, enable_automatic_upgrade=enable_automatic_upgrade) for avid in application_version_ids])
 
     if order_applications:
         index = 1