aaz update vm/vmss

Author: naga-nandyala

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

@@ -1835,9 +1835,10 @@ def _output(self, *args, **kwargs):
         os_type = vm.storage_profile.os_disk.os_type if vm.storage_profile.os_disk.os_type else None
         _set_data_source_for_workspace(cmd, os_type, resource_group_name, workspace_name)
 
-    aux_subscriptions = None
-    if vm and vm.storage_profile and vm.storage_profile.image_reference and 'id' in vm.storage_profile.image_reference:
-        aux_subscriptions = _parse_aux_subscriptions(vm.storage_profile.image_reference['id'])
+    # aux_subscriptions = None
+    # if (vm and vm.storage_profile and vm.storage_profile.image_reference
+    #    and 'id' in vm.storage_profile.image_reference):
+    #     aux_subscriptions = _parse_aux_subscriptions(vm.storage_profile.image_reference['id'])
 
     if size is not None:
         if vm.hardware_profile.vm_size == size:
@@ -1902,7 +1903,7 @@ def _output(self, *args, **kwargs):
     # ==============================================================================
     # All SDK modifications to 'vm' object have already happened above (storage, security, etc.)
     # Now we just need to apply those changes using AAZ's Update command
-    
+
     class VMUpdateCustom(UpdateVM):
         def pre_instance_update(self, instance):
             """
@@ -1911,13 +1912,13 @@ def pre_instance_update(self, instance):
             """
             vm_dict = vm.serialize()
             vm_dict.pop('resources', None)  # Remove conflicting resources
-            
+
             for key, value in vm_dict.items():
                 try:
                     instance[key] = value
                 except Exception as e:
-                    logger.debug(f"Could not set {key}: {e}")
-        
+                    logger.debug("Could not set %s: %s", key, e)
+
         def post_instance_update(self, instance):
             """
             Set AAZ-only fields after AAZ processes command_args.
@@ -1927,35 +1928,34 @@ def post_instance_update(self, instance):
                 props = getattr(instance, 'properties', None)
                 sp = getattr(props, 'security_profile', None) if props else None
                 pas = getattr(sp, 'proxy_agent_settings', None) if sp else None
-                
+
                 if pas:
                     pas.add_proxy_agent_extension = add_proxy_agent_extension
                 else:
                     logger.debug("Cannot set add_proxy_agent_extension - proxy_agent_settings does not exist")
-        
+
         def _output(self, *args, **kwargs):
             """
             Remove resources before flattening to avoid KeyError issues during output serialization.
             """
             self.ctx.vars.instance.resources = None
             return self.deserialize_output(self.ctx.vars.instance, client_flatten=True)
-    
+
     args = {
         'resource_group': resource_group_name,
         'vm_name': vm_name,
         'no_wait': no_wait
     }
-    
-    return VMUpdateCustom(cli_ctx=cmd.cli_ctx)(command_args=args)    
-
 
+    return VMUpdateCustom(cli_ctx=cmd.cli_ctx)(command_args=args)
 
     # client = _compute_client_factory(cmd.cli_ctx, aux_subscriptions=aux_subscriptions)
     # if wire_server_access_control_profile_reference_id is not None or \
     #         imds_access_control_profile_reference_id is not None or \
     #         add_proxy_agent_extension is not None:
     #     kwargs['parameters'] = vm
-    # return sdk_no_wait(no_wait, client.virtual_machines.begin_create_or_update, resource_group_name, vm_name, **kwargs)
+    # return sdk_no_wait(no_wait, client.virtual_machines.begin_create_or_update, resource_group_name, vm_name
+    #                   , **kwargs)
 # endregion
 
 
@@ -4598,34 +4598,34 @@ def _output(self, *args, **kwargs):
 
     if zone_balance is not None:
         vmss.zone_balance = zone_balance
-    
+
     # if wire_server_access_control_profile_reference_id is not None or \
     #         imds_access_control_profile_reference_id is not None or \
     #         add_proxy_agent_extension is not None:
     #     kwargs['parameters'] = vmss
 
     # return sdk_no_wait(no_wait, client.virtual_machine_scale_sets.begin_create_or_update,
-    #                    resource_group_name, name, **kwargs)    
+    #                    resource_group_name, name, **kwargs)
 
     # ==============================================================================
     # AAZ Update: Use SDK for complex logic, then apply via AAZ (same pattern as update_vm)
     # ==============================================================================
     from .aaz.latest.vmss import Update as UpdateVMSS
-    
+
     class VMSSUpdateCustom(UpdateVMSS):
         def pre_instance_update(self, instance):
             """
             Copy SDK VMSS object properties into the AAZ instance before the update.
             This ensures AAZ sees the current state, but does NOT override user args.
             """
             vmss_dict = vmss.serialize()
-            
+
             for key, value in vmss_dict.items():
                 try:
                     instance[key] = value
                 except Exception as e:
-                    logger.debug(f"Could not set {key}: {e}")
-        
+                    logger.debug("Could not set %s: %s", key, e)
+
         def post_instance_update(self, instance):
             """
             Set AAZ-only fields after AAZ processes command_args.
@@ -4637,18 +4637,18 @@ def post_instance_update(self, instance):
                 vm_profile = getattr(props, 'virtual_machine_profile', None) if props else None
                 sp = getattr(vm_profile, 'security_profile', None) if vm_profile else None
                 pas = getattr(sp, 'proxy_agent_settings', None) if sp else None
-                
+
                 if pas:
                     pas.add_proxy_agent_extension = add_proxy_agent_extension
                 else:
                     logger.debug("Cannot set add_proxy_agent_extension - proxy_agent_settings does not exist")
-    
+
     args = {
         'resource_group': resource_group_name,
         'vm_scale_set_name': name,
         'no_wait': no_wait
     }
-    
+
     return VMSSUpdateCustom(cli_ctx=cmd.cli_ctx)(command_args=args)
 
 # endregion