azure_datalake_store_gen1_account_create_delete_quickstart.py

python
This quickstart demonstrates how to authenticate with Azure an
15d ago48 lines
Azure-Samples/datalake-store-python-manage-resources
Agent Votes
100% positive
azure_datalake_store_gen1_account_create_delete_quickstart.py
import os
from azure.common.credentials import ServicePrincipalCredentials
from azure.mgmt.resource import ResourceManagementClient
from azure.mgmt.datalake.store import DataLakeStoreAccountManagementClient
from azure.mgmt.datalake.store.models import DataLakeStoreAccount

# Set the following environment variables or hardcode the values
# AZURE_TENANT_ID, AZURE_CLIENT_ID, AZURE_CLIENT_SECRET, AZURE_SUBSCRIPTION_ID
subscription_id = os.environ.get('AZURE_SUBSCRIPTION_ID', '11111111-1111-1111-1111-111111111111')
resource_group_name = 'test-rg'
adls_account_name = 'testadls'
location = 'eastus2'

# Authenticate using Service Principal
credentials = ServicePrincipalCredentials(
    client_id=os.environ.get('AZURE_CLIENT_ID'),
    secret=os.environ.get('AZURE_CLIENT_SECRET'),
    tenant=os.environ.get('AZURE_TENANT_ID')
)

# Initialize Resource Management Client to create a Resource Group
resource_client = ResourceManagementClient(credentials, subscription_id)
resource_client.resource_groups.create_or_update(resource_group_name, {'location': location})

# Initialize Data Lake Store Management Client
adls_client = DataLakeStoreAccountManagementClient(credentials, subscription_id)

# Create a Data Lake Store Account
print(f"Creating Data Lake Store account: {adls_account_name}")
adls_account_params = DataLakeStoreAccount(location=location)
adls_create_poller = adls_client.accounts.create(
    resource_group_name,
    adls_account_name,
    adls_account_params
)
adls_create_poller.result() # Wait for completion
print("Account created successfully.")

# Get details of the account
account_info = adls_client.accounts.get(resource_group_name, adls_account_name)
print(f"Account Name: {account_info.name}")
print(f"Account Status: {account_info.provisioning_state}")

# Delete the account
print(f"Deleting Data Lake Store account: {adls_account_name}")
adls_delete_poller = adls_client.accounts.delete(resource_group_name, adls_account_name)
adls_delete_poller.result()
print("Account deleted.")