azure-mgmt-databricks

import os from azure.identity import DefaultAzureCredential from azure.mgmt.databricks import AzureDatabricksManagementClient # Set these environment variables or replace with actual values # AZURE_SUBSCRIPTION_ID, AZURE_CLIENT_ID, AZURE_TENANT_ID, AZURE_CLIENT_SECRET subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "") if not subscription_id: raise ValueError("AZURE_SUBSCRIPTION_ID environment variable not set.") # Authenticate using DefaultAzureCredential # This will try to authenticate using various methods (env vars, managed identity, Azure CLI, etc.) credential = DefaultAzureCredential() # Create the Databricks Management Client client = AzureDatabricksManagementClient(credential=credential, subscription_id=subscription_id) # Example: List all Databricks workspaces in the subscription print("Listing Databricks workspaces:") for workspace in client.workspaces.list_by_subscription(): print(f" - {workspace.name} (Resource Group: {workspace.id.split('/')[4]}) (Location: {workspace.location})")