Fix vm get boot log (#18830)

Azure · Jul 16, 2021 · 0ec28e8 · 0ec28e8
1 parent 53b5c5b
commit 0ec28e8
Show file tree

Hide file tree

Showing 3 changed files with 535 additions and 355 deletions.
diff --git a/src/azure-cli/azure/cli/command_modules/vm/custom.py b/src/azure-cli/azure/cli/command_modules/vm/custom.py
@@ -1587,9 +1587,8 @@ def get_boot_log(cmd, resource_group_name, vm_name):
     # Managed storage
     if blob_uri is None:
         try:
-            poller = client.virtual_machines.retrieve_boot_diagnostics_data(resource_group_name, vm_name)
-            uris = LongRunningOperation(cmd.cli_ctx)(poller)
-            blob_uri = uris.serial_console_log_blob_uri
+            boot_diagnostics_data = client.virtual_machines.retrieve_boot_diagnostics_data(resource_group_name, vm_name)
+            blob_uri = boot_diagnostics_data.serial_console_log_blob_uri
         except CloudError:
             pass
         if blob_uri is None: