{SSH} Migrate commands calling Compute module to aaz-based implementation

[William (william051200)]

Migration from mgmt.compute to aaz-based

---

This checklist is used to make sure that common guidelines for a pull request are followed.

Related command

General Guidelines

• Have you run azdev style <YOUR_EXT> locally? (pip install azdev required)
• Have you run python scripts/ci/test_index.py -q locally? (pip install wheel==0.30.0 required)
• My extension version conforms to the Extension version schema

For new extensions:

• My extension description/summary conforms to the Extension Summary Guidelines.

About Extension Publish

There is a pipeline to automatically build, upload and publish extension wheels.
Once your pull request is merged into main branch, a new pull request will be created to update src/index.json automatically.
You only need to update the version information in file setup.py and historical information in file HISTORY.rst in your PR but do not modify src/index.json.

[azure-client-tools-bot-prd]

️✔️Azure CLI Extensions Breaking Change Test

️✔️Non Breaking Changes

[azure-client-tools-bot-prd]

Hi William (@william051200),
Please write the description of changes which can be perceived by customers into HISTORY.rst.
If you want to release a new extension version, please update the version in setup.py as well.

[Yong Zhang (yonzhan)]

Thank you for your contribution! We will review the pull request and get back to you soon.

[github-actions]

The git hooks are available for azure-cli and azure-cli-extensions repos. They could help you run required checks before creating the PR.

Please sync the latest code with latest dev branch (for azure-cli) or main branch (for azure-cli-extensions).
After that please run the following commands to enable git hooks:

pip install azdev --upgrade
azdev setup -c <your azure-cli repo path> -r <your azure-cli-extensions repo path>

[github-actions]

CodeGen Tools Feedback Collection

Thank you for using our CodeGen tool. We value your feedback, and we would like to know how we can improve our product. Please take a few minutes to fill our codegen survey

[github-actions]

• For more info about extension versioning, please refer to Extension version schema

[Copilot]

Pull request overview

Migrates the ssh extension’s Azure VM compute lookups from the management SDK (mgmt.compute) to AAZ-based command implementations, aligning compute operations with the extension’s existing AAZ usage.

Changes:

• Bump ssh extension version to 2.0.7 and add release notes.
• Switch Azure VM “show” calls to AAZ (azure.cli.command_modules.vm.aaz...) and update downstream field access to dict-style.
• Update VM NIC enumeration logic to read from AAZ VM output shape.

Reviewed changes

Copilot reviewed 4 out of 4 changed files in this pull request and generated 2 comments.

File	Description
src/ssh/setup.py	Version bump to reflect the migration release.
src/ssh/azext_ssh/target_os_utils.py	Replace mgmt compute VM GET with AAZ VM Show; adjust OS type extraction.
src/ssh/azext_ssh/ip_utils.py	Replace mgmt compute VM GET with AAZ VM Show; adjust NIC reference iteration.
src/ssh/HISTORY.md	Add 2.0.7 release entry describing the compute migration.

[Copilot]

Add/update unit tests for _get_azure_vm_os to cover the new AAZ-based VM retrieval (successful OS type extraction and the exception path). test_target_os_utils.py currently covers Arc/VMware but not Azure VM, so this change can regress silently if the AAZ output shape or invocation changes.

Suggested change

	if vm and vm.get('storageProfile', {}).get('osDisk', {}).get('osType'):
	os_type = vm['storageProfile']['osDisk']['osType']
	if vm:
	os_type = vm.get('storageProfile', {}).get('osDisk', {}).get('osType')

[Copilot]

Release note wording: consider hyphenating "AAZ-based" (compound modifier) for readability/consistency.

Suggested change

	* Migrate code from Azure SDK to AAZ based commands for compute operations (VM)
	* Migrate code from Azure SDK to AAZ-based commands for compute operations (VM)

[Julie Zhu (yanzhudd)]

please use the overriden VMShow under the operations folder, as it already includes a fix for a deserialization issue

[Julie Zhu (yanzhudd)]

the same as above

[William (william051200)]

/azp run

[azure-pipelines]

Azure Pipelines successfully started running 2 pipeline(s).

[Azure CLI Bot (azclibot)]

[Release] Update index.json for extension [ ssh-2.0.9 ] : https://dev.azure.com/msazure/One/_build/results?buildId=167603859&view=results