apache-airflow-providers-microsoft-azure 12.2.0__py3-none-any.whl → 12.2.1__py3-none-any.whl

airflow/providers/microsoft/azure/__init__.py

@@ -29,7 +29,7 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "12.2.0"
+__version__ = "12.2.1"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "2.9.0"

airflow/providers/microsoft/azure/fs/adls.py

@@ -18,9 +18,10 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any
 
+from azure.identity import ClientSecretCredential
+
 from airflow.hooks.base import BaseHook
 from airflow.providers.microsoft.azure.utils import get_field, parse_blob_account_url
-from azure.identity import ClientSecretCredential
 
 if TYPE_CHECKING:
     from fsspec import AbstractFileSystem

airflow/providers/microsoft/azure/get_provider_info.py

@@ -27,8 +27,9 @@ def get_provider_info():
         "name": "Microsoft Azure",
         "description": "`Microsoft Azure <https://azure.microsoft.com/>`__\n",
         "state": "ready",
-        "source-date-epoch": 1739963698,
+        "source-date-epoch": 1741508992,
         "versions": [
+            "12.2.1",
             "12.2.0",
             "12.0.0",
             "11.1.0",

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

@@ -30,14 +30,15 @@ import warnings
 from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
+from azure.kusto.data import ClientRequestProperties, KustoClient, KustoConnectionStringBuilder
+from azure.kusto.data.exceptions import KustoServiceError
+
 from airflow.exceptions import AirflowException
 from airflow.hooks.base import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     add_managed_identity_connection_widgets,
     get_sync_default_azure_credential,
 )
-from azure.kusto.data import ClientRequestProperties, KustoClient, KustoConnectionStringBuilder
-from azure.kusto.data.exceptions import KustoServiceError
 
 if TYPE_CHECKING:
     from azure.kusto.data.response import KustoResponseDataSet

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

@@ -19,12 +19,6 @@ from __future__ import annotations
 from typing import TYPE_CHECKING, Any, Callable
 from uuid import uuid4
 
-from airflow.hooks.base import BaseHook
-from airflow.providers.microsoft.azure.utils import (
-    add_managed_identity_connection_widgets,
-    get_field,
-    get_sync_default_azure_credential,
-)
 from azure.core.exceptions import ResourceNotFoundError
 from azure.servicebus import (
     ServiceBusClient,
@@ -42,12 +36,20 @@ from azure.servicebus.management import (
     SubscriptionProperties,
 )
 
+from airflow.hooks.base import BaseHook
+from airflow.providers.microsoft.azure.utils import (
+    add_managed_identity_connection_widgets,
+    get_field,
+    get_sync_default_azure_credential,
+)
+
 if TYPE_CHECKING:
     import datetime
 
-    from airflow.utils.context import Context
     from azure.identity import DefaultAzureCredential
 
+    from airflow.utils.context import Context
+
     MessageCallback = Callable[[ServiceBusMessage, Context], None]
 
 

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

@@ -18,14 +18,15 @@ from __future__ import annotations
 
 from typing import Any
 
+from azure.common.client_factory import get_client_from_auth_file, get_client_from_json_dict
+from azure.common.credentials import ServicePrincipalCredentials
+
 from airflow.exceptions import AirflowException
 from airflow.hooks.base import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     AzureIdentityCredentialAdapter,
     add_managed_identity_connection_widgets,
 )
-from azure.common.client_factory import get_client_from_auth_file, get_client_from_json_dict
-from azure.common.credentials import ServicePrincipalCredentials
 
 
 class AzureBaseHook(BaseHook):

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

@@ -22,6 +22,8 @@ from datetime import timedelta
 from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
