apache-airflow-providers-microsoft-azure 8.1.0rc1__py3-none-any.whl → 8.2.0rc1__py3-none-any.whl

airflow/providers/microsoft/azure/__init__.py

@@ -28,7 +28,7 @@ import packaging.version
 
 __all__ = ["__version__"]
 
-__version__ = "8.1.0"
+__version__ = "8.2.0"
 
 try:
     from airflow import __version__ as airflow_version

airflow/providers/microsoft/azure/get_provider_info.py

@@ -29,6 +29,7 @@ def get_provider_info():
         "description": "`Microsoft Azure <https://azure.microsoft.com/>`__\n",
         "suspended": False,
         "versions": [
+            "8.2.0",
             "8.1.0",
             "8.0.0",
             "7.0.0",

airflow/providers/microsoft/azure/hooks/adx.py

@@ -29,12 +29,15 @@ import warnings
 from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
-from azure.identity import DefaultAzureCredential
 from azure.kusto.data import ClientRequestProperties, KustoClient, KustoConnectionStringBuilder
 from azure.kusto.data.exceptions import KustoServiceError
 
 from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
 from airflow.hooks.base import BaseHook
+from airflow.providers.microsoft.azure.utils import (
+    add_managed_identity_connection_widgets,
+    get_sync_default_azure_credential,
+)
 
 if TYPE_CHECKING:
     from azure.kusto.data.response import KustoResponseDataSetV2
@@ -80,6 +83,7 @@ class AzureDataExplorerHook(BaseHook):
     hook_name = "Azure Data Explorer"
 
     @classmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3PasswordFieldWidget, BS3TextFieldWidget
@@ -192,9 +196,15 @@ class AzureDataExplorerHook(BaseHook):
         elif auth_method == "AAD_DEVICE":
             kcsb = KustoConnectionStringBuilder.with_aad_device_authentication(cluster)
         elif auth_method == "AZURE_TOKEN_CRED":
+            managed_identity_client_id = conn.extra_dejson.get("managed_identity_client_id")
+            workload_identity_tenant_id = conn.extra_dejson.get("workload_identity_tenant_id")
+            credential = get_sync_default_azure_credential(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
             kcsb = KustoConnectionStringBuilder.with_azure_token_credential(
                 connection_string=cluster,
-                credential=DefaultAzureCredential(),
+                credential=credential,
             )
         else:
             raise AirflowException(f"Unknown authentication method: {auth_method}")

airflow/providers/microsoft/azure/hooks/asb.py

@@ -16,14 +16,20 @@
 # under the License.
 from __future__ import annotations
 
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
-from azure.identity import DefaultAzureCredential
 from azure.servicebus import ServiceBusClient, ServiceBusMessage, ServiceBusSender
 from azure.servicebus.management import QueueProperties, ServiceBusAdministrationClient
 
 from airflow.hooks.base import BaseHook
-from airflow.providers.microsoft.azure.utils import get_field
+from airflow.providers.microsoft.azure.utils import (
+    add_managed_identity_connection_widgets,
+    get_field,
+    get_sync_default_azure_credential,
+)
+
+if TYPE_CHECKING:
+    from azure.identity import DefaultAzureCredential
 
 
 class BaseAzureServiceBusHook(BaseHook):
@@ -40,6 +46,7 @@ class BaseAzureServiceBusHook(BaseHook):
     hook_name = "Azure Service Bus"
 
     @staticmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets() -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
@@ -64,7 +71,7 @@ class BaseAzureServiceBusHook(BaseHook):
                     "<Resource group>.servicebus.windows.net (for Azure AD authenticaltion)"
                 ),
                 "credential": "credential",
-                "schema": "Endpoint=sb://<Resource group>.servicebus.windows.net/;SharedAccessKeyName=<AccessKeyName>;SharedAccessKey=<SharedAccessKey>",  # noqa
+                "schema": "Endpoint=sb://<Resource group>.servicebus.windows.net/;SharedAccessKeyName=<AccessKeyName>;SharedAccessKey=<SharedAccessKey>",
             },
         }
 
@@ -106,7 +113,16 @@ class AdminClientHook(BaseAzureServiceBusHook):
             credential: str | DefaultAzureCredential = self._get_field(extras=extras, field_name="credential")
             fully_qualified_namespace = self._get_field(extras=extras, field_name="fully_qualified_namespace")
             if not credential:
