apache-airflow-providers-microsoft-azure 12.7.0__py3-none-any.whl → 12.8.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.7.0"
+__version__ = "12.8.1"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "2.10.0"

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

@@ -20,8 +20,8 @@ from typing import TYPE_CHECKING, Any
 
 from azure.identity import ClientSecretCredential
 
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import get_field, parse_blob_account_url
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 
 if TYPE_CHECKING:
     from fsspec import AbstractFileSystem

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

@@ -0,0 +1,111 @@
+# 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.
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from airflow.providers.common.compat.sdk import BaseHook
+from airflow.providers.microsoft.azure.utils import get_field
+
+if TYPE_CHECKING:
+    from fsspec import AbstractFileSystem
+
+schemes = ["msgraph", "sharepoint", "onedrive", "msgd"]
+
+
+def get_fs(conn_id: str | None, storage_options: dict[str, Any] | None = None) -> AbstractFileSystem:
+    from msgraphfs import MSGDriveFS
+
+    if conn_id is None:
+        return MSGDriveFS({})
+
+    conn = BaseHook.get_connection(conn_id)
+    extras = conn.extra_dejson
+    conn_type = conn.conn_type or "msgraph"
+
+    options: dict[str, Any] = {}
+
+    # Get authentication parameters with fallback handling
+    client_id = conn.login or get_field(
+        conn_id=conn_id, conn_type=conn_type, extras=extras, field_name="client_id"
+    )
+    client_secret = conn.password or get_field(
+        conn_id=conn_id, conn_type=conn_type, extras=extras, field_name="client_secret"
+    )
+    tenant_id = conn.host or get_field(
+        conn_id=conn_id, conn_type=conn_type, extras=extras, field_name="tenant_id"
+    )
+
+    if client_id:
+        options["client_id"] = client_id
+    if client_secret:
+        options["client_secret"] = client_secret
+    if tenant_id:
+        options["tenant_id"] = tenant_id
+
+    # Process additional fields from extras
+    fields = [
+        "drive_id",
+        "scope",
+        "token_endpoint",
+        "redirect_uri",
+        "token_endpoint_auth_method",
+        "code_challenge_method",
+        "update_token",
+        "username",
+        "password",
+    ]
+    for field in fields:
+        value = get_field(conn_id=conn_id, conn_type=conn_type, extras=extras, field_name=field)
+        if value is not None:
+            if value == "":
+                options.pop(field, "")
+            else:
+                options[field] = value
+
+    # Update with storage options
+    options.update(storage_options or {})
+
+    # Create oauth2 client parameters if authentication is provided
+    oauth2_client_params = {}
+    if options.get("client_id") and options.get("client_secret") and options.get("tenant_id"):
+        oauth2_client_params = {
+            "client_id": options["client_id"],
+            "client_secret": options["client_secret"],
+            "tenant_id": options["tenant_id"],
+        }
+
+        # Add additional oauth2 parameters supported by authlib
+        oauth2_params = [
+            "scope",
+            "token_endpoint",
+            "redirect_uri",
+            "token_endpoint_auth_method",
+            "code_challenge_method",
+            "update_token",
+            "username",
+            "password",
+        ]
+        for param in oauth2_params:
+            if param in options:
+                oauth2_client_params[param] = options[param]
+
+    # Determine which filesystem to return based on drive_id
+    drive_id = options.get("drive_id")
+
+    return MSGDriveFS(drive_id=drive_id, oauth2_client_params=oauth2_client_params)

airflow/providers/microsoft/azure/get_provider_info.py

@@ -37,6 +37,7 @@ def get_provider_info():
             {
                 "integration-name": "Microsoft Azure Blob Storage",
                 "external-doc-url": "https://azure.microsoft.com/en-us/services/storage/blobs/",
+                "how-to-guide": ["/docs/apache-airflow-providers-microsoft-azure/sensors/wasb_sensors.rst"],
                 "logo": "/docs/integration-logos/Blob-Storage.svg",
                 "tags": ["azure"],
             },