+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 (
@@ -30,7 +32,6 @@ from airflow.providers.microsoft.azure.utils import (
     get_field,
 )
 from airflow.utils import timezone
-from azure.batch import BatchServiceClient, batch_auth, models as batch_models
 
 if TYPE_CHECKING:
     from azure.batch.models import JobAddParameter, PoolAddParameter, TaskAddParameter

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

@@ -20,13 +20,14 @@ from __future__ import annotations
 from functools import cached_property
 from typing import TYPE_CHECKING, Any, cast
 
-from airflow.exceptions import AirflowException
-from airflow.providers.microsoft.azure.hooks.base_azure import AzureBaseHook
-from airflow.providers.microsoft.azure.utils import get_sync_default_azure_credential
 from azure.common.client_factory import get_client_from_auth_file, get_client_from_json_dict
 from azure.identity import ClientSecretCredential, DefaultAzureCredential
 from azure.mgmt.containerinstance import ContainerInstanceManagementClient
 
+from airflow.exceptions import AirflowException
+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 (
         ContainerGroup,

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

@@ -22,14 +22,15 @@ from __future__ import annotations
 from functools import cached_property
 from typing import Any
 
+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 (
     add_managed_identity_connection_widgets,
     get_field,
     get_sync_default_azure_credential,
 )
-from azure.mgmt.containerinstance.models import ImageRegistryCredential
-from azure.mgmt.containerregistry import ContainerRegistryManagementClient
 
 
 class AzureContainerRegistryHook(BaseHook):

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

@@ -18,14 +18,15 @@ from __future__ import annotations
 
 from typing import Any
 
+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 (
     add_managed_identity_connection_widgets,
     get_field,
     get_sync_default_azure_credential,
 )
-from azure.mgmt.containerinstance.models import AzureFileVolume, Volume
-from azure.mgmt.storage import StorageManagementClient
 
 
 class AzureContainerVolumeHook(BaseHook):

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

@@ -30,6 +30,11 @@ import uuid
 from typing import TYPE_CHECKING, Any, Union
 from urllib.parse import urlparse
 
+from azure.cosmos import PartitionKey
+from azure.cosmos.cosmos_client import CosmosClient
+from azure.cosmos.exceptions import CosmosHttpResponseError
+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 (
@@ -37,10 +42,6 @@ from airflow.providers.microsoft.azure.utils import (
     get_field,
     get_sync_default_azure_credential,
 )
-from azure.cosmos import PartitionKey
-from azure.cosmos.cosmos_client import CosmosClient
-from azure.cosmos.exceptions import CosmosHttpResponseError
-from azure.mgmt.cosmosdb import CosmosDBManagementClient
 
 if TYPE_CHECKING:
     PartitionKeyType = Union[str, list[str]]

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

@@ -39,6 +39,13 @@ from functools import wraps
 from typing import IO, TYPE_CHECKING, Any, Callable, TypeVar, Union, cast
 
 from asgiref.sync import sync_to_async
+from azure.identity import ClientSecretCredential, DefaultAzureCredential
+from azure.identity.aio import (
+    ClientSecretCredential as AsyncClientSecretCredential,
+    DefaultAzureCredential as AsyncDefaultAzureCredential,
+)
+from azure.mgmt.datafactory import DataFactoryManagementClient
+from azure.mgmt.datafactory.aio import DataFactoryManagementClient as AsyncDataFactoryManagementClient
 
 from airflow.exceptions import AirflowException
 from airflow.hooks.base import BaseHook
@@ -47,13 +54,6 @@ from airflow.providers.microsoft.azure.utils import (
     get_async_default_azure_credential,
     get_sync_default_azure_credential,
 )
-from azure.identity import ClientSecretCredential, DefaultAzureCredential
-from azure.identity.aio import (
-    ClientSecretCredential as AsyncClientSecretCredential,
-    DefaultAzureCredential as AsyncDefaultAzureCredential,
-)
-from azure.mgmt.datafactory import DataFactoryManagementClient
-from azure.mgmt.datafactory.aio import DataFactoryManagementClient as AsyncDataFactoryManagementClient
 
 if TYPE_CHECKING:
     from azure.core.polling import LROPoller

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

@@ -20,14 +20,6 @@ from __future__ import annotations
 from functools import cached_property
 from typing import Any, Union
 
-from airflow.exceptions import AirflowException
-from airflow.hooks.base import BaseHook
-from airflow.providers.microsoft.azure.utils import (
-    AzureIdentityCredentialAdapter,
-    add_managed_identity_connection_widgets,
-    get_field,
-    get_sync_default_azure_credential,
-)
 from azure.core.exceptions import ResourceExistsError, ResourceNotFoundError
 from azure.datalake.store import core, lib, multithread
 from azure.identity import ClientSecretCredential, DefaultAzureCredential
@@ -40,6 +32,15 @@ from azure.storage.filedatalake import (
     FileSystemProperties,
 )
 
+from airflow.exceptions import AirflowException
+from airflow.hooks.base import BaseHook
+from airflow.providers.microsoft.azure.utils import (
+    AzureIdentityCredentialAdapter,
+    add_managed_identity_connection_widgets,
+    get_field,
+    get_sync_default_azure_credential,
+)
+
 Credentials = Union[ClientSecretCredential, AzureIdentityCredentialAdapter, DefaultAzureCredential]
 
 

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

@@ -19,12 +19,13 @@ from __future__ import annotations
 
 from typing import IO, Any
 
+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,
 )
-from azure.storage.fileshare import FileProperties, ShareDirectoryClient, ShareFileClient, ShareServiceClient
 
 
 class AzureFileShareHook(BaseHook):

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

@@ -27,6 +27,7 @@ from typing import TYPE_CHECKING, Any
 from urllib.parse import quote, urljoin, urlparse
 
 import httpx
+from azure.identity import CertificateCredential, ClientSecretCredential
 from httpx import AsyncHTTPTransport, Timeout
 from kiota_abstractions.api_error import APIError
 from kiota_abstractions.method import Method
@@ -50,9 +51,9 @@ from airflow.exceptions import (
     AirflowNotFoundException,
 )
 from airflow.hooks.base import BaseHook
-from azure.identity import CertificateCredential, ClientSecretCredential
 
 if TYPE_CHECKING:
+    from azure.identity._internal.client_credential_base import ClientCredentialBase
     from kiota_abstractions.request_adapter import RequestAdapter
     from kiota_abstractions.request_information import QueryParams
     from kiota_abstractions.response_handler import NativeResponseType
@@ -60,7 +61,6 @@ if TYPE_CHECKING:
     from kiota_http.httpx_request_adapter import ResponseType
 
     from airflow.models import Connection
-    from azure.identity._internal.client_credential_base import ClientCredentialBase
 
 
 class DefaultResponseHandler(ResponseHandler):

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

@@ -19,6 +19,11 @@ from __future__ import annotations
 import time
 from typing import TYPE_CHECKING, Any, Union
 
+from azure.core.exceptions import ServiceRequestError
+from azure.identity import ClientSecretCredential, DefaultAzureCredential
+from azure.synapse.artifacts import ArtifactsClient
+from azure.synapse.spark import SparkClient
+
 from airflow.exceptions import AirflowException, AirflowTaskTimeout
 from airflow.hooks.base import BaseHook
 from airflow.providers.microsoft.azure.utils import (
@@ -26,10 +31,6 @@ from airflow.providers.microsoft.azure.utils import (
     get_field,
     get_sync_default_azure_credential,
 )
-from azure.core.exceptions import ServiceRequestError
-from azure.identity import ClientSecretCredential, DefaultAzureCredential
-from azure.synapse.artifacts import ArtifactsClient
-from azure.synapse.spark import SparkClient
 
 if TYPE_CHECKING:
     from azure.synapse.artifacts.models import CreateRunResponse, PipelineRun

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

@@ -32,15 +32,6 @@ from functools import cached_property
 from typing import TYPE_CHECKING, Any, Union
 
 from asgiref.sync import sync_to_async
-
-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,
-    parse_blob_account_url,
-)
 from azure.core.exceptions import HttpResponseError, ResourceExistsError, ResourceNotFoundError
 from azure.identity import ClientSecretCredential
 from azure.identity.aio import (
@@ -54,6 +45,15 @@ from azure.storage.blob.aio import (
     ContainerClient as AsyncContainerClient,
 )
 
+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,
+    parse_blob_account_url,
+)
+
 if TYPE_CHECKING:
     from azure.storage.blob._models import BlobProperties
 

airflow/providers/microsoft/azure/log/wasb_task_handler.py

@@ -23,10 +23,12 @@ from functools import cached_property
 from pathlib import Path
 from typing import TYPE_CHECKING
 
+from azure.core.exceptions import HttpResponseError
+
 from airflow.configuration import conf
+from airflow.providers.microsoft.azure.version_compat import AIRFLOW_V_3_0_PLUS
 from airflow.utils.log.file_task_handler import FileTaskHandler
 from airflow.utils.log.logging_mixin import LoggingMixin
-from azure.core.exceptions import HttpResponseError
 
 if TYPE_CHECKING:
     import logging
@@ -136,9 +138,13 @@ class WasbTaskHandler(FileTaskHandler, LoggingMixin):
 
         if blob_names:
             uris = [f"https://{self.wasb_container}.blob.core.windows.net/{b}" for b in blob_names]
-            messages.extend(["Found remote logs:", *[f"  * {x}" for x in sorted(uris)]])
+            if AIRFLOW_V_3_0_PLUS:
+                messages = uris
+            else:
+                messages.extend(["Found remote logs:", *[f"  * {x}" for x in sorted(uris)]])
         else:
-            messages.append(f"No logs found in WASB; ti=%s {ti}")
+            if not AIRFLOW_V_3_0_PLUS:
+                messages.append(f"No logs found in WASB; ti={ti}")
 
         for name in sorted(blob_names):
             remote_log = ""

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

@@ -28,9 +28,10 @@ from airflow.models import BaseOperator
 from airflow.providers.microsoft.azure.hooks.adx import AzureDataExplorerHook
 
 if TYPE_CHECKING:
-    from airflow.utils.context import Context
     from azure.kusto.data._models import KustoResultTable
 
+    from airflow.utils.context import Context
+
 
 class AzureDataExplorerQueryOperator(BaseOperator):
     """

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

@@ -25,10 +25,11 @@ from airflow.providers.microsoft.azure.hooks.asb import AdminClientHook, Message
 if TYPE_CHECKING:
     import datetime
 
-    from airflow.utils.context import Context
     from azure.servicebus import ServiceBusMessage
     from azure.servicebus.management import AuthorizationRule, CorrelationRuleFilter, SqlRuleFilter
 
+    from airflow.utils.context import Context
+
     MessageCallback = Callable[[ServiceBusMessage, Context], None]
 
 

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

@@ -21,10 +21,11 @@ from collections.abc import Sequence
 from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
+from azure.batch import models as batch_models
+
 from airflow.exceptions import AirflowException
 from airflow.models import BaseOperator
 from airflow.providers.microsoft.azure.hooks.batch import AzureBatchHook
-from azure.batch import models as batch_models
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

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

@@ -23,13 +23,6 @@ from collections import namedtuple
 from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any
 
-from msrestazure.azure_exceptions import CloudError
-
-from airflow.exceptions import AirflowException, AirflowTaskTimeout
-from airflow.models import BaseOperator
-from airflow.providers.microsoft.azure.hooks.container_instance import AzureContainerInstanceHook
-from airflow.providers.microsoft.azure.hooks.container_registry import AzureContainerRegistryHook
-from airflow.providers.microsoft.azure.hooks.container_volume import AzureContainerVolumeHook
 from azure.mgmt.containerinstance.models import (
     Container,
     ContainerGroup,
@@ -44,6 +37,13 @@ from azure.mgmt.containerinstance.models import (
     Volume as _AzureVolume,
     VolumeMount,
 )
+from msrestazure.azure_exceptions import CloudError
+
+from airflow.exceptions import AirflowException, AirflowTaskTimeout
+from airflow.models import BaseOperator
+from airflow.providers.microsoft.azure.hooks.container_instance import AzureContainerInstanceHook
+from airflow.providers.microsoft.azure.hooks.container_registry import AzureContainerRegistryHook
+from airflow.providers.microsoft.azure.hooks.container_volume import AzureContainerVolumeHook
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

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

@@ -25,7 +25,7 @@ from typing import TYPE_CHECKING, Any
 from airflow.configuration import conf
 from airflow.exceptions import AirflowException
 from airflow.hooks.base import BaseHook
-from airflow.models import BaseOperator, BaseOperatorLink, XCom
+from airflow.models import BaseOperator, XCom
 from airflow.providers.microsoft.azure.hooks.data_factory import (
     AzureDataFactoryHook,
     AzureDataFactoryPipelineRunException,
@@ -39,6 +39,13 @@ if TYPE_CHECKING:
     from airflow.models.taskinstancekey import TaskInstanceKey
     from airflow.utils.context import Context
 
+from airflow.providers.microsoft.azure.version_compat import AIRFLOW_V_3_0_PLUS
+
+if AIRFLOW_V_3_0_PLUS:
+    from airflow.sdk import BaseOperatorLink
+else:
+    from airflow.models.baseoperatorlink import BaseOperatorLink  # type: ignore[no-redef]
+
 
 class AzureDataFactoryPipelineRunLink(LoggingMixin, BaseOperatorLink):
     """Construct a link to monitor a pipeline run in Azure Data Factory."""

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

@@ -319,7 +319,15 @@ class MSGraphAsyncOperator(BaseOperator):
 
     def trigger_next_link(self, response, method_name: str, context: Context) -> None:
         if isinstance(response, dict):
-            url, query_parameters = self.pagination_function(self, response, **dict(context.items()))  # type: ignore
+            try:
+                url, query_parameters = self.pagination_function(self, response, **dict(context.items()))  # type: ignore
+            except TypeError:
+                warnings.warn(
+                    "pagination_function signature has changed, context parameter should be a kwargs argument!",
+                    AirflowProviderDeprecationWarning,
+                    stacklevel=2,
+                )
+                url, query_parameters = self.pagination_function(self, response, context)  # type: ignore
 
             self.log.debug("url: %s", url)
             self.log.debug("query_parameters: %s", query_parameters)

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

@@ -21,7 +21,7 @@ from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any
 
 from airflow.exceptions import AirflowException
-from airflow.models import BaseOperator, BaseOperatorLink
+from airflow.models import BaseOperator
 from airflow.providers.microsoft.azure.hooks.powerbi import PowerBIHook
 from airflow.providers.microsoft.azure.triggers.powerbi import PowerBITrigger
 
@@ -31,6 +31,13 @@ if TYPE_CHECKING:
     from airflow.models.taskinstancekey import TaskInstanceKey
     from airflow.utils.context import Context
 
+from airflow.providers.microsoft.azure.version_compat import AIRFLOW_V_3_0_PLUS
+
+if AIRFLOW_V_3_0_PLUS:
+    from airflow.sdk import BaseOperatorLink
+else:
+    from airflow.models.baseoperatorlink import BaseOperatorLink  # type: ignore[no-redef]
+
 
 class PowerBILink(BaseOperatorLink):
     """Construct a link to monitor a dataset in Power BI."""

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

@@ -23,7 +23,7 @@ from urllib.parse import urlencode
 
 from airflow.exceptions import AirflowException
 from airflow.hooks.base import BaseHook
-from airflow.models import BaseOperator, BaseOperatorLink, XCom
+from airflow.models import BaseOperator, XCom
 from airflow.providers.microsoft.azure.hooks.synapse import (
     AzureSynapseHook,
     AzureSynapsePipelineHook,
@@ -33,9 +33,17 @@ from airflow.providers.microsoft.azure.hooks.synapse import (
 )
 
 if TYPE_CHECKING:
+    from azure.synapse.spark.models import SparkBatchJobOptions
+
     from airflow.models.taskinstancekey import TaskInstanceKey
     from airflow.utils.context import Context
-    from azure.synapse.spark.models import SparkBatchJobOptions
+
+from airflow.providers.microsoft.azure.version_compat import AIRFLOW_V_3_0_PLUS
+
+if AIRFLOW_V_3_0_PLUS:
+    from airflow.sdk import BaseOperatorLink
+else:
+    from airflow.models.baseoperatorlink import BaseOperatorLink  # type: ignore[no-redef]
 
 
 class AzureSynapseRunSparkBatchOperator(BaseOperator):

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

@@ -28,13 +28,14 @@ import logging
 import os
 from functools import cached_property
 
-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 azure.core.exceptions import ResourceNotFoundError
 from azure.identity import ClientSecretCredential, DefaultAzureCredential
 from azure.keyvault.secrets import SecretClient
 
+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
+
 
 class AzureKeyVaultBackend(BaseSecretsBackend, LoggingMixin):
     """

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

@@ -21,12 +21,13 @@ import time
 from collections.abc import AsyncIterator
 from typing import Any
 
+from azure.core.exceptions import ServiceRequestError
+
 from airflow.providers.microsoft.azure.hooks.data_factory import (
     AzureDataFactoryAsyncHook,
     AzureDataFactoryPipelineRunStatus,
 )
 from airflow.triggers.base import BaseTrigger, TriggerEvent
-from azure.core.exceptions import ServiceRequestError
 
 
 class ADFPipelineRunStatusSensorTrigger(BaseTrigger):

airflow/providers/microsoft/azure/utils.py

@@ -21,13 +21,12 @@ import warnings
 from functools import partial, wraps
 from urllib.parse import urlparse, urlunparse
 
-from msrest.authentication import BasicTokenAuthentication
-
 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
 
 
 def get_field(*, conn_id: str, conn_type: str, extras: dict, field_name: str):

airflow/providers/microsoft/azure/version_compat.py

@@ -0,0 +1,36 @@
+# 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
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+#
+# NOTE! THIS FILE IS COPIED MANUALLY IN OTHER PROVIDERS DELIBERATELY TO AVOID ADDING UNNECESSARY
+# DEPENDENCIES BETWEEN PROVIDERS. IF YOU WANT TO ADD CONDITIONAL CODE IN YOUR PROVIDER THAT DEPENDS
+# ON AIRFLOW VERSION, PLEASE COPY THIS FILE TO THE ROOT PACKAGE OF YOUR PROVIDER AND IMPORT
+# THOSE CONSTANTS FROM IT RATHER THAN IMPORTING THEM FROM ANOTHER PROVIDER OR TEST CODE
+#
+from __future__ import annotations
+
+
+def get_base_airflow_version_tuple() -> tuple[int, int, int]:
+    from packaging.version import Version
+
+    from airflow import __version__
+
+    airflow_version = Version(__version__)
+    return airflow_version.major, airflow_version.minor, airflow_version.micro
+
+
+AIRFLOW_V_2_10_PLUS = get_base_airflow_version_tuple() >= (2, 10, 0)
+AIRFLOW_V_3_0_PLUS = get_base_airflow_version_tuple() >= (3, 0, 0)

{apache_airflow_providers_microsoft_azure-12.2.0.dist-info → apache_airflow_providers_microsoft_azure-12.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: apache-airflow-providers-microsoft-azure
-Version: 12.2.0
+Version: 12.2.1
 Summary: Provider package apache-airflow-providers-microsoft-azure for Apache Airflow
 Keywords: airflow-provider,microsoft.azure,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -52,8 +52,8 @@ Requires-Dist: apache-airflow-providers-common-compat ; extra == "common-compat"
 Requires-Dist: apache-airflow-providers-oracle ; extra == "oracle"
 Requires-Dist: apache-airflow-providers-sftp ; extra == "sftp"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.2.0/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.2.0
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.2.1/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.2.1
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
 Project-URL: Twitter, https://x.com/ApacheAirflow
@@ -64,32 +64,31 @@ Provides-Extra: oracle
 Provides-Extra: sftp
 
 
- .. 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
-    to you under the Apache License, Version 2.0 (the
-    "License"); you may not use this file except in compliance
-    with the License.  You may obtain a copy of the License at
+.. 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
+   to you under the Apache License, Version 2.0 (the
+   "License"); you may not use this file except in compliance
+   with the License.  You may obtain a copy of the License at
 
- ..   http://www.apache.org/licenses/LICENSE-2.0
+..   http://www.apache.org/licenses/LICENSE-2.0
 
- .. Unless required by applicable law or agreed to in writing,
-    software distributed under the License is distributed on an
-    "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-    KIND, either express or implied.  See the License for the
-    specific language governing permissions and limitations
-    under the License.
+.. Unless required by applicable law or agreed to in writing,
+   software distributed under the License is distributed on an
+   "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+   KIND, either express or implied.  See the License for the
+   specific language governing permissions and limitations
+   under the License.
 
- .. NOTE! THIS FILE IS AUTOMATICALLY GENERATED AND WILL BE OVERWRITTEN!
-
- .. IF YOU WANT TO MODIFY TEMPLATE FOR THIS FILE, YOU SHOULD MODIFY THE TEMPLATE
-    `PROVIDER_README_TEMPLATE.rst.jinja2` IN the `dev/breeze/src/airflow_breeze/templates` DIRECTORY
+.. NOTE! THIS FILE IS AUTOMATICALLY GENERATED AND WILL BE OVERWRITTEN!
 
+.. IF YOU WANT TO MODIFY TEMPLATE FOR THIS FILE, YOU SHOULD MODIFY THE TEMPLATE
+   ``PROVIDER_README_TEMPLATE.rst.jinja2`` IN the ``dev/breeze/src/airflow_breeze/templates`` DIRECTORY
 
 Package ``apache-airflow-providers-microsoft-azure``
 
-Release: ``12.2.0``
+Release: ``12.2.1``
 
 
 `Microsoft Azure <https://azure.microsoft.com/>`__
@@ -102,7 +101,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/12.2.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.2.1/>`_.
 
 Installation
 ------------
@@ -171,5 +170,5 @@ Dependent package
 ==================================================================================================================  =================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.2.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.2.1/changelog.html>`_.
 

{apache_airflow_providers_microsoft_azure-12.2.0.dist-info → apache_airflow_providers_microsoft_azure-12.2.1.dist-info}/RECORD

@@ -1,41 +1,42 @@
 airflow/providers/microsoft/azure/LICENSE,sha256=gXPVwptPlW1TJ4HSuG5OMPg-a3h43OGMkZRR1rpwfJA,10850
-airflow/providers/microsoft/azure/__init__.py,sha256=akJ1rrzguxpg6Ie6GFTSIrKozfazd6HiDYrirm8FO7M,1503
-airflow/providers/microsoft/azure/get_provider_info.py,sha256=Kk2zV-N0wB-qU6Ds0JVJhnW2No-6XJqDyiSK4fXL5AI,21948
-airflow/providers/microsoft/azure/utils.py,sha256=D084V_bnrW9pfm5A4UBQgYFkqjgBArTR4hUe7TI-hSU,8430
+airflow/providers/microsoft/azure/__init__.py,sha256=t-zJdwMuY6XxnuoOKBQZdA_zJ2snJrGqM8oIju2yjfQ,1503
+airflow/providers/microsoft/azure/get_provider_info.py,sha256=07vD_khbf9kZngDpDmIECu1vBFKQYJvzVUWm__TBmTg,21970
+airflow/providers/microsoft/azure/utils.py,sha256=pvBiwBKRYMtEnqkQWLTqvhrsCOkMTRiPH97GkIYHUQc,8429
+airflow/providers/microsoft/azure/version_compat.py,sha256=aHg90_DtgoSnQvILFICexMyNlHlALBdaeWqkX3dFDug,1605
 airflow/providers/microsoft/azure/fs/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/microsoft/azure/fs/adls.py,sha256=0B5vXz9emkKd9CIbH8xWEjVBdKZR3tHs_wOT23XUI4Y,3678
+airflow/providers/microsoft/azure/fs/adls.py,sha256=kXZOVulLNfqwJNR0X9MBN23kcYr3dyv8K_0KqAcWvnk,3679
 airflow/providers/microsoft/azure/hooks/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/microsoft/azure/hooks/adx.py,sha256=Lw8DhtSVS1xOuCVBtBs0abx9vZiYjPbbN_qRdVDYdvM,9575
-airflow/providers/microsoft/azure/hooks/asb.py,sha256=1ZG-NKpe_L_aFAcNaGFnwUfc1ayRVsNgnQqnIhK9ddg,29802
-airflow/providers/microsoft/azure/hooks/base_azure.py,sha256=1Ct-uptTWBFzL7iW3QEp2wYJglslyAh24h3i7CFtExw,5388
-airflow/providers/microsoft/azure/hooks/batch.py,sha256=xNBNAnQgR50Dj8L3ACMM1yDrnaunY4YgCliKhsD7zns,16113
-airflow/providers/microsoft/azure/hooks/container_instance.py,sha256=KPo9_dREUrjgEc37_0MlK98DjRPdL-nLAcmfsQqj7o4,7137
-airflow/providers/microsoft/azure/hooks/container_registry.py,sha256=UGsXLj4SR8iwxYprU1oj-z_vRrwQYOGOlivaOYNoLZ0,4854
-airflow/providers/microsoft/azure/hooks/container_volume.py,sha256=uEb6MTZSptYY2Nve9jjdQnfn-VQV5BDWJhQQ0ydUW9E,5757
-airflow/providers/microsoft/azure/hooks/cosmos.py,sha256=4lj9J3bUvmveHWpjDnQRbu6YAg51CASBtjZ57e2AbD0,17465
-airflow/providers/microsoft/azure/hooks/data_factory.py,sha256=AtcQR9ZFqyzYX5x-jdS7_5qZZWmM4F-5cbT2Q3qnqPs,44986
-airflow/providers/microsoft/azure/hooks/data_lake.py,sha256=uar5VrbejiGkPzIenyLRkh1p5vMje3kb9i-siHDZew4,23823
-airflow/providers/microsoft/azure/hooks/fileshare.py,sha256=LQTOFe0AfX9ENjoSX91TKcV8WRBQdrgqEWItqxoIr_I,10883
-airflow/providers/microsoft/azure/hooks/msgraph.py,sha256=tFcAx5yjfqDuP_n4bIihASbTtTtVXyUrSTVntEJC3fM,20609
+airflow/providers/microsoft/azure/hooks/adx.py,sha256=FR7EC6ygDhcMK9clw-IDuoJqYtsrq-6WJOZF8ixl8Rw,9576
+airflow/providers/microsoft/azure/hooks/asb.py,sha256=gBNM_YiqrM2YamrugAbr5HNWAj2yW3qFGo39puxrA-I,29804
+airflow/providers/microsoft/azure/hooks/base_azure.py,sha256=ZNTjfKiHM91dQ8imc2VPxorj3fOlzkv2SgTxy008Zr0,5389
+airflow/providers/microsoft/azure/hooks/batch.py,sha256=BK2ciseUyGNFxOC0mEnPohZRfBlHf_UTyE0mB0tvn0g,16114
+airflow/providers/microsoft/azure/hooks/container_instance.py,sha256=BbhWTGhrTOejMJkSLZUG3msQXTpk8L0Ol2FQEDOFIGA,7138
+airflow/providers/microsoft/azure/hooks/container_registry.py,sha256=ii6GiOvl3p0lSGEyerHuhpvJC1bh18Asz4QmatXipYY,4855
+airflow/providers/microsoft/azure/hooks/container_volume.py,sha256=ZSe2C3y5xttxp8Ir9dgihZnQzcArckhPazxHg1iknYk,5758
+airflow/providers/microsoft/azure/hooks/cosmos.py,sha256=pE7MyNJx-u6pr_NlK5W2hv--wGBuDjSjPfeneZhky70,17466
+airflow/providers/microsoft/azure/hooks/data_factory.py,sha256=Yj65yLD2e_uIOyV8b4FN528tMj-6zz6PcHIvepdfvPE,44986
+airflow/providers/microsoft/azure/hooks/data_lake.py,sha256=3Ra6wCCp2qVnvXZb407W2G6UqRvM1J3JIMr4wfRMwyo,23824
+airflow/providers/microsoft/azure/hooks/fileshare.py,sha256=jaHSD_xZxposSD5FbdlpZ7JK_CugFHNrgejZkZbHJXM,10884
+airflow/providers/microsoft/azure/hooks/msgraph.py,sha256=zetub75VPUfhy5tOaqUVoo5z5juuM5gGGnLO2eOnX_k,20609
 airflow/providers/microsoft/azure/hooks/powerbi.py,sha256=RdhiHeJExf7D1M3HslagprGvM37aN_BWqQQ-xafz2z4,7501
-airflow/providers/microsoft/azure/hooks/synapse.py,sha256=GSSQgLM6fXhVVwsYK6Jc-eObrqTPbPK6xUkAuzeiDQc,16077
-airflow/providers/microsoft/azure/hooks/wasb.py,sha256=okwEvAKe_3zF8CQdRguQwtSDTeYAjRoegLa9JVZD4SE,30827
+airflow/providers/microsoft/azure/hooks/synapse.py,sha256=ya4bByYKgfvUyqs3H_jCUyu2HfI0lmhNSuoKguJrPHU,16078
+airflow/providers/microsoft/azure/hooks/wasb.py,sha256=NKChczpWD1IJ45Wr8PQR9I5Y0VH6mdGkk593Q8MZS0I,30827
 airflow/providers/microsoft/azure/log/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/microsoft/azure/log/wasb_task_handler.py,sha256=Ycof-KW8KXTMQKXPFfMfmhjbP2yIlIXbBxYLW6_iH20,8458
+airflow/providers/microsoft/azure/log/wasb_task_handler.py,sha256=jGK9oFfh-sYQYkt-ElebYAeLpGa0hYLbCEaaivQK6tE,8668
 airflow/providers/microsoft/azure/operators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
 airflow/providers/microsoft/azure/operators/adls.py,sha256=xLcaYKMFkygveTSGzFY1Y1oEzzliu3b77fumxYzYss8,5816
-airflow/providers/microsoft/azure/operators/adx.py,sha256=tMvCMrza3xQKi8CQem9-rfWSwxx5ZZIxiaIDrk0_yc0,3172
-airflow/providers/microsoft/azure/operators/asb.py,sha256=22hKBYocYZadIWLlEZTX7oksVp2uvOU1fjLv2aqvgEg,29643
-airflow/providers/microsoft/azure/operators/batch.py,sha256=ZP39_3zt3O1DTIo2fKJX9yJUcKmQSMvszFZWzulz7ns,16269
-airflow/providers/microsoft/azure/operators/container_instances.py,sha256=gFyrbWY_nBWrnTVb2Ept9NbgOpFESKztf0POsWRhYYg,18410
+airflow/providers/microsoft/azure/operators/adx.py,sha256=eUHEhJ18gL3kdCqNaQ-3BL5rSoorOxtMLBjIAx7xcxY,3173
+airflow/providers/microsoft/azure/operators/asb.py,sha256=xEgObtCO6oF88A0e0yeDk3JpLULmT5OEdPp9UfOhzyE,29644
+airflow/providers/microsoft/azure/operators/batch.py,sha256=QgtmaUYgejLtER68o0dDhLRJSGM4WqmstW5Guy7I30U,16270
+airflow/providers/microsoft/azure/operators/container_instances.py,sha256=EL2c4uBlFTgvi4gm8xNkxGnyhFnwI8IvltG-tcklOFY,18410
 airflow/providers/microsoft/azure/operators/cosmos.py,sha256=apq0wVYslRaO-ThXGnVyf7v_DovK951vaE7LxcVYU4E,2814
-airflow/providers/microsoft/azure/operators/data_factory.py,sha256=IzEdTC1vd3NSFFQSlKLIi2n8hbS3WaEWVA3AKdIKoss,12468
-airflow/providers/microsoft/azure/operators/msgraph.py,sha256=S5euMx896RyJcRFcJeiD06Hg_tCurNE8zfFTa6ok6D0,13150
-airflow/providers/microsoft/azure/operators/powerbi.py,sha256=oPHyAIalDvI5kvGqklaWEsHS34OPmZljhIbDlEFpe7Q,6055
-airflow/providers/microsoft/azure/operators/synapse.py,sha256=nGRk4BIWLo3Hf-lgkSXq-EK6CdyhN8TwNlbu12YEzP0,12412
+airflow/providers/microsoft/azure/operators/data_factory.py,sha256=ZS6Tz_yRypWM4x5SOdvqW8RGaXsJq0oaLWDWCYJO4ag,12697
+airflow/providers/microsoft/azure/operators/msgraph.py,sha256=jb0yA7GLaoK3QVhlRq_PcsmCQm8SrM6LkbHEKxw9PxA,13558
+airflow/providers/microsoft/azure/operators/powerbi.py,sha256=Lp6uzuLw_hNNrQJVy6XhCMCS2wNqhA4OEyTuDHUxMNo,6284
+airflow/providers/microsoft/azure/operators/synapse.py,sha256=Cc2L-tI3CHHNgVUhAkjLz-1NQhMDxWMGek2HtjLEOZ8,12642
 airflow/providers/microsoft/azure/operators/wasb_delete_blob.py,sha256=RQQz_xvcjyfY3AuIVkK-J-rkPF3rT-z4SJNPY-fpyZQ,2714
 airflow/providers/microsoft/azure/secrets/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/microsoft/azure/secrets/key_vault.py,sha256=7095n1X99UEYfHayI5VQRqeaSG3qNtfNNdSJa-A7MCs,8961
+airflow/providers/microsoft/azure/secrets/key_vault.py,sha256=EmtGfyBtfefGu1ZTtZ5WswIleOx-nx8wst7xfcua2rI,8962
 airflow/providers/microsoft/azure/sensors/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
 airflow/providers/microsoft/azure/sensors/cosmos.py,sha256=MI1MTpWavmMkb4-vG56I6B8dKISj4xTjx8yWPT_Fbpc,2664
 airflow/providers/microsoft/azure/sensors/data_factory.py,sha256=EEkXMZvB5vZyq0UPHyifRzcwSHgHqdwjtsrT49VsKU4,5014
@@ -48,11 +49,11 @@ airflow/providers/microsoft/azure/transfers/oracle_to_azure_data_lake.py,sha256=
 airflow/providers/microsoft/azure/transfers/s3_to_wasb.py,sha256=53If9gvBbMnCnjdncwzXQgUHMG1DB4wYftfX82LYieg,12509
 airflow/providers/microsoft/azure/transfers/sftp_to_wasb.py,sha256=UU_PFJlmsd8xz3rM7qZdndIUmWYfM7E4QYUN293hHD4,8222
 airflow/providers/microsoft/azure/triggers/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/microsoft/azure/triggers/data_factory.py,sha256=Y3H6Wa1iasCxvTzTolc4vaq5WGXk4CTij2V4QQtviVM,11164
+airflow/providers/microsoft/azure/triggers/data_factory.py,sha256=U3vY_pj4yORxE7X6YR7CP3Jl73K9euCdUczy3dLmikU,11165
 airflow/providers/microsoft/azure/triggers/msgraph.py,sha256=Hsyh9XLtxGhRzxK34xjh5iW9_ghFUHsN0_I7QbLe9Uk,8850
 airflow/providers/microsoft/azure/triggers/powerbi.py,sha256=14RFR5HfuoX4qKG_FkFc72GhEf34ISg3TxQkBAD9TpA,9758
 airflow/providers/microsoft/azure/triggers/wasb.py,sha256=RF-C6iqDEs6_pWireCWZXqxcqWK-sFJ695Okdd_EJOA,7456
-apache_airflow_providers_microsoft_azure-12.2.0.dist-info/entry_points.txt,sha256=6iWHenOoUC3YZBb3OKn6g0HlJsV58Ba56i8USmQrcJI,111
-apache_airflow_providers_microsoft_azure-12.2.0.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
-apache_airflow_providers_microsoft_azure-12.2.0.dist-info/METADATA,sha256=tQAKBY1fyp9AmSGKnqEDUs3DLtCrpRvkUlpQxF1pc_E,8380
-apache_airflow_providers_microsoft_azure-12.2.0.dist-info/RECORD,,
+apache_airflow_providers_microsoft_azure-12.2.1.dist-info/entry_points.txt,sha256=6iWHenOoUC3YZBb3OKn6g0HlJsV58Ba56i8USmQrcJI,111
+apache_airflow_providers_microsoft_azure-12.2.1.dist-info/WHEEL,sha256=_2ozNFCLWc93bK4WKHCO-eDUENDlo-dgc9cU3qokYO4,82
+apache_airflow_providers_microsoft_azure-12.2.1.dist-info/METADATA,sha256=qlpdR62vYTHlpwpUDFqhcvOnVa488AMH9ERJ4mDUFos,8366
+apache_airflow_providers_microsoft_azure-12.2.1.dist-info/RECORD,,

{apache_airflow_providers_microsoft_azure-12.2.0.dist-info → apache_airflow_providers_microsoft_azure-12.2.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: flit 3.10.1
+Generator: flit 3.11.0
 Root-Is-Purelib: true
 Tag: py3-none-any