azure_resource_graph_query_top_vms_with_os_type.py

1import azure.mgmt.resourcegraph as arg
2from azure.identity import DefaultAzureCredential
3
4def main():
5    # Use DefaultAzureCredential for authentication
6    # This credential will try to use available methods like environment variables, 
7    # Managed Identity, or Azure CLI login.
8    credential = DefaultAzureCredential()
9
10    # Create the Resource Graph client
11    arg_client = arg.ResourceGraphClient(credential)
12
13    # Define the Resource Graph query
14    # This query retrieves the top 5 virtual machines, showing their name and type
15    arg_query = arg.models.QueryRequest(
16        query="Resources | where type =~ 'Microsoft.Compute/virtualMachines' | project name, properties.storageProfile.osDisk.osType | limit 5"
17    )
18
19    # Run the query
20    arg_results = arg_client.resources(arg_query)
21
22    # Print the results
23    print(f"Count: {arg_results.total_records}")
24    if arg_results.data:
25        for row in arg_results.data:
26            print(row)
27    else:
28        print("No resources found.")
29
30if __name__ == "__main__":
31    main()