@@ -49,6 +50,9 @@ def get_provider_info():
             {
                 "integration-name": "Microsoft Azure Cosmos DB",
                 "external-doc-url": "https://azure.microsoft.com/en-us/services/cosmos-db/",
+                "how-to-guide": [
+                    "/docs/apache-airflow-providers-microsoft-azure/sensors/cosmos_document_sensor.rst"
+                ],
                 "logo": "/docs/integration-logos/Azure-Cosmos-DB.svg",
                 "tags": ["azure"],
             },
@@ -117,7 +121,10 @@ def get_provider_info():
                 "integration-name": "Microsoft Graph API",
                 "external-doc-url": "https://learn.microsoft.com/en-us/graph/use-the-api/",
                 "logo": "/docs/integration-logos/Microsoft-Graph-API.png",
-                "how-to-guide": ["/docs/apache-airflow-providers-microsoft-azure/operators/msgraph.rst"],
+                "how-to-guide": [
+                    "/docs/apache-airflow-providers-microsoft-azure/operators/msgraph.rst",
+                    "/docs/apache-airflow-providers-microsoft-azure/sensors/msgraph.rst",
+                ],
                 "tags": ["azure"],
             },
             {
@@ -191,7 +198,10 @@ def get_provider_info():
                 "python-modules": ["airflow.providers.microsoft.azure.sensors.msgraph"],
             },
         ],
-        "filesystems": ["airflow.providers.microsoft.azure.fs.adls"],
+        "filesystems": [
+            "airflow.providers.microsoft.azure.fs.adls",
+            "airflow.providers.microsoft.azure.fs.msgraphfs",
+        ],
         "hooks": [
             {
                 "integration-name": "Microsoft Azure Container Instances",

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

@@ -34,11 +34,11 @@ from azure.kusto.data import ClientRequestProperties, KustoClient, KustoConnecti
 from azure.kusto.data.exceptions import KustoServiceError
 
 from airflow.exceptions import AirflowException
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     add_managed_identity_connection_widgets,
     get_sync_default_azure_credential,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 
 if TYPE_CHECKING:
     from azure.kusto.data.response import KustoResponseDataSet

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

@@ -37,12 +37,12 @@ from azure.servicebus.management import (
     SubscriptionProperties,
 )
 
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     add_managed_identity_connection_widgets,
     get_field,
     get_sync_default_azure_credential,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 
 if TYPE_CHECKING:
     import datetime

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

@@ -16,17 +16,24 @@
 # under the License.
 from __future__ import annotations
 
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from azure.common.client_factory import get_client_from_auth_file, get_client_from_json_dict
 from azure.common.credentials import ServicePrincipalCredentials
+from azure.identity import ClientSecretCredential, DefaultAzureCredential
 
 from airflow.exceptions import AirflowException
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     AzureIdentityCredentialAdapter,
     add_managed_identity_connection_widgets,
+    get_sync_default_azure_credential,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
+
+if TYPE_CHECKING:
+    from azure.core.credentials import AccessToken
+
+    from airflow.sdk import Connection
 
 
 class AzureBaseHook(BaseHook):
@@ -85,7 +92,7 @@ class AzureBaseHook(BaseHook):
             },
         }
 
-    def __init__(self, sdk_client: Any, conn_id: str = "azure_default"):
+    def __init__(self, sdk_client: Any = None, conn_id: str = "azure_default"):
         self.sdk_client = sdk_client
         self.conn_id = conn_id
         super().__init__()
