azure_cosmosdb_nosql_account_creation_with_default_credential.py

1import os
2from azure.identity import DefaultAzureCredential
3from azure.mgmt.cosmosdb import CosmosDBManagementClient
4
5def main():
6    # Acquisition of constants from environment variables
7    SUBSCRIPTION_ID = os.environ.get("AZURE_SUBSCRIPTION_ID", "your-subscription-id")
8    RESOURCE_GROUP_NAME = "myResourceGroup"
9    ACCOUNT_NAME = "my-cosmos-db-account"
10    LOCATION = "westus"
11
12    # Authenticate using default Azure credentials
13    # Ensure you have AZURE_TENANT_ID, AZURE_CLIENT_ID, and AZURE_CLIENT_SECRET set
14    credential = DefaultAzureCredential()
15
16    # Initialize the Cosmos DB Management Client
17    client = CosmosDBManagementClient(
18        credential=credential,
19        subscription_id=SUBSCRIPTION_ID
20    )
21
22    # Define the parameters for the new Cosmos DB account
23    # This creates a standard SQL (NoSQL) API account
24    parameters = {
25        "location": LOCATION,
26        "database_account_offer_type": "Standard",
27        "locations": [{"location_name": LOCATION, "failover_priority": 0, "is_zone_redundant": False}],
28    }
29
30    print(f"Creating Cosmos DB account: {ACCOUNT_NAME}...")
31
32    # Create the account
33    # This is a long-running operation; .result() waits for completion
34    poller = client.database_accounts.begin_create_or_update(
35        RESOURCE_GROUP_NAME,
36        ACCOUNT_NAME,
37        parameters
38    )
39    account = poller.result()
40
41    print(f"Successfully created account: {account.name}")
42
43if __name__ == "__main__":
44    main()