-                credential = DefaultAzureCredential()
+                managed_identity_client_id = self._get_field(
+                    extras=extras, field_name="managed_identity_client_id"
+                )
+                workload_identity_tenant_id = self._get_field(
+                    extras=extras, field_name="workload_identity_tenant_id"
+                )
+                credential = get_sync_default_azure_credential(
+                    managed_identity_client_id=managed_identity_client_id,
+                    workload_identity_tenant_id=workload_identity_tenant_id,
+                )
             client = ServiceBusAdministrationClient(
                 fully_qualified_namespace=fully_qualified_namespace,
                 credential=credential,  # type: ignore[arg-type]
@@ -190,7 +206,16 @@ class MessageHook(BaseAzureServiceBusHook):
             credential: str | DefaultAzureCredential = self._get_field(extras=extras, field_name="credential")
             fully_qualified_namespace = self._get_field(extras=extras, field_name="fully_qualified_namespace")
             if not credential:
-                credential = DefaultAzureCredential()
+                managed_identity_client_id = self._get_field(
+                    extras=extras, field_name="managed_identity_client_id"
+                )
+                workload_identity_tenant_id = self._get_field(
+                    extras=extras, field_name="workload_identity_tenant_id"
+                )
+                credential = get_sync_default_azure_credential(
+                    managed_identity_client_id=managed_identity_client_id,
+                    workload_identity_tenant_id=workload_identity_tenant_id,
+                )
             client = ServiceBusClient(
                 fully_qualified_namespace=fully_qualified_namespace,
                 credential=credential,  # type: ignore[arg-type]

airflow/providers/microsoft/azure/hooks/base_azure.py

@@ -24,7 +24,10 @@ from azure.common.credentials import ServicePrincipalCredentials
 
 from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
 from airflow.hooks.base import BaseHook
-from airflow.providers.microsoft.azure.utils import AzureIdentityCredentialAdapter
+from airflow.providers.microsoft.azure.utils import (
+    AzureIdentityCredentialAdapter,
+    add_managed_identity_connection_widgets,
+)
 
 
 class AzureBaseHook(BaseHook):
@@ -45,6 +48,7 @@ class AzureBaseHook(BaseHook):
     hook_name = "Azure"
 
     @staticmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets() -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
@@ -133,7 +137,12 @@ class AzureBaseHook(BaseHook):
             )
         else:
             self.log.info("Using DefaultAzureCredential as credential")
-            credentials = AzureIdentityCredentialAdapter()
+            managed_identity_client_id = conn.extra_dejson.get("managed_identity_client_id")
+            workload_identity_tenant_id = conn.extra_dejson.get("workload_identity_tenant_id")
+            credentials = AzureIdentityCredentialAdapter(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
 
         return self.sdk_client(
             credentials=credentials,

airflow/providers/microsoft/azure/hooks/batch.py

@@ -26,7 +26,11 @@ from azure.batch import BatchServiceClient, batch_auth, models as batch_models
 
 from airflow.exceptions import AirflowException
 from airflow.hooks.base import BaseHook
-from airflow.providers.microsoft.azure.utils import AzureIdentityCredentialAdapter, get_field
+from airflow.providers.microsoft.azure.utils import (
+    AzureIdentityCredentialAdapter,
+    add_managed_identity_connection_widgets,
+    get_field,
+)
 from airflow.utils import timezone
 
 if TYPE_CHECKING:
@@ -46,15 +50,8 @@ class AzureBatchHook(BaseHook):
     conn_type = "azure_batch"
     hook_name = "Azure Batch Service"
 
-    def _get_field(self, extras, name):
-        return get_field(
-            conn_id=self.conn_id,
-            conn_type=self.conn_type,
-            extras=extras,
-            field_name=name,
-        )
-
     @classmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
@@ -80,6 +77,14 @@ class AzureBatchHook(BaseHook):
         super().__init__()
         self.conn_id = azure_batch_conn_id
 
+    def _get_field(self, extras, name):
+        return get_field(
+            conn_id=self.conn_id,
+            conn_type=self.conn_type,
+            extras=extras,
+            field_name=name,
+        )
+
     @cached_property
     def connection(self) -> BatchServiceClient:
         """Get the Batch client connection (cached)."""
@@ -101,8 +106,13 @@ class AzureBatchHook(BaseHook):
         if all([conn.login, conn.password]):
             credentials = batch_auth.SharedKeyCredentials(conn.login, conn.password)
         else:
+            managed_identity_client_id = conn.extra_dejson.get("managed_identity_client_id")
+            workload_identity_tenant_id = conn.extra_dejson.get("workload_identity_tenant_id")
             credentials = AzureIdentityCredentialAdapter(
-                None, resource_id="https://batch.core.windows.net/.default"
+                None,
+                resource_id="https://batch.core.windows.net/.default",
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
             )
 
         batch_client = BatchServiceClient(credentials, batch_url=batch_account_url)

airflow/providers/microsoft/azure/hooks/container_instance.py

@@ -27,6 +27,7 @@ from azure.mgmt.containerinstance import ContainerInstanceManagementClient
 
 from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
 from airflow.providers.microsoft.azure.hooks.base_azure import AzureBaseHook
+from airflow.providers.microsoft.azure.utils import get_sync_default_azure_credential
 
 if TYPE_CHECKING:
     from azure.mgmt.containerinstance.models import (
@@ -92,7 +93,12 @@ class AzureContainerInstanceHook(AzureBaseHook):
             )
         else:
             self.log.info("Using DefaultAzureCredential as credential")
-            credential = DefaultAzureCredential()
+            managed_identity_client_id = conn.extra_dejson.get("managed_identity_client_id")
+            workload_identity_tenant_id = conn.extra_dejson.get("workload_identity_tenant_id")
+            credential = get_sync_default_azure_credential(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
 
         subscription_id = cast(str, conn.extra_dejson.get("subscriptionId"))
         return ContainerInstanceManagementClient(

airflow/providers/microsoft/azure/hooks/container_registry.py

@@ -21,12 +21,15 @@ from __future__ import annotations
 from functools import cached_property
 from typing import Any
 
-from azure.identity import DefaultAzureCredential
 from azure.mgmt.containerinstance.models import ImageRegistryCredential
 from azure.mgmt.containerregistry import ContainerRegistryManagementClient
 
 from airflow.hooks.base import BaseHook
-from airflow.providers.microsoft.azure.utils import get_field
+from airflow.providers.microsoft.azure.utils import (
+    add_managed_identity_connection_widgets,
+    get_field,
+    get_sync_default_azure_credential,
+)
 
 
 class AzureContainerRegistryHook(BaseHook):
@@ -44,6 +47,7 @@ class AzureContainerRegistryHook(BaseHook):
     hook_name = "Azure Container Registry"
 
     @staticmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets() -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
@@ -103,8 +107,15 @@ class AzureContainerRegistryHook(BaseHook):
             extras = conn.extra_dejson
             subscription_id = self._get_field(extras, "subscription_id")
             resource_group = self._get_field(extras, "resource_group")
+            managed_identity_client_id = self._get_field(extras, "managed_identity_client_id")
+            workload_identity_tenant_id = self._get_field(extras, "workload_identity_tenant_id")
+            credential = get_sync_default_azure_credential(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
             client = ContainerRegistryManagementClient(
-                credential=DefaultAzureCredential(), subscription_id=subscription_id
+                credential=credential,
+                subscription_id=subscription_id,
             )
             credentials = client.registries.list_credentials(resource_group, conn.login).as_dict()
             password = credentials["passwords"][0]["value"]

airflow/providers/microsoft/azure/hooks/container_volume.py

@@ -1,4 +1,3 @@
-#
 # Licensed to the Apache Software Foundation (ASF) under one
 # or more contributor license agreements.  See the NOTICE file
 # distributed with this work for additional information
@@ -19,12 +18,15 @@ from __future__ import annotations
 
 from typing import Any
 
-from azure.identity import DefaultAzureCredential
 from azure.mgmt.containerinstance.models import AzureFileVolume, Volume
 from azure.mgmt.storage import StorageManagementClient
 
 from airflow.hooks.base import BaseHook
-from airflow.providers.microsoft.azure.utils import get_field
+from airflow.providers.microsoft.azure.utils import (
+    add_managed_identity_connection_widgets,
+    get_field,
+    get_sync_default_azure_credential,
+)
 
 
 class AzureContainerVolumeHook(BaseHook):
@@ -41,19 +43,8 @@ class AzureContainerVolumeHook(BaseHook):
     conn_type = "azure_container_volume"
     hook_name = "Azure Container Volume"
 
-    def __init__(self, azure_container_volume_conn_id: str = "azure_container_volume_default") -> None:
-        super().__init__()
-        self.conn_id = azure_container_volume_conn_id
-
-    def _get_field(self, extras, name):
-        return get_field(
-            conn_id=self.conn_id,
-            conn_type=self.conn_type,
-            extras=extras,
-            field_name=name,
-        )
-
     @staticmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets() -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3PasswordFieldWidget, BS3TextFieldWidget
@@ -92,6 +83,18 @@ class AzureContainerVolumeHook(BaseHook):
             },
         }
 
+    def __init__(self, azure_container_volume_conn_id: str = "azure_container_volume_default") -> None:
+        super().__init__()
+        self.conn_id = azure_container_volume_conn_id
+
+    def _get_field(self, extras, name):
+        return get_field(
+            conn_id=self.conn_id,
+            conn_type=self.conn_type,
+            extras=extras,
+            field_name=name,
+        )
+
     def get_storagekey(self, *, storage_account_name: str | None = None) -> str:
         """Get Azure File Volume storage key."""
         conn = self.get_connection(self.conn_id)
@@ -106,8 +109,13 @@ class AzureContainerVolumeHook(BaseHook):
         subscription_id = self._get_field(extras, "subscription_id")
         resource_group = self._get_field(extras, "resource_group")
         if subscription_id and storage_account_name and resource_group:
-            credentials = DefaultAzureCredential()
-            storage_client = StorageManagementClient(credentials, subscription_id)
+            managed_identity_client_id = self._get_field(extras, "managed_identity_client_id")
+            workload_identity_tenant_id = self._get_field(extras, "workload_identity_tenant_id")
+            credential = get_sync_default_azure_credential(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
+            storage_client = StorageManagementClient(credential, subscription_id)
             storage_account_list_keys_result = storage_client.storage_accounts.list_keys(
                 resource_group, storage_account_name
             )

airflow/providers/microsoft/azure/hooks/cosmos.py

@@ -31,12 +31,15 @@ from urllib.parse import urlparse
 
 from azure.cosmos.cosmos_client import CosmosClient
 from azure.cosmos.exceptions import CosmosHttpResponseError
-from azure.identity import DefaultAzureCredential
 from azure.mgmt.cosmosdb import CosmosDBManagementClient
 
 from airflow.exceptions import AirflowBadRequest, AirflowException
 from airflow.hooks.base import BaseHook
-from airflow.providers.microsoft.azure.utils import get_field
+from airflow.providers.microsoft.azure.utils import (
+    add_managed_identity_connection_widgets,
+    get_field,
+    get_sync_default_azure_credential,
+)
 
 
 class AzureCosmosDBHook(BaseHook):
@@ -57,6 +60,7 @@ class AzureCosmosDBHook(BaseHook):
     hook_name = "Azure CosmosDB"
 
     @staticmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets() -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
@@ -126,9 +130,16 @@ class AzureCosmosDBHook(BaseHook):
             if conn.password:
                 master_key = conn.password
             elif resource_group_name:
+                managed_identity_client_id = self._get_field(extras, "managed_identity_client_id")
+                workload_identity_tenant_id = self._get_field(extras, "workload_identity_tenant_id")
+                subscritption_id = self._get_field(extras, "subscription_id")
+                credential = get_sync_default_azure_credential(
+                    managed_identity_client_id=managed_identity_client_id,
+                    workload_identity_tenant_id=workload_identity_tenant_id,
+                )
                 management_client = CosmosDBManagementClient(
-                    credential=DefaultAzureCredential(),
-                    subscription_id=self._get_field(extras, "subscription_id"),
+                    credential=credential,
+                    subscription_id=subscritption_id,
                 )
 
                 database_account = urlparse(conn.login).netloc.split(".")[0]

airflow/providers/microsoft/azure/hooks/data_factory.py

@@ -48,6 +48,11 @@ from azure.mgmt.datafactory.aio import DataFactoryManagementClient as AsyncDataF
 
 from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
 from airflow.hooks.base import BaseHook
+from airflow.providers.microsoft.azure.utils import (
+    add_managed_identity_connection_widgets,
+    get_async_default_azure_credential,
+    get_sync_default_azure_credential,
+)
 
 if TYPE_CHECKING:
     from azure.core.polling import LROPoller
@@ -152,6 +157,7 @@ class AzureDataFactoryHook(BaseHook):
     hook_name: str = "Azure Data Factory"
 
     @staticmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets() -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
@@ -205,7 +211,12 @@ class AzureDataFactoryHook(BaseHook):
                 client_id=conn.login, client_secret=conn.password, tenant_id=tenant
             )
         else:
-            credential = DefaultAzureCredential()
+            managed_identity_client_id = get_field(extras, "managed_identity_client_id")
+            workload_identity_tenant_id = get_field(extras, "workload_identity_tenant_id")
+            credential = get_sync_default_azure_credential(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
         self._conn = self._create_client(credential, subscription_id)
 
         return self._conn
@@ -1140,7 +1151,12 @@ class AzureDataFactoryAsyncHook(AzureDataFactoryHook):
                 client_id=conn.login, client_secret=conn.password, tenant_id=tenant
             )
         else:
-            credential = AsyncDefaultAzureCredential()
+            managed_identity_client_id = get_field(extras, "managed_identity_client_id")
+            workload_identity_tenant_id = get_field(extras, "workload_identity_tenant_id")
+            credential = get_async_default_azure_credential(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
 
         self._async_conn = AsyncDataFactoryManagementClient(
             credential=credential,

airflow/providers/microsoft/azure/hooks/data_lake.py

@@ -34,7 +34,11 @@ from azure.storage.filedatalake import (
 
 from airflow.exceptions import AirflowException
 from airflow.hooks.base import BaseHook
-from airflow.providers.microsoft.azure.utils import AzureIdentityCredentialAdapter, get_field
+from airflow.providers.microsoft.azure.utils import (
+    AzureIdentityCredentialAdapter,
+    add_managed_identity_connection_widgets,
+    get_field,
+)
 
 Credentials = Union[ClientSecretCredential, AzureIdentityCredentialAdapter]
 
@@ -62,6 +66,7 @@ class AzureDataLakeHook(BaseHook):
     hook_name = "Azure Data Lake"
 
     @staticmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets() -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
@@ -118,7 +123,12 @@ class AzureDataLakeHook(BaseHook):
             if tenant:
                 credential = lib.auth(tenant_id=tenant, client_secret=conn.password, client_id=conn.login)
             else:
-                credential = AzureIdentityCredentialAdapter()
+                managed_identity_client_id = self._get_field(extras, "managed_identity_client_id")
+                workload_identity_tenant_id = self._get_field(extras, "workload_identity_tenant_id")
+                credential = AzureIdentityCredentialAdapter(
+                    managed_identity_client_id=managed_identity_client_id,
+                    workload_identity_tenant_id=workload_identity_tenant_id,
+                )
             self._conn = core.AzureDLFileSystem(credential, store_name=self.account_name)
             self._conn.connect()
         return self._conn
@@ -265,6 +275,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
     hook_name = "Azure Date Lake Storage V2"
 
     @classmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3PasswordFieldWidget, BS3TextFieldWidget
@@ -305,6 +316,17 @@ class AzureDataLakeStorageV2Hook(BaseHook):
         self.conn_id = adls_conn_id
         self.public_read = public_read
 
+    def _get_field(self, extra_dict, field_name):
+        prefix = "extra__adls__"
+        if field_name.startswith("extra__"):
+            raise ValueError(
+                f"Got prefixed name {field_name}; please remove the '{prefix}' prefix "
+                f"when using this method."
+            )
+        if field_name in extra_dict:
+            return extra_dict[field_name] or None
+        return extra_dict.get(f"{prefix}{field_name}") or None
+
     @cached_property
     def service_client(self) -> DataLakeServiceClient:
         """Return the DataLakeServiceClient object (cached)."""
@@ -330,7 +352,12 @@ class AzureDataLakeStorageV2Hook(BaseHook):
         elif conn.password:
             credential = conn.password
         else:
-            credential = AzureIdentityCredentialAdapter()
+            managed_identity_client_id = self._get_field(extra, "managed_identity_client_id")
+            workload_identity_tenant_id = self._get_field(extra, "workload_identity_tenant_id")
+            credential = AzureIdentityCredentialAdapter(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
 
         return DataLakeServiceClient(
             account_url=f"https://{conn.host}.dfs.core.windows.net",
@@ -338,17 +365,6 @@ class AzureDataLakeStorageV2Hook(BaseHook):
             **extra,
         )
 
-    def _get_field(self, extra_dict, field_name):
-        prefix = "extra__adls__"
-        if field_name.startswith("extra__"):
-            raise ValueError(
-                f"Got prefixed name {field_name}; please remove the '{prefix}' prefix "
-                f"when using this method."
-            )
-        if field_name in extra_dict:
-            return extra_dict[field_name] or None
-        return extra_dict.get(f"{prefix}{field_name}") or None
-
     def create_file_system(self, file_system_name: str) -> None:
         """Create a new file system under the specified account.
 

airflow/providers/microsoft/azure/hooks/fileshare.py

@@ -19,10 +19,13 @@ from __future__ import annotations
 
 from typing import IO, Any
 
-from azure.identity import DefaultAzureCredential
 from azure.storage.fileshare import FileProperties, ShareDirectoryClient, ShareFileClient, ShareServiceClient
 
 from airflow.hooks.base import BaseHook
+from airflow.providers.microsoft.azure.utils import (
+    add_managed_identity_connection_widgets,
+    get_sync_default_azure_credential,
+)
 
 
 class AzureFileShareHook(BaseHook):
@@ -39,24 +42,8 @@ class AzureFileShareHook(BaseHook):
     conn_type = "azure_fileshare"
     hook_name = "Azure FileShare"
 
-    def __init__(
-        self,
-        share_name: str | None = None,
-        file_path: str | None = None,
-        directory_path: str | None = None,
-        azure_fileshare_conn_id: str = "azure_fileshare_default",
-    ) -> None:
-        super().__init__()
-        self._conn_id = azure_fileshare_conn_id
-        self.share_name = share_name
-        self.file_path = file_path
-        self.directory_path = directory_path
-        self._account_url: str | None = None
-        self._connection_string: str | None = None
-        self._account_access_key: str | None = None
-        self._sas_token: str | None = None
-
     @staticmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets() -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3PasswordFieldWidget, BS3TextFieldWidget
@@ -87,6 +74,23 @@ class AzureFileShareHook(BaseHook):
             },
         }
 
+    def __init__(
+        self,
+        share_name: str | None = None,
+        file_path: str | None = None,
+        directory_path: str | None = None,
+        azure_fileshare_conn_id: str = "azure_fileshare_default",
+    ) -> None:
+        super().__init__()
+        self._conn_id = azure_fileshare_conn_id
+        self.share_name = share_name
+        self.file_path = file_path
+        self.directory_path = directory_path
+        self._account_url: str | None = None
+        self._connection_string: str | None = None
+        self._account_access_key: str | None = None
+        self._sas_token: str | None = None
+
     def get_conn(self) -> None:
         conn = self.get_connection(self._conn_id)
         extras = conn.extra_dejson
@@ -102,6 +106,16 @@ class AzureFileShareHook(BaseHook):
             return f"https://{account_url}.file.core.windows.net"
         return account_url
 
+    def _get_sync_default_azure_credential(self):
+        conn = self.get_connection(self._conn_id)
+        extras = conn.extra_dejson
+        managed_identity_client_id = extras.get("managed_identity_client_id")
+        workload_identity_tenant_id = extras.get("workload_identity_tenant_id")
+        return get_sync_default_azure_credential(
+            managed_identity_client_id=managed_identity_client_id,
+            workload_identity_tenant_id=workload_identity_tenant_id,
+        )
+
     @property
     def share_service_client(self):
         self.get_conn()
@@ -114,7 +128,9 @@ class AzureFileShareHook(BaseHook):
             return ShareServiceClient(account_url=self._account_url, credential=credential)
         else:
             return ShareServiceClient(
-                account_url=self._account_url, credential=DefaultAzureCredential(), token_intent="backup"
+                account_url=self._account_url,
+                credential=self._get_sync_default_azure_credential(),
+                token_intent="backup",
             )
 
     @property
@@ -138,7 +154,7 @@ class AzureFileShareHook(BaseHook):
                 account_url=self._account_url,
                 share_name=self.share_name,
                 directory_path=self.directory_path,
-                credential=DefaultAzureCredential(),
+                credential=self._get_sync_default_azure_credential(),
                 token_intent="backup",
             )
 
@@ -163,7 +179,7 @@ class AzureFileShareHook(BaseHook):
                 account_url=self._account_url,
                 share_name=self.share_name,
                 file_path=self.file_path,
-                credential=DefaultAzureCredential(),
+                credential=self._get_sync_default_azure_credential(),
                 token_intent="backup",
             )
 

airflow/providers/microsoft/azure/hooks/synapse.py

@@ -24,7 +24,11 @@ from azure.synapse.spark import SparkClient
 
 from airflow.exceptions import AirflowTaskTimeout
 from airflow.hooks.base import BaseHook
-from airflow.providers.microsoft.azure.utils import get_field
+from airflow.providers.microsoft.azure.utils import (
+    add_managed_identity_connection_widgets,
+    get_field,
+    get_sync_default_azure_credential,
+)
 
 if TYPE_CHECKING:
     from azure.synapse.spark.models import SparkBatchJobOptions
@@ -63,6 +67,7 @@ class AzureSynapseHook(BaseHook):
     hook_name: str = "Azure Synapse"
 
     @staticmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets() -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
@@ -79,7 +84,11 @@ class AzureSynapseHook(BaseHook):
         """Returns custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "extra"],
-            "relabeling": {"login": "Client ID", "password": "Secret", "host": "Synapse Workspace URL"},
+            "relabeling": {
+                "login": "Client ID",
+                "password": "Secret",
+                "host": "Synapse Workspace URL",
+            },
         }
 
     def __init__(self, azure_synapse_conn_id: str = default_conn_name, spark_pool: str = ""):
@@ -120,7 +129,12 @@ class AzureSynapseHook(BaseHook):
                 client_id=conn.login, client_secret=conn.password, tenant_id=tenant
             )
         else:
-            credential = DefaultAzureCredential()
+            managed_identity_client_id = self._get_field(extras, "managed_identity_client_id")
+            workload_identity_tenant_id = self._get_field(extras, "workload_identity_tenant_id")
+            credential = get_sync_default_azure_credential(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
 
         self._conn = self._create_client(credential, conn.host, spark_pool, livy_api_version, subscription_id)
 

airflow/providers/microsoft/azure/hooks/wasb.py

@@ -33,7 +33,7 @@ from urllib.parse import urlparse
 
 from asgiref.sync import sync_to_async
 from azure.core.exceptions import HttpResponseError, ResourceExistsError, ResourceNotFoundError
-from azure.identity import ClientSecretCredential, DefaultAzureCredential
+from azure.identity import ClientSecretCredential
 from azure.identity.aio import (
     ClientSecretCredential as AsyncClientSecretCredential,
     DefaultAzureCredential as AsyncDefaultAzureCredential,
@@ -47,6 +47,11 @@ from azure.storage.blob.aio import (
 
 from airflow.exceptions import AirflowException
 from airflow.hooks.base import BaseHook
+from airflow.providers.microsoft.azure.utils import (
+    add_managed_identity_connection_widgets,
+    get_async_default_azure_credential,
+    get_sync_default_azure_credential,
+)
 
 if TYPE_CHECKING:
     from azure.storage.blob._models import BlobProperties
@@ -77,6 +82,7 @@ class WasbHook(BaseHook):
     hook_name = "Azure Blob Storage"
 
     @staticmethod
+    @add_managed_identity_connection_widgets
     def get_connection_form_widgets() -> dict[str, Any]:
         """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3PasswordFieldWidget, BS3TextFieldWidget
@@ -207,7 +213,12 @@ class WasbHook(BaseHook):
         # Fall back to old auth (password) or use managed identity if not provided.
         credential = conn.password
         if not credential:
-            credential = DefaultAzureCredential()
+            managed_identity_client_id = self._get_field(extra, "managed_identity_client_id")
+            workload_identity_tenant_id = self._get_field(extra, "workload_identity_tenant_id")
+            credential = get_sync_default_azure_credential(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
             self.log.info("Using DefaultAzureCredential as credential")
         return BlobServiceClient(
             account_url=account_url,
@@ -600,7 +611,9 @@ class WasbAsyncHook(WasbHook):
                 tenant, app_id, app_secret, **client_secret_auth_config
             )
             self.blob_service_client = AsyncBlobServiceClient(
-                account_url=account_url, credential=token_credential, **extra  # type:ignore[arg-type]
+                account_url=account_url,
+                credential=token_credential,
+                **extra,  # type:ignore[arg-type]
             )
             return self.blob_service_client
 
@@ -632,7 +645,12 @@ class WasbAsyncHook(WasbHook):
         # Fall back to old auth (password) or use managed identity if not provided.
         credential = conn.password
         if not credential:
-            credential = AsyncDefaultAzureCredential()
+            managed_identity_client_id = self._get_field(extra, "managed_identity_client_id")
+            workload_identity_tenant_id = self._get_field(extra, "workload_identity_tenant_id")
+            credential = get_async_default_azure_credential(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
             self.log.info("Using DefaultAzureCredential as credential")
         self.blob_service_client = AsyncBlobServiceClient(
             account_url=account_url,

airflow/providers/microsoft/azure/operators/container_instances.py

@@ -278,12 +278,11 @@ class AzureContainerInstancesOperator(BaseOperator):
                 self.on_kill()
 
     def on_kill(self) -> None:
-        if self.remove_on_error:
-            self.log.info("Deleting container group")
-            try:
-                self._ci_hook.delete(self.resource_group, self.name)
-            except Exception:
-                self.log.exception("Could not delete container group")
+        self.log.info("Deleting container group")
+        try:
+            self._ci_hook.delete(self.resource_group, self.name)
+        except Exception:
+            self.log.exception("Could not delete container group")
 
     def _monitor_logging(self, resource_group: str, name: str) -> int:
         last_state = None

airflow/providers/microsoft/azure/secrets/key_vault.py

@@ -14,6 +14,14 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+"""
+This module contains Azure Key Vault Backend.
+
+.. spelling:word-list::
+
+    Entra
+"""
+
 from __future__ import annotations
 
 import logging
@@ -27,6 +35,7 @@ from azure.identity import ClientSecretCredential, DefaultAzureCredential
 from azure.keyvault.secrets import SecretClient
 
 from airflow.exceptions import AirflowProviderDeprecationWarning
+from airflow.providers.microsoft.azure.utils import get_sync_default_azure_credential
 from airflow.secrets import BaseSecretsBackend
 from airflow.utils.log.logging_mixin import LoggingMixin
 from airflow.version import version as airflow_version
@@ -76,8 +85,11 @@ class AzureKeyVaultBackend(BaseSecretsBackend, LoggingMixin):
         If not given, it falls back to ``DefaultAzureCredential``
     :param client_id: The client id of an Azure Key Vault to use.
         If not given, it falls back to ``DefaultAzureCredential``
-    :param client_secret: The client secret of an Azure Key Vault to use.
-        If not given, it falls back to ``DefaultAzureCredential``
+    :param managed_identity_client_id: The client ID of a user-assigned managed identity.
+        If provided with `workload_identity_tenant_id`, they'll pass to ``DefaultAzureCredential``.
+    :param workload_identity_tenant_id: ID of the application's Microsoft Entra tenant.
+        Also called its "directory" ID.
+        If provided with `managed_identity_client_id`, they'll pass to ``DefaultAzureCredential``.
     """
 
     def __init__(
@@ -91,6 +103,8 @@ class AzureKeyVaultBackend(BaseSecretsBackend, LoggingMixin):
         tenant_id: str = "",
         client_id: str = "",
         client_secret: str = "",
+        managed_identity_client_id: str = "",
+        workload_identity_tenant_id: str = "",
         **kwargs,
     ) -> None:
         super().__init__()
@@ -118,6 +132,8 @@ class AzureKeyVaultBackend(BaseSecretsBackend, LoggingMixin):
         self.tenant_id = tenant_id
         self.client_id = client_id
         self.client_secret = client_secret
+        self.managed_identity_client_id = managed_identity_client_id
+        self.workload_identity_tenant_id = workload_identity_tenant_id
         self.kwargs = kwargs
 
     @cached_property
@@ -127,7 +143,10 @@ class AzureKeyVaultBackend(BaseSecretsBackend, LoggingMixin):
         if all([self.tenant_id, self.client_id, self.client_secret]):
             credential = ClientSecretCredential(self.tenant_id, self.client_id, self.client_secret)
         else:
-            credential = DefaultAzureCredential()
+            credential = get_sync_default_azure_credential(
+                managed_identity_client_id=self.managed_identity_client_id,
+                workload_identity_tenant_id=self.workload_identity_tenant_id,
+            )
         client = SecretClient(vault_url=self.vault_url, credential=credential, **self.kwargs)
         return client
 

airflow/providers/microsoft/azure/utils.py

@@ -18,11 +18,13 @@
 from __future__ import annotations
 
 import warnings
+from functools import partial, wraps
 
 from azure.core.pipeline import PipelineContext, PipelineRequest
 from azure.core.pipeline.policies import BearerTokenCredentialPolicy
 from azure.core.pipeline.transport import HttpRequest
 from azure.identity import DefaultAzureCredential
+from azure.identity.aio import DefaultAzureCredential as AsyncDefaultAzureCredential
 from msrest.authentication import BasicTokenAuthentication
 
 
@@ -51,6 +53,64 @@ def get_field(*, conn_id: str, conn_type: str, extras: dict, field_name: str):
     return ret
 
 
+def _get_default_azure_credential(
+    *,
+    managed_identity_client_id: str | None = None,
+    workload_identity_tenant_id: str | None = None,
+    use_async: bool = False,
+) -> DefaultAzureCredential | AsyncDefaultAzureCredential:
+    """Get DefaultAzureCredential based on provided arguments.
+
+    If managed_identity_client_id and workload_identity_tenant_id are provided, this function returns
+    DefaultAzureCredential with managed identity.
+    """
+    credential_cls: type[AsyncDefaultAzureCredential] | type[DefaultAzureCredential] = (
+        AsyncDefaultAzureCredential if use_async else DefaultAzureCredential
+    )
+    if managed_identity_client_id and workload_identity_tenant_id:
+        return credential_cls(
+            managed_identity_client_id=managed_identity_client_id,
+            workload_identity_tenant_id=workload_identity_tenant_id,
+            additionally_allowed_tenants=[workload_identity_tenant_id],
+        )
+    else:
+        return credential_cls()
+
+
+get_sync_default_azure_credential: partial[DefaultAzureCredential] = partial(
+    _get_default_azure_credential,  #  type: ignore[arg-type]
+    use_async=False,
+)
+
+get_async_default_azure_credential: partial[AsyncDefaultAzureCredential] = partial(
+    _get_default_azure_credential,  #  type: ignore[arg-type]
+    use_async=True,
+)
+
+
+def add_managed_identity_connection_widgets(func):
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
+        from flask_babel import lazy_gettext
+        from wtforms import StringField
+
+        widgets = func(*args, **kwargs)
+        widgets.update(
+            {
+                "managed_identity_client_id": StringField(
+                    lazy_gettext("Managed Identity Client ID"), widget=BS3TextFieldWidget()
+                ),
+                "workload_identity_tenant_id": StringField(
+                    lazy_gettext("Workload Identity Tenant ID"), widget=BS3TextFieldWidget()
+                ),
+            }
+        )
+        return widgets
+
+    return wrapper
+
+
 class AzureIdentityCredentialAdapter(BasicTokenAuthentication):
     """Adapt azure-identity credentials for backward compatibility.
 
@@ -60,15 +120,31 @@ class AzureIdentityCredentialAdapter(BasicTokenAuthentication):
     Check https://stackoverflow.com/questions/63384092/exception-attributeerror-defaultazurecredential-object-has-no-attribute-sig
     """
 
-    def __init__(self, credential=None, resource_id="https://management.azure.com/.default", **kwargs):
+    def __init__(
+        self,
+        credential=None,
+        resource_id="https://management.azure.com/.default",
+        *,
+        managed_identity_client_id: str | None = None,
+        workload_identity_tenant_id: str | None = None,
+        **kwargs,
+    ):
         """Adapt azure-identity credentials for backward compatibility.
 
         :param credential: Any azure-identity credential (DefaultAzureCredential by default)
-        :param str resource_id: The scope to use to get the token (default ARM)
+        :param resource_id: The scope to use to get the token (default ARM)
+        :param managed_identity_client_id: The client ID of a user-assigned managed identity.
+            If provided with `workload_identity_tenant_id`, they'll pass to ``DefaultAzureCredential``.
+        :param workload_identity_tenant_id: ID of the application's Microsoft Entra tenant.
+            Also called its "directory" ID.
+            If provided with `managed_identity_client_id`, they'll pass to ``DefaultAzureCredential``.
         """
-        super().__init__(None)
+        super().__init__(None)  # type: ignore[arg-type]
         if credential is None:
-            credential = DefaultAzureCredential()
+            credential = get_sync_default_azure_credential(
+                managed_identity_client_id=managed_identity_client_id,
+                workload_identity_tenant_id=workload_identity_tenant_id,
+            )
         self._policy = BearerTokenCredentialPolicy(credential, resource_id, **kwargs)
 
     def _make_request(self):

{apache_airflow_providers_microsoft_azure-8.1.0rc1.dist-info → apache_airflow_providers_microsoft_azure-8.2.0rc1.dist-info}/METADATA

@@ -1,14 +1,14 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-microsoft-azure
-Version: 8.1.0rc1
+Version: 8.2.0rc1
 Summary: Provider for Apache Airflow. Implements apache-airflow-providers-microsoft-azure package
 Home-page: https://airflow.apache.org/
 Download-URL: https://archive.apache.org/dist/airflow/providers
 Author: Apache Software Foundation
 Author-email: dev@airflow.apache.org
 License: Apache License 2.0
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/8.1.0/
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/8.1.0/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/8.2.0/
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/8.2.0/changelog.html
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
 Project-URL: Source Code, https://github.com/apache/airflow
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
@@ -76,8 +76,7 @@ Requires-Dist: apache-airflow-providers-sftp ; extra == 'sftp'
    KIND, either express or implied.  See the License for the
    specific language governing permissions and limitations
    under the License.
-
- .. Licensed to the Apache Software Foundation (ASF) under one
+.. Licensed to the Apache Software Foundation (ASF) under one
     or more contributor license agreements.  See the NOTICE file
     distributed with this work for additional information
     regarding copyright ownership.  The ASF licenses this file
@@ -97,7 +96,7 @@ Requires-Dist: apache-airflow-providers-sftp ; extra == 'sftp'
 
 Package ``apache-airflow-providers-microsoft-azure``
 
-Release: ``8.1.0rc1``
+Release: ``8.2.0rc1``
 
 
 `Microsoft Azure <https://azure.microsoft.com/>`__
@@ -110,8 +109,7 @@ This is a provider package for ``microsoft.azure`` provider. All classes for thi
 are in ``airflow.providers.microsoft.azure`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/8.1.0/>`_.
-
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/8.2.0/>`_.
 
 Installation
 ------------
@@ -173,4 +171,4 @@ Dependent package
 ====================================================================================================  ==========
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/8.1.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/8.2.0/changelog.html>`_.

{apache_airflow_providers_microsoft_azure-8.1.0rc1.dist-info → apache_airflow_providers_microsoft_azure-8.2.0rc1.dist-info}/RECORD

@@ -1,22 +1,22 @@
-airflow/providers/microsoft/azure/__init__.py,sha256=nH38VasVt8Aj5JYfefzofjwlcgNeV1FRS9xDNjRux_I,1570
-airflow/providers/microsoft/azure/get_provider_info.py,sha256=AH0GdOjV7AknKHJG8ENA1Pte1L6a5E71d5mB94rkYdk,18322
-airflow/providers/microsoft/azure/utils.py,sha256=HWf2jbEKo78W9qWErjJ6e1V1GE7OZnkbbRFA7NjB3G8,4148
+airflow/providers/microsoft/azure/__init__.py,sha256=A4kvtHQo9XYh3bXkl1-QNHM5vSkAdX2833p-oSipxYA,1570
+airflow/providers/microsoft/azure/get_provider_info.py,sha256=bzX27y8jPmWH9fU8t6vXTRX2lXprj2kzybcSrGHZb0M,18343
+airflow/providers/microsoft/azure/utils.py,sha256=0JAjaDDZ0dIT_K9sMUsWaSZzBBQvGtkQUeG2SLrYdzg,7127
 airflow/providers/microsoft/azure/fs/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/microsoft/azure/fs/adls.py,sha256=XV4uWBXO8C6U2gy4KyZjf9ni9zJamoRpKaQIMT2ttdM,2183
 airflow/providers/microsoft/azure/hooks/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/microsoft/azure/hooks/adx.py,sha256=p_MT-jBuBdCyubxy-xveWLu5UUreSLTHlJG5Z1M7OJA,9407
-airflow/providers/microsoft/azure/hooks/asb.py,sha256=Ke2V45aiokrsHxkh23T5-VuMuNPDzfsOw099QPhjPQ8,13275
-airflow/providers/microsoft/azure/hooks/base_azure.py,sha256=zwXLtzraBh9oJqUYVM7Cp1d78RiPEc5unVT6LX1pX98,5838
-airflow/providers/microsoft/azure/hooks/batch.py,sha256=_TA0AYHMWFvhXD_O1S9km4lqToRIStZBPsivHB-tLso,15662
-airflow/providers/microsoft/azure/hooks/container_instance.py,sha256=IwETh1n9jdA3E-Tgj7Ljmw9lvJ5jFIufZiD7Ho-8wBE,8612
-airflow/providers/microsoft/azure/hooks/container_registry.py,sha256=XvrEipY0N_-nkTuow_2LeyirNUR5sAPv5enQOzphMjE,4353
-airflow/providers/microsoft/azure/hooks/container_volume.py,sha256=ZBNpJcjAWpecrk7plZ9ugQ0X7tUX39e38XH8hdEv_cw,5312
-airflow/providers/microsoft/azure/hooks/cosmos.py,sha256=v2nYHFBdsFYiP9HoRqiunY9-7xocTYGli85twnT20bo,16082
-airflow/providers/microsoft/azure/hooks/data_factory.py,sha256=dLm_3_pAbkJeAbKGDwzD3eAwr2mZTvIYXNH52VBVm8g,44636
-airflow/providers/microsoft/azure/hooks/data_lake.py,sha256=Irgmvs73sPIKqQwY4YTafOMpFYJ2RYm94l8ATMPbNVk,22471
-airflow/providers/microsoft/azure/hooks/fileshare.py,sha256=v0ALFQULNiBagjjEr8ZhtC5KIfr3rj9ykIIdONZOSTY,10125
-airflow/providers/microsoft/azure/hooks/synapse.py,sha256=_S0PMVqiSar2zB5sfZpnfjNgNgDE_6fP06fTyNjQDCo,7353
-airflow/providers/microsoft/azure/hooks/wasb.py,sha256=kKBA5BO_QLA6uv31xEsK64E0t4M7oD0EqNr3AIHmWYk,30165
+airflow/providers/microsoft/azure/hooks/adx.py,sha256=4a2oBs-x3VNJrjxCqSA6l7hjSn8z8kbGAnmpC_hJCM0,9934
+airflow/providers/microsoft/azure/hooks/asb.py,sha256=VKLLgGSj_SnBB-74vPxtc2N0mVduBJEtXqU_EAXvouE,14427
+airflow/providers/microsoft/azure/hooks/base_azure.py,sha256=yVO6cXYcdaHwLLrXqtXvCN9HeQRGpnyk60lOmntOehA,6282
+airflow/providers/microsoft/azure/hooks/batch.py,sha256=VdAix6NcsAH9_uDWPz7pUNeTh8sTmC8mUgo6zBI_f8c,16114
+airflow/providers/microsoft/azure/hooks/container_instance.py,sha256=B9uCgpy-q7Pd_FQ0BdvcJ_opYQbRJfVvw0dOdTXfl3c,9054
+airflow/providers/microsoft/azure/hooks/container_registry.py,sha256=QCoTcqVY5o1O5fR-5KxoHRcdgTtyM9Spc6ktOrz1Tho,4854
+airflow/providers/microsoft/azure/hooks/container_volume.py,sha256=2b31RtjMQvLIyyRvKJ1yyykeMrPhWF-To5V8OkGefUM,5756
+airflow/providers/microsoft/azure/hooks/cosmos.py,sha256=zgtGHqeAOiFpFhvIyumAB6c89vjeP28NkQZN9KV9H88,16642
+airflow/providers/microsoft/azure/hooks/data_factory.py,sha256=Mpv6FsldMp_ToEJ1JkWZ-eJ4HxQhxzWrFK6U_DEnCqc,45553
+airflow/providers/microsoft/azure/hooks/data_lake.py,sha256=_OpZUzng8SvV0FDOprBFFWRKuXubxbEouVIxhobpwYE,23335
+airflow/providers/microsoft/azure/hooks/fileshare.py,sha256=8evFa4csjcRScW_XJBgBw-ZJWSH0SL9CGfJIcOkO9bU,10834
+airflow/providers/microsoft/azure/hooks/synapse.py,sha256=MEzC4nsWjKbQHEydFljDVb6AZapKOtSXLj_wo1AmqwQ,7914
+airflow/providers/microsoft/azure/hooks/wasb.py,sha256=ZgxbL0CXuzBWZG3hcwDSYoox60b6nzOZoxDlqM7gdMQ,31111
 airflow/providers/microsoft/azure/log/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/microsoft/azure/log/wasb_task_handler.py,sha256=uq7CPplHQCAJCMxxeTdTnmsKwq82olihBXYPh9LmbNI,9949
 airflow/providers/microsoft/azure/operators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
@@ -24,13 +24,13 @@ airflow/providers/microsoft/azure/operators/adls.py,sha256=msx7uZNlSshiZsNASESk6
 airflow/providers/microsoft/azure/operators/adx.py,sha256=H7Q6BLdO-WZUV0UUxqtR8e-rRdvJ8qo2dk-6VpJj3K4,3408
 airflow/providers/microsoft/azure/operators/asb.py,sha256=DDOH-EMSuQNn8PfSqdsFw5Ks81GAfavvxzKjchdZgTk,29420
 airflow/providers/microsoft/azure/operators/batch.py,sha256=BxhI8H9vzCtvPZo7bjQtNirRAlRCOyo8UK2D7GZCJkk,16538
-airflow/providers/microsoft/azure/operators/container_instances.py,sha256=L_eXOkACe0YiHCqujgDdtNV4MVCDTkkGZ2P2wJfAddg,16137
+airflow/providers/microsoft/azure/operators/container_instances.py,sha256=uZahD2uIYO0sFeUWeM7Nk3J5QWzCcq2gcKv4SEgSfwY,16084
 airflow/providers/microsoft/azure/operators/cosmos.py,sha256=Y1Hj4p6W8soVFaq1rx8LFgchNISjkq8vjdaQ0j8Tnqs,2782
 airflow/providers/microsoft/azure/operators/data_factory.py,sha256=-AZNA51CTiFfyl0DUJL_0LlDHg6uGVVtt0YID7iX910,12377
 airflow/providers/microsoft/azure/operators/synapse.py,sha256=j3HC4SbjJKXwso8gkw7gKwvnjPByWBKGxVecH5HaVvM,4541
 airflow/providers/microsoft/azure/operators/wasb_delete_blob.py,sha256=GUfV9DLU1bGYvn2TE54iTYBTbxn3Jm_e985pXp_0IsE,2687
 airflow/providers/microsoft/azure/secrets/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/microsoft/azure/secrets/key_vault.py,sha256=W3UbjroulgbVG4ILedHYO5nZ2JLk-XI65q9EcCBpQNM,9063
+airflow/providers/microsoft/azure/secrets/key_vault.py,sha256=S9VpCe-qUqTyU-Ym8_kpAL3Q47uUL_w6yGaf480mAS8,9928
 airflow/providers/microsoft/azure/sensors/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
 airflow/providers/microsoft/azure/sensors/cosmos.py,sha256=vRrJ8zJnApvuKxHia53tNZUZ7wILWFT3_5cEyMA2M1I,2637
 airflow/providers/microsoft/azure/sensors/data_factory.py,sha256=P2XkREfInc0kAvTB1kdKejs_7kvbBTcxmu_ShkW3cCo,5553
@@ -44,10 +44,10 @@ airflow/providers/microsoft/azure/transfers/sftp_to_wasb.py,sha256=sx3fCZdjQ_AVx
 airflow/providers/microsoft/azure/triggers/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/microsoft/azure/triggers/data_factory.py,sha256=Td65obpw4fvXjjmF8kB-b9W994GueXTMm0mfR3Gy8FA,11130
 airflow/providers/microsoft/azure/triggers/wasb.py,sha256=PizofSGAwdgCuswenlbzlH0Jr7GHEzT89BluucDt8dE,7382
-apache_airflow_providers_microsoft_azure-8.1.0rc1.dist-info/LICENSE,sha256=gXPVwptPlW1TJ4HSuG5OMPg-a3h43OGMkZRR1rpwfJA,10850
-apache_airflow_providers_microsoft_azure-8.1.0rc1.dist-info/METADATA,sha256=RuHaisSCWlHughnw1PFZCfdW8K25-k6E00zOitbf1yE,7753
-apache_airflow_providers_microsoft_azure-8.1.0rc1.dist-info/NOTICE,sha256=m-6s2XynUxVSUIxO4rVablAZCvFq-wmLrqV91DotRBw,240
-apache_airflow_providers_microsoft_azure-8.1.0rc1.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-apache_airflow_providers_microsoft_azure-8.1.0rc1.dist-info/entry_points.txt,sha256=mAQpo-U_MYTG3pa7xmaY5vG849FnAziWsdGpfV-16NM,112
-apache_airflow_providers_microsoft_azure-8.1.0rc1.dist-info/top_level.txt,sha256=OeMVH5md7fr2QQWpnZoOWWxWO-0WH1IP70lpTVwopPg,8
-apache_airflow_providers_microsoft_azure-8.1.0rc1.dist-info/RECORD,,
+apache_airflow_providers_microsoft_azure-8.2.0rc1.dist-info/LICENSE,sha256=gXPVwptPlW1TJ4HSuG5OMPg-a3h43OGMkZRR1rpwfJA,10850
+apache_airflow_providers_microsoft_azure-8.2.0rc1.dist-info/METADATA,sha256=7AUdA4dGMQb98FkFN1v-2lWMGvI8ojL5bPq09dwJ8Ro,7750
+apache_airflow_providers_microsoft_azure-8.2.0rc1.dist-info/NOTICE,sha256=m-6s2XynUxVSUIxO4rVablAZCvFq-wmLrqV91DotRBw,240
+apache_airflow_providers_microsoft_azure-8.2.0rc1.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
+apache_airflow_providers_microsoft_azure-8.2.0rc1.dist-info/entry_points.txt,sha256=mAQpo-U_MYTG3pa7xmaY5vG849FnAziWsdGpfV-16NM,112
+apache_airflow_providers_microsoft_azure-8.2.0rc1.dist-info/top_level.txt,sha256=OeMVH5md7fr2QQWpnZoOWWxWO-0WH1IP70lpTVwopPg,8
+apache_airflow_providers_microsoft_azure-8.2.0rc1.dist-info/RECORD,,

{apache_airflow_providers_microsoft_azure-8.1.0rc1.dist-info → apache_airflow_providers_microsoft_azure-8.2.0rc1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.41.2)
+Generator: bdist_wheel (0.41.3)
 Root-Is-Purelib: true
 Tag: py3-none-any