@@ -96,8 +103,9 @@ class AzureBaseHook(BaseHook):
 
         :return: the authenticated client.
         """
+        if not self.sdk_client:
+            raise ValueError("`sdk_client` must be provided to AzureBaseHook to use `get_conn` method.")
         conn = self.get_connection(self.conn_id)
-        tenant = conn.extra_dejson.get("tenantId")
         subscription_id = conn.extra_dejson.get("subscriptionId")
         key_path = conn.extra_dejson.get("key_path")
         if key_path:
@@ -111,22 +119,90 @@ class AzureBaseHook(BaseHook):
             self.log.info("Getting connection using a JSON config.")
             return get_client_from_json_dict(client_class=self.sdk_client, config_dict=key_json)
 
-        credentials: ServicePrincipalCredentials | AzureIdentityCredentialAdapter
+        credentials = self.get_credential(conn=conn)
+
+        return self.sdk_client(
+            credentials=credentials,
+            subscription_id=subscription_id,
+        )
+
+    def get_credential(
+        self, *, conn: Connection | None = None
+    ) -> (
+        ServicePrincipalCredentials
+        | AzureIdentityCredentialAdapter
+        | ClientSecretCredential
+        | DefaultAzureCredential
+    ):
+        """
+        Get Azure credential object for the connection.
+
+        Azure Identity based credential object (``ClientSecretCredential``, ``DefaultAzureCredential``) can be used to get OAuth token using ``get_token`` method.
+        Older Credential objects (``ServicePrincipalCredentials``, ``AzureIdentityCredentialAdapter``) are supported for backward compatibility.
+
+        :return: The Azure credential object
+        """
+        if not conn:
+            conn = self.get_connection(self.conn_id)
+        tenant = conn.extra_dejson.get("tenantId")
+        credential: (
+            ServicePrincipalCredentials
+            | AzureIdentityCredentialAdapter
+            | ClientSecretCredential
+            | DefaultAzureCredential
+        )
         if all([conn.login, conn.password, tenant]):
-            self.log.info("Getting connection using specific credentials and subscription_id.")
-            credentials = ServicePrincipalCredentials(
-                client_id=conn.login, secret=conn.password, tenant=tenant
-            )
+            credential = self._get_client_secret_credential(conn)
         else:
-            self.log.info("Using DefaultAzureCredential as credential")
-            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(
+            credential = self._get_default_azure_credential(conn)
+        return credential
+
+    def _get_client_secret_credential(
+        self, conn: Connection
+    ) -> ServicePrincipalCredentials | ClientSecretCredential:
+        self.log.info("Getting credentials using specific credentials and subscription_id.")
+        extra_dejson = conn.extra_dejson
+        tenant = extra_dejson.get("tenantId")
+        use_azure_identity_object = extra_dejson.get("use_azure_identity_object", False)
+        if use_azure_identity_object:
+            return ClientSecretCredential(
+                client_id=conn.login,  # type: ignore[arg-type]
+                client_secret=conn.password,  # type: ignore[arg-type]
+                tenant_id=tenant,  # type: ignore[arg-type]
+            )
+        return ServicePrincipalCredentials(client_id=conn.login, secret=conn.password, tenant=tenant)
+
+    def _get_default_azure_credential(
+        self, conn: Connection
+    ) -> DefaultAzureCredential | AzureIdentityCredentialAdapter:
+        self.log.info("Using DefaultAzureCredential as credential")
+        extra_dejson = conn.extra_dejson
+        managed_identity_client_id = extra_dejson.get("managed_identity_client_id")
+        workload_identity_tenant_id = extra_dejson.get("workload_identity_tenant_id")
+        use_azure_identity_object = extra_dejson.get("use_azure_identity_object", False)
+        if use_azure_identity_object:
+            return get_sync_default_azure_credential(
                 managed_identity_client_id=managed_identity_client_id,
                 workload_identity_tenant_id=workload_identity_tenant_id,
             )
-
-        return self.sdk_client(
-            credentials=credentials,
-            subscription_id=subscription_id,
+        return AzureIdentityCredentialAdapter(
+            managed_identity_client_id=managed_identity_client_id,
+            workload_identity_tenant_id=workload_identity_tenant_id,
         )
+
+    def get_token(self, *scopes, **kwargs) -> AccessToken:
+        """
+        Request an access token for `scopes`.
+
+        To use this method, set `use_azure_identity_object: True` in the connection extra field.
+        ServicePrincipalCredentials and AzureIdentityCredentialAdapter don't support `get_token` method.
+        """
+        credential = self.get_credential()
+        if isinstance(credential, ServicePrincipalCredentials) or isinstance(
+            credential, AzureIdentityCredentialAdapter
+        ):
+            raise AttributeError(
+                "ServicePrincipalCredentials and AzureIdentityCredentialAdapter don't support get_token method. "
+                "Please set `use_azure_identity_object: True` in the connection extra field to use credential that support get_token method."
+            )
+        return credential.get_token(*scopes, **kwargs)

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

@@ -25,12 +25,12 @@ from typing import TYPE_CHECKING, Any
 from azure.batch import BatchServiceClient, batch_auth, models as batch_models
 
 from airflow.exceptions import AirflowException
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     AzureIdentityCredentialAdapter,
     add_managed_identity_connection_widgets,
     get_field,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 from airflow.utils import timezone
 
 if TYPE_CHECKING:

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

@@ -25,12 +25,12 @@ from typing import Any, cast
 from azure.mgmt.containerinstance.models import ImageRegistryCredential
 from azure.mgmt.containerregistry import ContainerRegistryManagementClient
 
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     add_managed_identity_connection_widgets,
     get_field,
     get_sync_default_azure_credential,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 
 
 class AzureContainerRegistryHook(BaseHook):

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

@@ -21,12 +21,12 @@ from typing import Any, cast
 from azure.mgmt.containerinstance.models import AzureFileVolume, Volume
 from azure.mgmt.storage import StorageManagementClient
 
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     add_managed_identity_connection_widgets,
     get_field,
     get_sync_default_azure_credential,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 
 
 class AzureContainerVolumeHook(BaseHook):

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

@@ -36,12 +36,12 @@ from azure.cosmos.exceptions import CosmosHttpResponseError
 from azure.mgmt.cosmosdb import CosmosDBManagementClient
 
 from airflow.exceptions import AirflowBadRequest, AirflowException
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     add_managed_identity_connection_widgets,
     get_field,
     get_sync_default_azure_credential,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 
 if TYPE_CHECKING:
     PartitionKeyType = str | list[str]

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

@@ -49,12 +49,12 @@ from azure.mgmt.datafactory import DataFactoryManagementClient
 from azure.mgmt.datafactory.aio import DataFactoryManagementClient as AsyncDataFactoryManagementClient
 
 from airflow.exceptions import AirflowException
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     add_managed_identity_connection_widgets,
     get_async_default_azure_credential,
     get_sync_default_azure_credential,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 
 if TYPE_CHECKING:
     from azure.core.polling import LROPoller
@@ -1214,7 +1214,4 @@ class AzureDataFactoryAsyncHook(AzureDataFactoryHook):
         :param config: Extra parameters for the ADF client.
         """
         client = await self.get_async_conn()
