azure_data_lake_store_gen1_account_management_quickstart.py

1import os
2from azure.common.credentials import ServicePrincipalCredentials
3from azure.mgmt.resource import ResourceManagementClient
4from azure.mgmt.datalake.store import DataLakeStoreAccountManagementClient
5from azure.mgmt.datalake.store.models import DataLakeStoreAccount
6
7# Set your variables here or ensure they are set in your environment
8SUBSCRIPTION_ID = os.environ.get("AZURE_SUBSCRIPTION_ID")
9TENANT_ID = os.environ.get("AZURE_TENANT_ID")
10CLIENT_ID = os.environ.get("AZURE_CLIENT_ID")
11CLIENT_SECRET = os.environ.get("AZURE_CLIENT_SECRET")
12RESOURCE_GROUP_NAME = "my-resource-group"
13LOCATION = "eastus2"
14ADLS_ACCOUNT_NAME = "mydatalakestore"
15
16# 1. Authenticate
17credentials = ServicePrincipalCredentials(
18    client_id=CLIENT_ID,
19    secret=CLIENT_SECRET,
20    tenant=TENANT_ID
21)
22
23# 2. Initialize Clients
24resource_client = ResourceManagementClient(credentials, SUBSCRIPTION_ID)
25adls_mgmt_client = DataLakeStoreAccountManagementClient(credentials, SUBSCRIPTION_ID)
26
27# 3. Create Resource Group
28resource_client.resource_groups.create_or_update(
29    RESOURCE_GROUP_NAME,
30    {'location': LOCATION}
31)
32
33# 4. Create Data Lake Store Account
34print(f"Creating Data Lake Store account: {ADLS_ACCOUNT_NAME}...")
35params = DataLakeStoreAccount(location=LOCATION)
36adls_account_future = adls_mgmt_client.accounts.create(
37    RESOURCE_GROUP_NAME,
38    ADLS_ACCOUNT_NAME,
39    params
40)
41adls_account = adls_account_future.result()
42print(f"Account created: {adls_account.name}")
43
44# 5. List Data Lake Store Accounts
45print("Listing accounts in subscription:")
46for account in adls_mgmt_client.accounts.list():
47    print(f"\t{account.name}")
48
49# 6. Delete the account (Optional)
50# adls_mgmt_client.accounts.delete(RESOURCE_GROUP_NAME, ADLS_ACCOUNT_NAME)