Azure Python SDK列出虚拟机并生成自定义JSON响应

Question

在az cli中，我有一个cmd来查找所有正在运行的VM，它将所有信息作为JSON文件返回，其中包含Idefine键

我的问题是如何使用Python SDK实现这一点？我想要JSON格式的相同键/值。我正在用Python做一些后期处理，所以我更喜欢从头到尾使用Python来生成信息

工作代码如下：

az vm list -d -o json --query `
    "[?powerState=='VM running'].{Name:name, admin:osProfile.adminUsername, STATUS:powerState RG:resourceGroup TAGS:tags}" 

返回

[
 {
    "Name": "boston",
    "RG": "r_group",
    "STATUS": "VM running",
    "TAGS": {
      "tag_1": "tag_value_1",
      "tag_2": "tag_value_2"
    },
    "admin": "waldo"
  }
]

Answer 1

您可以使用以下代码列出VM

import azure.mgmt.compute as Compute
from azure.common.client_factory import get_client_from_cli_profile

import retry
import sys
import logging

@retry.retry(RuntimeError, tries=2)
def listVMs(resourceGroup):

    VM = get_client_from_cli_profile(Compute.ComputeManagementClient)

    try:
        VMs = VM.virtual_machines.list(resourceGroup)
        for vm in VMs:
            print(vm)

    except Exception as e:
        logging.error(e)

resourceGroup = sys.argv[1]

if __name__ == '__main__':
    listVMs(resourceGroup)

else:
    print('Running as imported module')
    listVMs(resourceGroup)

Answer 2

最后，在一位同事的帮助下，我通过大量搜索找到了答案。将此贴在这里，希望其他人能从中受益

欢迎@Charles Xu让我从这个答案开始how-could-i-list-azure-virtual-machines-using-python

我很难从微软那里找到具体和相关的文件来解决我的问题

注意：这是一项正在进行的工作，我确信还有很大的改进空间。我认为生成的json并不完美，但json2html并没有抱怨

现在是代码

#!/usr/bin/env python

from azure.common.credentials import ServicePrincipalCredentials
from azure.mgmt.compute import ComputeManagementClient
import json
from json2html import *

# Tenant ID for your Azure subscription
TENANT_ID = 'XXXXXXXXXXXXXXXXXXXXXXXXXXX'

# Your service principal App ID
CLIENT = 'XXXXXXXXXXXXXXXXXXXXXXXXXXX'

# Your service principal password
KEY = 'XXXXXXXXXXXXXXXXXXXXXXXXXXX'

# Your Azure Subscription ID
subscription_id = 'XXXXXXXXXXXXXXXXXXXXXXXXXXX'

credentials = ServicePrincipalCredentials(
    client_id=CLIENT,
    secret=KEY,
    tenant=TENANT_ID
)

result = []

# Create a Resource Management client
compute_client = ComputeManagementClient(credentials, subscription_id)
vm_list = compute_client.virtual_machines.list_all()
for vm in vm_list:
    resource_group = vm.id.split("/")[4]
    vm_name = vm.name
    details = compute_client.virtual_machines.instance_view(resource_group, vm_name, expand='instanceView')

    status = len(details.statuses) >= 2 and details.statuses[1]
    if status and status.code == 'PowerState/running':
        tagz = compute_client.virtual_machines.get(resource_group, vm_name).tags
        if not tagz:
            tagz = ""
        admin = ''
        if hasattr(details, 'os_profile'):
            admin = details.os_profile.admin_username
        row = {"ComputerName": vm_name, "ResourceGroup": resource_group, "Admin": admin,
               "Status": status.code, "Tags": tagz}
        result.append(row)

data = json.dumps(result)
html = json2html.convert(json=data)
print(html)