-        try:
-            await client.pipeline_runs.cancel(resource_group_name, factory_name, run_id)
-        except Exception as e:
-            raise AirflowException(e)
+        await client.pipeline_runs.cancel(resource_group_name, factory_name, run_id)

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

@@ -33,13 +33,13 @@ from azure.storage.filedatalake import (
 )
 
 from airflow.exceptions import AirflowException
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     AzureIdentityCredentialAdapter,
     add_managed_identity_connection_widgets,
     get_field,
     get_sync_default_azure_credential,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 
 Credentials = ClientSecretCredential | AzureIdentityCredentialAdapter | DefaultAzureCredential
 

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

@@ -21,11 +21,11 @@ from typing import IO, Any
 
 from azure.storage.fileshare import FileProperties, ShareDirectoryClient, ShareFileClient, ShareServiceClient
 
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     add_managed_identity_connection_widgets,
     get_sync_default_azure_credential,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 
 
 class AzureFileShareHook(BaseHook):

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

@@ -17,6 +17,7 @@
 # under the License.
 from __future__ import annotations
 
+import asyncio
 import json
 import warnings
 from ast import literal_eval
@@ -28,7 +29,6 @@ from typing import TYPE_CHECKING, Any, cast
 from urllib.parse import quote, urljoin, urlparse
 
 import httpx
