lib.DataLakeCredential doesn't work after the latest release 1.0.1

[dylanw-oss]

I have been using Managed Identity in the ADLS client, and it worked well until the release of version 1.0.1 on June 3.

Got error message:

adlsCredential = lib.DataLakeCredential(
AttributeError: module 'azure.datalake.store.lib' has no attribute 'DataLakeCredential'

The function I'm using to create ADLS client:

def get_adls_gen1_client(tenant_id: str, client_id: str, store_name: str):
    resource = "https://datalake.azure.net/.default"
    scopes = [resource]
    credential = DefaultAzureCredential(exclude_interactive_browser_credential=False)
    token = credential.get_token(resource)
    adlsCredential = lib.DataLakeCredential(
        {
            "access_token": token.token,
            "access": token.token,
            "expires_in": token.expires_on,
            "token_type": "Bearer",
            "resource": resource,
            "scopes": scopes,
            "refresh": False,
            "time": time.time(),
            "tenant": tenant_id,
            "client_id": client_id,
            "scopes": scopes,
        }
    )
    return core.AzureDLFileSystem(token=adlsCredential, store_name=store_name)

[dylanw-oss]

@akharit

[akharit]

This is expected. We moved to using TokenCredential class in 1.x and removed any auth related code from storage layer. All auth code is now in the Identity library.

You would need to change to using one of the methods from azure identity or implement your own that satisfies the
TokenCredential interface from Azure Identity team.

https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/identity/azure-identity/README.md

[dylanw-oss]

My co-worker shared a code change he made, I'll try it, please feel free to close it, and it would better to update document.

from azure.datalake.store.core import AzureDLFileSystem

cred = DefaultAzureCredential(  
        managed_identity_client_id=settings.AZURE_CLIENT_ID
    )
return AzureDLFileSystem(token_credential=cred, store_name=store_name)