azure_redis_enterprise_cluster_and_database_creation.py

1import os
2from azure.identity import DefaultAzureCredential
3from azure.mgmt.redisenterprise import RedisEnterpriseManagementClient
4from azure.mgmt.redisenterprise.models import Cluster, SKU, Database, Capability
5
6def main():
7    # Substitution variables
8    subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "your-subscription-id")
9    resource_group = "your-resource-group"
10    cluster_name = "your-redis-enterprise-cluster"
11    location = "eastus"
12
13    # Authenticate using default credentials
14    credential = DefaultAzureCredential()
15    client = RedisEnterpriseManagementClient(credential, subscription_id)
16
17    # 1. Create a Redis Enterprise Cluster
18    # Note: SKU name can be Enterprise_E10, Enterprise_E20, etc.
19    poller = client.redis_enterprise.begin_create(
20        resource_group_name=resource_group,
21        cluster_name=cluster_name,
22        parameters=Cluster(
23            location=location,
24            sku=SKU(name="Enterprise_E10", capacity=2)
25        )
26    )
27    cluster = poller.result()
28    print(f"Created Cluster: {cluster.name}")
29
30    # 2. Create a Database in the cluster
31    # Default database name is "default"
32    db_poller = client.databases.begin_create(
33        resource_group_name=resource_group,
34        cluster_name=cluster_name,
35        database_name="default",
36        parameters=Database(
37            client_protocol="Encrypted",
38            port=10000,
39            clustering_policy="EnterpriseCluster",
40            eviction_policy="NoEviction"
41        )
42    )
43    database = db_poller.result()
44    print(f"Created Database: {database.name}")
45
46if __name__ == "__main__":
47    main()