-from asgiref.sync import sync_to_async
 from azure.identity import CertificateCredential, ClientSecretCredential
 from httpx import AsyncHTTPTransport, Response, Timeout
 from kiota_abstractions.api_error import APIError
@@ -53,7 +53,7 @@ from airflow.exceptions import (
     AirflowNotFoundException,
     AirflowProviderDeprecationWarning,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
+from airflow.providers.common.compat.sdk import BaseHook
 
 if TYPE_CHECKING:
     from azure.identity._internal.client_credential_base import ClientCredentialBase
@@ -61,7 +61,7 @@ if TYPE_CHECKING:
     from kiota_abstractions.response_handler import NativeResponseType
     from kiota_abstractions.serialization import ParsableFactory
 
-    from airflow.models import Connection
+    from airflow.providers.common.compat.sdk import Connection
 
 
 class DefaultResponseHandler(ResponseHandler):
@@ -152,6 +152,7 @@ class KiotaRequestAdapterHook(BaseHook):
 
         return {
             "tenant_id": StringField(lazy_gettext("Tenant ID"), widget=BS3TextFieldWidget()),
+            "drive_id": StringField(lazy_gettext("Drive ID"), widget=BS3TextFieldWidget()),
             "api_version": StringField(
                 lazy_gettext("API Version"), widget=BS3TextFieldWidget(), default=APIVersion.v1.value
             ),
@@ -250,7 +251,9 @@ class KiotaRequestAdapterHook(BaseHook):
     def _build_request_adapter(self, connection) -> tuple[str, RequestAdapter]:
         client_id = connection.login
         client_secret = connection.password
-        config = connection.extra_dejson if connection.extra else {}
+        # TODO (#54350): do not use connection.extra_dejson until it's fixed in Airflow otherwise expect:
+        #       RuntimeError: You cannot use AsyncToSync in the same thread as an async event loop.
+        config = json.loads(connection.extra) if connection.extra else {}
         api_version = self.get_api_version(config)
         host = self.get_host(connection)  # type: ignore[arg-type]
         base_url = self.get_base_url(host, api_version, config)
@@ -342,6 +345,15 @@ class KiotaRequestAdapterHook(BaseHook):
         self.api_version = api_version
         return request_adapter
 
+    @classmethod
+    async def get_async_connection(cls, conn_id: str) -> Connection:
+        if hasattr(BaseHook, "aget_connection"):
+            return await BaseHook.aget_connection(conn_id=conn_id)
+
+        from asgiref.sync import sync_to_async
+
+        return await sync_to_async(BaseHook.get_connection)(conn_id=conn_id)
+
     async def get_async_conn(self) -> RequestAdapter:
         """Initiate a new RequestAdapter connection asynchronously."""
         if not self.conn_id:
@@ -350,7 +362,7 @@ class KiotaRequestAdapterHook(BaseHook):
         api_version, request_adapter = self.cached_request_adapters.get(self.conn_id, (None, None))
 
         if not request_adapter:
-            connection = await sync_to_async(self.get_connection)(conn_id=self.conn_id)
+            connection = await self.get_async_connection(conn_id=self.conn_id)
             api_version, request_adapter = self._build_request_adapter(connection)
         self.api_version = api_version
         return request_adapter
@@ -417,7 +429,7 @@ class KiotaRequestAdapterHook(BaseHook):
     def test_connection(self):
         """Test HTTP Connection."""
         try:
-            self.run()
+            asyncio.run(self.run())
             return True, "Connection successfully tested"
         except Exception as e:
             return False, str(e)

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

@@ -25,12 +25,12 @@ from azure.synapse.artifacts import ArtifactsClient
 from azure.synapse.spark import SparkClient
 
 from airflow.exceptions import AirflowException, AirflowTaskTimeout
+from airflow.providers.common.compat.sdk import BaseHook
 from airflow.providers.microsoft.azure.utils import (
     add_managed_identity_connection_widgets,
     get_field,
     get_sync_default_azure_credential,
 )
-from airflow.providers.microsoft.azure.version_compat import BaseHook
 
 if TYPE_CHECKING:
     from azure.synapse.artifacts.models import CreateRunResponse, PipelineRun

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

@@ -45,13 +45,13 @@ from azure.storage.blob.aio import (
 )
 
 from airflow.exceptions import AirflowException
+from airflow.providers.common.compat.sdk 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 airflow.providers.microsoft.azure.version_compat import BaseHook
 
 if TYPE_CHECKING:
     from azure.core.credentials import TokenCredential

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

@@ -188,9 +188,15 @@ class WasbTaskHandler(FileTaskHandler, LoggingMixin):
         base_log_folder: str,
         wasb_log_folder: str,
         wasb_container: str,
+        max_bytes: int = 0,
+        backup_count: int = 0,
+        delay: bool = False,
         **kwargs,
     ) -> None:
-        super().__init__(base_log_folder)
+        # support log file size handling of FileTaskHandler
+        super().__init__(
+            base_log_folder=base_log_folder, max_bytes=max_bytes, backup_count=backup_count, delay=delay
+        )
         self.handler: logging.FileHandler | None = None
         self.log_relative_path = ""
         self.closed = False

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

@@ -19,8 +19,8 @@ from __future__ import annotations
 from collections.abc import Iterable, Sequence
 from typing import IO, TYPE_CHECKING, Any, AnyStr
 
+from airflow.providers.common.compat.sdk import BaseOperator
 from airflow.providers.microsoft.azure.hooks.data_lake import AzureDataLakeHook, AzureDataLakeStorageV2Hook
-from airflow.providers.microsoft.azure.version_compat import BaseOperator
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

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

@@ -24,8 +24,8 @@ from functools import cached_property
 from typing import TYPE_CHECKING
 
 from airflow.configuration import conf
+from airflow.providers.common.compat.sdk import BaseOperator
 from airflow.providers.microsoft.azure.hooks.adx import AzureDataExplorerHook
-from airflow.providers.microsoft.azure.version_compat import BaseOperator
 
 if TYPE_CHECKING:
     from azure.kusto.data._models import KustoResultTable

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

@@ -20,8 +20,8 @@ from collections.abc import Callable, Sequence
 from typing import TYPE_CHECKING, Any
 from uuid import UUID
 
+from airflow.providers.common.compat.sdk import BaseOperator
 from airflow.providers.microsoft.azure.hooks.asb import AdminClientHook, MessageHook
-from airflow.providers.microsoft.azure.version_compat import BaseOperator
 
 if TYPE_CHECKING:
     import datetime

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

@@ -24,8 +24,8 @@ from typing import TYPE_CHECKING, Any
 from azure.batch import models as batch_models
 
 from airflow.exceptions import AirflowException
+from airflow.providers.common.compat.sdk import BaseOperator
 from airflow.providers.microsoft.azure.hooks.batch import AzureBatchHook
-from airflow.providers.microsoft.azure.version_compat import BaseOperator
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context

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

@@ -40,10 +40,10 @@ from azure.mgmt.containerinstance.models import (
 from msrestazure.azure_exceptions import CloudError
 
 from airflow.exceptions import AirflowException, AirflowTaskTimeout
+from airflow.providers.common.compat.sdk 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 airflow.providers.microsoft.azure.version_compat import BaseOperator
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context