{vm-repair} Migrate commands calling Compute module to aaz-based implementation and update deprecated package

[william051200]

Migrate code from Azure SDK to AAZ based commands for compute operations
Replace deprecated pkgutil loader with importlib.import_module

After running some tests for version compatibility, here are the results I got,
az extension add vm-repair 2.2.1 package generated from azdev 0.2.12 & 0.2.13 will show version = null

Those successfully installed (with no null version) also runs properly, with no error.
This confirms the migration has no issue & the issue raised by user is caused by version incompatibility.

Azure CLI 0.87:
1. azdev 0.2.13 - fail (no version)
2. azdev 0.2.12 - fail (no version)
3. azdev 0.2.10 - success
4. azdev 0.2.9 - success
5. azdev 0.2.8 - success

Azure CLI 0.88:
1. azdev 0.2.13 - success
2. azdev 0.2.12 - success
3. azdev 0.2.10 - success
4. azdev 0.2.9 - success
5. azdev 0.2.8 - success

---

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 azdev 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.

🤖 PR Validation — ️✔️ All clear

Breaking Changes
️✔️ None

[azure-client-tools-bot-prd]

️✔️Azure CLI Extensions Breaking Change Test

️✔️Non Breaking Changes

[Copilot]

Pull request overview

This PR updates the vm-repair extension to use AAZ-based VM retrieval helpers (instead of Compute SDK objects) and adds defensive handling in extension-version checking to avoid crashes when version metadata is missing.

Changes:

• Bump vm-repair extension version to 2.2.1 and update release notes.
• Migrate VM object access patterns across custom.py, _validators.py, and repair_utils.py from SDK object attributes to AAZ JSON/dict access.
• Add handling in check_extension_version to avoid 'str' > 'NoneType' crashes when extension version is null.

Reviewed changes

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

Show a summary per file

File	Description
src/vm-repair/setup.py	Version bump to 2.2.1.
src/vm-repair/HISTORY.rst	Add 2.2.1 release notes describing the migration and crash fix.
src/vm-repair/azext_vm_repair/repair_utils.py	Switch VM parsing to dict access; add null-safe extension version check.
src/vm-repair/azext_vm_repair/custom.py	Switch command implementation to AAZ VM retrieval and dict-based VM field access.
src/vm-repair/azext_vm_repair/_validators.py	Switch validation logic to AAZ VM retrieval and dict-based access.

[yonzhan]

vm-repair

[william051200]

/azp run

[azure-pipelines]

Azure Pipelines:
Successfully started running 2 pipeline(s).

[william051200]

/azp run

[azure-pipelines]

Azure Pipelines:
Successfully started running 2 pipeline(s).

[william051200]

/azp run

[azure-pipelines]

Azure Pipelines:
Successfully started running 2 pipeline(s).

[Copilot]

Pull request overview

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

Comments suppressed due to low confidence (2)

src/vm-repair/azext_vm_repair/custom.py:541

• In the unmanaged-disk restore path, this code assumes parameters have gone through the CLI validator, but restore() is also called internally (e.g., from repair-and-restore flows) without validators. If dataDisks is missing or disk_name isn’t found, the list indexing will raise and get reported as a generic "unexpected error". Add a defensive check and raise a clearer error before indexing.

                repair_vm = get_vm_by_aaz(cmd, repair_vm_id['resource_group'], repair_vm_id['name'])
                data_disks = repair_vm.get('storageProfile', {}).get('dataDisks')

                # The params went through validator so no need for existence checks
                disk_uri = [disk.get('vhd', {}).get('uri') for disk in data_disks if disk.get('name') == disk_name][0]

src/vm-repair/azext_vm_repair/repair_utils.py:576

• Potential TypeError: os_image_ref.get('version') in urns[0] will raise if the version is None (or otherwise not a string). The previous implementation guarded this with an isinstance(..., str) check; keep that guard when switching to dict access.

    os_image_ref = source_vm.get('storageProfile', {}).get('imageReference', {})
    if os_image_ref and os_image_ref.get('version') in urns[0]: