resource_graph_query.py

# Import Azure Resource Graph library
import azure.mgmt.resourcegraph as arg

# Import specific methods and models from other libraries
from azure.mgmt.resource import SubscriptionClient
from azure.identity import DefaultAzureCredential

# Wrap all the work in a function
def getresources( strQuery ):
    # Get your credentials from Default Azure Credential (development only!) and get your subscription list
    credential = DefaultAzureCredential()
    subsClient = SubscriptionClient(credential)
    subsRaw = []
    for sub in subsClient.subscriptions.list():
        subsRaw.append(sub.as_dict())
    subsList = []
    for sub in subsRaw:
        subsList.append(sub.get('subscription_id'))

    # Create Azure Resource Graph client and set options
    argClient = arg.ResourceGraphClient(credential)
    argQueryOptions = arg.models.QueryRequestOptions(result_format="objectArray")

    # Create query
    argQuery = arg.models.QueryRequest(subscriptions=subsList, query=strQuery, options=argQueryOptions)

    # Run query
    argResults = argClient.resources(argQuery).data

    # Store query results in a file
    with open('resource_inventory.txt', 'w') as file:
        for resource in argResults:
            file.write(f"{resource}\n")

    # Show Python object
    print(argResults)

getresources("Resources")