vmss deallocate migration

Author: naga-nandyala

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

@@ -4,8 +4,9 @@
 # --------------------------------------------------------------------------------------------
 # pylint: disable=no-self-use, line-too-long, protected-access, too-few-public-methods, unused-argument
 from knack.log import get_logger
-
+from azure.cli.core.aaz import register_command
 from ..aaz.latest.vmss import ListInstances as _VMSSListInstances
+from ..aaz.latest.vmss import Deallocate as _VMSSDeallocate
 
 logger = get_logger(__name__)
 
@@ -25,3 +26,17 @@ def _output(self, *args, **kwargs):
         result = self.deserialize_output(self.ctx.vars.instance.value, client_flatten=True)
         next_link = self.deserialize_output(self.ctx.vars.instance.next_link)
         return result, next_link
+
+class VMSSDeallocate(_VMSSDeallocate):
+    def _output(self, *args, **kwargs):
+        from azure.cli.core.aaz import AAZUndefined, has_value
+
+        # Resolve flatten conflict
+        # When the type field conflicts, the type in inner layer is ignored and the outer layer is applied
+        if has_value(self.ctx.vars.instance.resources):
+            for resource in self.ctx.vars.instance.resources:
+                if has_value(resource.type):
+                    resource.type = AAZUndefined
+
+        result = self.deserialize_output(self.ctx.vars.instance, client_flatten=True)
+        return result