from azure.storage.blob import BlobServiceClient

from pyinfra.config import CONFIG


def get_azure_client(connection_string=None) -> BlobServiceClient:

    if not connection_string:
        connection_string = CONFIG.storage.azure.connection_string

    return BlobServiceClient.from_connection_string(conn_str=connection_string)