azure_data_lake_store_gen1_account_management_quickstart.py

1import os
2from azure.identity import DefaultAzureCredential
3from azure.mgmt.resource import ResourceManagementClient
4from azure.mgmt.datalake.store import DataLakeStoreAccountManagementClient
5from azure.mgmt.datalake.store.models import CreateDataLakeStoreAccountParameters
6
7# 1. Set up variables (Ensure these environment variables are set or replace them)
8SUBSCRIPTION_ID = os.environ.get("AZURE_SUBSCRIPTION_ID", "your-subscription-id")
9RESOURCE_GROUP_NAME = "my-datalake-rg"
10LOCATION = "eastus"
11ADLS_ACCOUNT_NAME = "mydatalakestoregen1"
12
13# 2. Authenticate
14# DefaultAzureCredential will use managed identity, environment variables, or CLI login
15credential = DefaultAzureCredential()
16
17# 3. Initialize Clients
18resource_client = ResourceManagementClient(credential, SUBSCRIPTION_ID)
19adls_mgmt_client = DataLakeStoreAccountManagementClient(credential, SUBSCRIPTION_ID)
20
21# 4. Create a Resource Group
22print(f"Creating resource group: {RESOURCE_GROUP_NAME}")
23resource_client.resource_groups.create_or_update(
24    RESOURCE_GROUP_NAME,
25    {"location": LOCATION}
26)
27
28# 5. Create a Data Lake Storage Gen1 Account
29print(f"Creating Data Lake Store account: {ADLS_ACCOUNT_NAME}")
30adls_params = CreateDataLakeStoreAccountParameters(location=LOCATION)
31
32# The operation is long-running, so we wait for the result
33poller = adls_mgmt_client.accounts.begin_create(
34    RESOURCE_GROUP_NAME,
35    ADLS_ACCOUNT_NAME,
36    adls_params
37)
38adls_account = poller.result()
39
40print(f"Successfully created account: {adls_account.name}")
41
42# 6. List accounts in the resource group
43print(f"Listing accounts in resource group {RESOURCE_GROUP_NAME}:")
44for account in adls_mgmt_client.accounts.list_by_resource_group(RESOURCE_GROUP_NAME):
45    print(f"- {account.name}")