[refractor] - Added handling to assign_vm_identity function

Author: william051200

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

@@ -892,8 +892,8 @@ def setter(vm, external_identities=external_identities):
 
     vm = getter()
     return _construct_identity_info(identity_scope, identity_role,
-                                   vm.get('identity').get('principalId') or None,
-                                   vm.get('identity').get('userAssignedIdentities') or None)
+                                   vm.get('identity').get('principalId') or None if vm.get('identity') else None,
+                                   vm.get('identity').get('userAssignedIdentities') or None if vm.get('identity') else None)
 # endregion