如何升级 Azure Python SDK 到 Track2 预览版本

此文适用于想要把 Azure Python SDK 切换到 Track2 版本的 Azure 用户

升级 SDK

目前全部的 Track2 SDK 均处于预览版本, 用户能够从 pypi.org 上查询具体的 Service SDK 是否有 Track2 SDK, 如 azure-mgmt-compute 的 Track2 预览版本: https://pypi.org/project/azure-mgmt-compute/17.0.0b1/#history

经过 pypi.org 上提示的预览版本号进行安装升级, 如安装 compute: pip install azure-mgmt-compute==17.0.0b1.

使用新的 Identity 库

在 Track2 中再也不使用 azure.common 中的 ServicePrincipalCredentials 进行身份校验， 取而代之的是 azure-identity 中的校验工具, 例如: ClientSecretCredential

使用 pip install azure-identity 安装该库，并在代码中进行以下修改:

# Track1 (old)
import azure.mgmt.compute
from azure.common.credentials import ServicePrincipalCredentials

credentials = ServicePrincipalCredentials(
    client_id=client_id,
    secret=client_secret,
    tenant=tenant_id
)
compute_client = azure.mgmt.compute.ComputeManagementClient(credentials=credentials, subscription_id=self.subscription_id)

# Track2 (new)
import azure.mmgt.compute
from azure.identity import ClientSecretCredential

credential = ClientSecretCredential(
    tenant_id=tenant_id,
    client_id=client_id,
    client_secret=client_secret
)
compute_client = azure.mgmt.compute.ComputeManagementClient(credential=credential, subscription_id=self.subscription_id)

Client 的变更

大部分的 Client API 与 Track1 一致, 只有在异步的 API 部分(RestAPI 返回 202 或者须要显式调用 .result() 的 API), 为了明显的与通常 API 区分, API名称前加上了begin_的前缀.

以建立 VM 为例, 代码的改动:

# Track1 (old)
result = self.compute_client.virtual_machines.create_or_update(
    group_name,
    vm_name,
    parameters
)
result = result.result()

# Track2 (new)
result = self.compute_client.virtual_machines.begin_create_or_update(
    group_name,
    vm_name,
    parameters
)
vm = result.result()

完整的 Track1 与 Track2 Demo 能够参考:https://github.com/00Kai0/azure-playground/tree/master/samples/azure/createVM