azure_mgmt_msi_create_user_assigned_identity_with_default_credential.py

1import os
2from azure.identity import DefaultAzureCredential
3from azure.mgmt.msi import ManagedServiceIdentityClient
4from azure.mgmt.msi.models import Identity
5
6def create_user_assigned_identity():
7    # Acquisition of constants from environment variables
8    SUBSCRIPTION_ID = os.environ.get("AZURE_SUBSCRIPTION_ID", "your-subscription-id")
9    RESOURCE_GROUP = "sample-resource-group"
10    IDENTITY_NAME = "sample-identity"
11    LOCATION = "eastus"
12
13    # Authentication using DefaultAzureCredential
14    credential = DefaultAzureCredential()
15
16    # Initialize the Management Client
17    msi_client = ManagedServiceIdentityClient(
18        credential=credential,
19        subscription_id=SUBSCRIPTION_ID
20    )
21
22    # Create or update a User Assigned Identity
23    print(f"Creating identity: {IDENTITY_NAME}...")
24    identity = msi_client.user_assigned_identities.create_or_update(
25        resource_group_name=RESOURCE_GROUP,
26        resource_name=IDENTITY_NAME,
27        parameters=Identity(location=LOCATION)
28    )
29
30    print(f"Successfully created identity with ID: {identity.id}")
31    print(f"Client ID: {identity.client_id}")
32    print(f"Principal ID: {identity.principal_id}")
33
34if __name__ == "__main__":
35    create_user_assigned_identity()