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

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

@@ -18,6 +18,7 @@
 from __future__ import annotations
 
 import json
+import warnings
 from ast import literal_eval
 from contextlib import suppress
 from http import HTTPStatus
@@ -27,6 +28,7 @@ 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
@@ -49,6 +51,7 @@ from airflow.exceptions import (
     AirflowConfigException,
     AirflowException,
     AirflowNotFoundException,
+    AirflowProviderDeprecationWarning,
 )
 from airflow.providers.microsoft.azure.version_compat import BaseHook
 
@@ -114,7 +117,7 @@ class KiotaRequestAdapterHook(BaseHook):
 
     DEFAULT_HEADERS = {"Accept": "application/json;q=1"}
     DEFAULT_SCOPE = "https://graph.microsoft.com/.default"
-    cached_request_adapters: dict[str, tuple[APIVersion, RequestAdapter]] = {}
+    cached_request_adapters: dict[str, tuple[str, RequestAdapter]] = {}
     conn_type: str = "msgraph"
     conn_name_attr: str = "conn_id"
     default_conn_name: str = "msgraph_default"
@@ -138,7 +141,7 @@ class KiotaRequestAdapterHook(BaseHook):
             self.scopes = [scopes]
         else:
             self.scopes = scopes or [self.DEFAULT_SCOPE]
-        self._api_version = self.resolve_api_version_from_value(api_version)
+        self.api_version = self.resolve_api_version_from_value(api_version)
 
     @classmethod
     def get_connection_form_widgets(cls) -> dict[str, Any]:
@@ -186,11 +189,6 @@ class KiotaRequestAdapterHook(BaseHook):
             },
         }
 
-    @property
-    def api_version(self) -> str | None:
-        self.get_conn()  # Make sure config has been loaded through get_conn to have correct api version!
-        return self._api_version
-
     @staticmethod
     def resolve_api_version_from_value(
         api_version: APIVersion | str, default: str | None = None
@@ -200,7 +198,7 @@ class KiotaRequestAdapterHook(BaseHook):
         return api_version or default
 
     def get_api_version(self, config: dict) -> str:
-        return self._api_version or self.resolve_api_version_from_value(
+        return self.api_version or self.resolve_api_version_from_value(
             config.get("api_version"), APIVersion.v1.value
         )  # type: ignore
 
@@ -209,6 +207,13 @@ class KiotaRequestAdapterHook(BaseHook):
             return f"{connection.schema}://{connection.host}"
         return self.host
 
+    def get_base_url(self, host: str, api_version: str, config: dict) -> str:
+        base_url = config.get("base_url", urljoin(host, api_version)).strip()
+
+        if not base_url.endswith("/"):
+            return f"{base_url}/"
+        return base_url
+
     @staticmethod
     def format_no_proxy_url(url: str) -> str:
         if "://" not in url:
@@ -242,83 +247,112 @@ class KiotaRequestAdapterHook(BaseHook):
                         return None
         return proxies
 
+    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 {}
+        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)
+        authority = config.get("authority")
+        proxies = self.get_proxies(config)
+        httpx_proxies = self.to_httpx_proxies(proxies=proxies)
+        scopes = config.get("scopes", self.scopes)
+        if isinstance(scopes, str):
+            scopes = scopes.split(",")
+        verify = config.get("verify", True)
+        trust_env = config.get("trust_env", False)
+        allowed_hosts = (config.get("allowed_hosts", authority) or "").split(",")
+
+        self.log.info(
+            "Creating Microsoft Graph SDK client %s for conn_id: %s",
+            api_version,
+            self.conn_id,
+        )
+        self.log.info("Host: %s", host)
+        self.log.info("Base URL: %s", base_url)
+        self.log.info("Client id: %s", client_id)
+        self.log.info("Client secret: %s", client_secret)
+        self.log.info("API version: %s", api_version)
+        self.log.info("Scope: %s", scopes)
+        self.log.info("Verify: %s", verify)
+        self.log.info("Timeout: %s", self.timeout)
+        self.log.info("Trust env: %s", trust_env)
+        self.log.info("Authority: %s", authority)
+        self.log.info("Allowed hosts: %s", allowed_hosts)
+        self.log.info("Proxies: %s", proxies)
+        self.log.info("HTTPX Proxies: %s", httpx_proxies)
+        credentials = self.get_credentials(
+            login=connection.login,
+            password=connection.password,
+            config=config,
+            authority=authority,
+            verify=verify,
+            proxies=proxies,
+        )
+        http_client = GraphClientFactory.create_with_default_middleware(
+            api_version=api_version,
+            client=httpx.AsyncClient(
+                mounts=httpx_proxies,
+                timeout=Timeout(timeout=self.timeout),
+                verify=verify,
+                trust_env=trust_env,
+                base_url=base_url,
+            ),
+            host=host,
+        )
+        auth_provider = AzureIdentityAuthenticationProvider(
+            credentials=credentials,
+            scopes=scopes,
+            allowed_hosts=allowed_hosts,
+        )
+        parse_node_factory = ParseNodeFactoryRegistry()
+        parse_node_factory.CONTENT_TYPE_ASSOCIATED_FACTORIES["text/plain"] = TextParseNodeFactory()
+        parse_node_factory.CONTENT_TYPE_ASSOCIATED_FACTORIES["application/json"] = JsonParseNodeFactory()
+        request_adapter = HttpxRequestAdapter(
+            authentication_provider=auth_provider,
+            parse_node_factory=parse_node_factory,
+            http_client=http_client,
+            base_url=base_url,
+        )
+        self.cached_request_adapters[self.conn_id] = (api_version, request_adapter)
+        return api_version, request_adapter
+
     def get_conn(self) -> RequestAdapter:
+        """
+        Initiate a new RequestAdapter connection.
+
+        .. warning::
+           This method is deprecated.
+        """
         if not self.conn_id:
             raise AirflowException("Failed to create the KiotaRequestAdapterHook. No conn_id provided!")
 
+        warnings.warn(
+            "get_conn is deprecated, please use the async get_async_conn method!",
+            category=AirflowProviderDeprecationWarning,
+            stacklevel=2,
+        )
+
         api_version, request_adapter = self.cached_request_adapters.get(self.conn_id, (None, None))
 
         if not request_adapter:
             connection = self.get_connection(conn_id=self.conn_id)
-            client_id = connection.login
-            client_secret = connection.password
-            config = connection.extra_dejson if connection.extra else {}
-            api_version = self.get_api_version(config)
-            host = self.get_host(connection)  # type: ignore[arg-type]
-            base_url = config.get("base_url", urljoin(host, api_version))
-            authority = config.get("authority")
-            proxies = self.get_proxies(config)
-            httpx_proxies = self.to_httpx_proxies(proxies=proxies)
-            scopes = config.get("scopes", self.scopes)
-            if isinstance(scopes, str):
-                scopes = scopes.split(",")
-            verify = config.get("verify", True)
-            trust_env = config.get("trust_env", False)
-            allowed_hosts = (config.get("allowed_hosts", authority) or "").split(",")
-
-            self.log.info(
-                "Creating Microsoft Graph SDK client %s for conn_id: %s",
-                api_version,
-                self.conn_id,
-            )
-            self.log.info("Host: %s", host)
-            self.log.info("Base URL: %s", base_url)
-            self.log.info("Client id: %s", client_id)
-            self.log.info("Client secret: %s", client_secret)
-            self.log.info("API version: %s", api_version)
-            self.log.info("Scope: %s", scopes)
-            self.log.info("Verify: %s", verify)
-            self.log.info("Timeout: %s", self.timeout)
-            self.log.info("Trust env: %s", trust_env)
-            self.log.info("Authority: %s", authority)
-            self.log.info("Allowed hosts: %s", allowed_hosts)
-            self.log.info("Proxies: %s", proxies)
-            self.log.info("HTTPX Proxies: %s", httpx_proxies)
-            credentials = self.get_credentials(
-                login=connection.login,
-                password=connection.password,
-                config=config,
-                authority=authority,
-                verify=verify,
-                proxies=proxies,
-            )
-            http_client = GraphClientFactory.create_with_default_middleware(
-                api_version=api_version,
-                client=httpx.AsyncClient(
-                    mounts=httpx_proxies,
-                    timeout=Timeout(timeout=self.timeout),
-                    verify=verify,
-                    trust_env=trust_env,
-                    base_url=base_url,
-                ),
-                host=host,
-            )
-            auth_provider = AzureIdentityAuthenticationProvider(
-                credentials=credentials,
-                scopes=scopes,
-                allowed_hosts=allowed_hosts,
-            )
-            parse_node_factory = ParseNodeFactoryRegistry()
-            parse_node_factory.CONTENT_TYPE_ASSOCIATED_FACTORIES["text/plain"] = TextParseNodeFactory()
-            parse_node_factory.CONTENT_TYPE_ASSOCIATED_FACTORIES["application/json"] = JsonParseNodeFactory()
-            request_adapter = HttpxRequestAdapter(
-                authentication_provider=auth_provider,
-                parse_node_factory=parse_node_factory,
-                http_client=http_client,
-                base_url=base_url,
-            )
-            self.cached_request_adapters[self.conn_id] = (api_version, request_adapter)
-        self._api_version = api_version
+            api_version, request_adapter = self._build_request_adapter(connection)
+        self.api_version = api_version
+        return request_adapter
+
+    async def get_async_conn(self) -> RequestAdapter:
+        """Initiate a new RequestAdapter connection asynchronously."""
+        if not self.conn_id:
+            raise AirflowException("Failed to create the KiotaRequestAdapterHook. No conn_id provided!")
+
+        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)
+            api_version, request_adapter = self._build_request_adapter(connection)
+        self.api_version = api_version
         return request_adapter
 
     def get_proxies(self, config: dict) -> dict:
@@ -418,13 +452,15 @@ class KiotaRequestAdapterHook(BaseHook):
         return response
 
     async def send_request(self, request_info: RequestInformation, response_type: str | None = None):
+        conn = await self.get_async_conn()
+
         if response_type:
-            return await self.get_conn().send_primitive_async(
+            return await conn.send_primitive_async(
                 request_info=request_info,
                 response_type=response_type,
                 error_map=self.error_mapping(),
             )
-        return await self.get_conn().send_no_response_content_async(
+        return await conn.send_no_response_content_async(
             request_info=request_info,
             error_map=self.error_mapping(),
         )
@@ -437,7 +473,7 @@ class KiotaRequestAdapterHook(BaseHook):
         method: str = "GET",
         query_parameters: dict[str, Any] | None = None,
         headers: dict[str, str] | None = None,
-        data: dict[str, Any] | str | BytesIO | None = None,
+        data: dict[str, Any] | str | bytes | BytesIO | None = None,
     ) -> RequestInformation:
         request_information = RequestInformation()
         request_information.path_parameters = path_parameters or {}
@@ -468,7 +504,7 @@ class KiotaRequestAdapterHook(BaseHook):
                 header_name=RequestInformation.CONTENT_TYPE_HEADER, header_value="application/json"
             )
             request_information.content = json.dumps(data).encode("utf-8")
-        print("Request Information:", request_information.url)
+        self.log.debug("Request Information: %s", request_information.url)
         return request_information
 
     @staticmethod

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

@@ -28,7 +28,6 @@ from __future__ import annotations
 
 import logging
 import os
-from functools import cached_property
 from typing import TYPE_CHECKING, Any, cast
 
 from asgiref.sync import sync_to_async
@@ -137,6 +136,7 @@ class WasbHook(BaseHook):
         super().__init__()
         self.conn_id = wasb_conn_id
         self.public_read = public_read
+        self._blob_service_client: AsyncBlobServiceClient | BlobServiceClient | None = None
 
         logger = logging.getLogger("azure.core.pipeline.policies.http_logging_policy")
         try:
@@ -154,10 +154,17 @@ class WasbHook(BaseHook):
             return extra_dict[field_name] or None
         return extra_dict.get(f"{prefix}{field_name}") or None
 
-    @cached_property
-    def blob_service_client(self) -> BlobServiceClient:
+    @property
+    def blob_service_client(self) -> AsyncBlobServiceClient | BlobServiceClient:
         """Return the BlobServiceClient object (cached)."""
-        return self.get_conn()
+        if self._blob_service_client is None:
+            self._blob_service_client = self.get_conn()
+        return self._blob_service_client
+
+    @blob_service_client.setter
+    def blob_service_client(self, client: AsyncBlobServiceClient) -> None:
+        """Set the cached BlobServiceClient object."""
+        self._blob_service_client = client
 
     def get_conn(self) -> BlobServiceClient:
         """Return the BlobServiceClient object."""
@@ -220,13 +227,12 @@ class WasbHook(BaseHook):
             **extra,
         )
 
-    # TODO: rework the interface as it might also return AsyncContainerClient
-    def _get_container_client(self, container_name: str) -> ContainerClient:
+    def _get_container_client(self, container_name: str) -> AsyncContainerClient | ContainerClient:
         """
         Instantiate a container client.
 
         :param container_name: The name of the container
-        :return: ContainerClient
+        :return: AsyncContainerClient | ContainerClient
         """
         return self.blob_service_client.get_container_client(container_name)
 
@@ -266,6 +272,12 @@ class WasbHook(BaseHook):
         blobs = self.get_blobs_list(container_name=container_name, prefix=prefix, **kwargs)
         return bool(blobs)
 
+    def check_for_variable_type(self, variable_name: str, container: Any, expected_type: type[Any]) -> None:
+        if not isinstance(container, expected_type):
+            raise TypeError(
+                f"{variable_name} for {self.__class__.__name__} must be {expected_type.__name__}, got {type(container).__name__}"
+            )
+
     def get_blobs_list(
         self,
         container_name: str,
@@ -286,6 +298,9 @@ class WasbHook(BaseHook):
         :param delimiter: filters objects based on the delimiter (for e.g '.csv')
         """
         container = self._get_container_client(container_name)
+        self.check_for_variable_type("container", container, ContainerClient)
+        container = cast("ContainerClient", container)
+
         blob_list = []
         blobs = container.walk_blobs(name_starts_with=prefix, include=include, delimiter=delimiter, **kwargs)
         for blob in blobs:
@@ -312,8 +327,12 @@ class WasbHook(BaseHook):
         :param delimiter: filters objects based on the delimiter (for e.g '.csv')
         """
         container = self._get_container_client(container_name)
+        self.check_for_variable_type("container", container, ContainerClient)
+        container = cast("ContainerClient", container)
+
         blob_list = []
         blobs = container.list_blobs(name_starts_with=prefix, include=include, **kwargs)
+
         for blob in blobs:
             if blob.name.endswith(endswith):
                 blob_list.append(blob.name)
@@ -591,12 +610,16 @@ class WasbAsyncHook(WasbHook):
         """Initialize the hook instance."""
         self.conn_id = wasb_conn_id
         self.public_read = public_read
-        self.blob_service_client: AsyncBlobServiceClient = None  # type: ignore
+        self._blob_service_client: AsyncBlobServiceClient | BlobServiceClient | None = None
 
     async def get_async_conn(self) -> AsyncBlobServiceClient:
         """Return the Async BlobServiceClient object."""
-        if self.blob_service_client is not None:
-            return self.blob_service_client
+        if self._blob_service_client is not None:
+            self.check_for_variable_type(
+                "self._blob_service_client", self._blob_service_client, AsyncBlobServiceClient
+            )
+            self._blob_service_client = cast("AsyncBlobServiceClient", self._blob_service_client)
+            return self._blob_service_client
 
         conn = await sync_to_async(self.get_connection)(self.conn_id)
         extra = conn.extra_dejson or {}
@@ -605,7 +628,7 @@ class WasbAsyncHook(WasbHook):
         connection_string = self._get_field(extra, "connection_string")
         if connection_string:
             # connection_string auth takes priority
-            self.blob_service_client = AsyncBlobServiceClient.from_connection_string(
+            self.blob_service_client: AsyncBlobServiceClient = AsyncBlobServiceClient.from_connection_string(
                 connection_string, **extra
             )
             return self.blob_service_client
@@ -703,18 +726,6 @@ class WasbAsyncHook(WasbHook):
             return False
         return True
 
-    # TODO: rework the interface as in parent Hook it returns ContainerClient
-    def _get_container_client(self, container_name: str) -> AsyncContainerClient:  # type: ignore[override]
-        """
-        Instantiate a container client.
-
-        :param container_name: the name of the container
-        """
-        if self.blob_service_client is None:
-            raise AirflowException("BlobServiceClient is not initialized")
-
-        return self.blob_service_client.get_container_client(container_name)
-
     async def get_blobs_list_async(
         self,
         container_name: str,
@@ -735,6 +746,9 @@ class WasbAsyncHook(WasbHook):
         :param delimiter: filters objects based on the delimiter (for e.g '.csv')
         """
         container = self._get_container_client(container_name)
+        container = cast("AsyncContainerClient", container)
+        self.check_for_variable_type("container", container, AsyncContainerClient)
+
         blob_list: list[BlobProperties | BlobPrefix] = []
         blobs = container.walk_blobs(name_starts_with=prefix, include=include, delimiter=delimiter, **kwargs)
         async for blob in blobs:

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

@@ -247,7 +247,7 @@ class MSGraphAsyncOperator(BaseOperator):
     @classmethod
     def append_result(
         cls,
-        results: list[Any],
+        results: Any,
         result: Any,
         append_result_as_list_if_absent: bool = False,
     ) -> list[Any]:

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

@@ -88,7 +88,7 @@ class AzureSynapseRunSparkBatchOperator(BaseOperator):
         **kwargs,
     ) -> None:
         super().__init__(**kwargs)
-        self.job_id = None
+        self.job_id: Any = None
         self.azure_synapse_conn_id = azure_synapse_conn_id
         self.wait_for_termination = wait_for_termination
         self.spark_pool = spark_pool

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

@@ -100,15 +100,18 @@ class AzureKeyVaultBackend(BaseSecretsBackend, LoggingMixin):
     ) -> None:
         super().__init__()
         self.vault_url = vault_url
-        if connections_prefix is not None:
+        self.connections_prefix: str | None
+        if connections_prefix:
             self.connections_prefix = connections_prefix.rstrip(sep)
         else:
             self.connections_prefix = connections_prefix
-        if variables_prefix is not None:
+        self.variables_prefix: str | None
+        if variables_prefix:
             self.variables_prefix = variables_prefix.rstrip(sep)
         else:
             self.variables_prefix = variables_prefix
-        if config_prefix is not None:
+        self.config_prefix: str | None
+        if config_prefix:
             self.config_prefix = config_prefix.rstrip(sep)
         else:
             self.config_prefix = config_prefix

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

@@ -23,6 +23,7 @@ from base64 import b64encode
 from collections.abc import AsyncIterator, Sequence
 from contextlib import suppress
 from datetime import datetime
+from functools import cached_property
 from json import JSONDecodeError
 from typing import (
     TYPE_CHECKING,
@@ -125,13 +126,11 @@ class MSGraphTrigger(BaseTrigger):
         serializer: type[ResponseSerializer] = ResponseSerializer,
     ):
         super().__init__()
-        self.hook = KiotaRequestAdapterHook(
-            conn_id=conn_id,
-            timeout=timeout,
-            proxies=proxies,
-            scopes=scopes,
-            api_version=api_version,
-        )
+        self.conn_id = conn_id
+        self.timeout = timeout
+        self.proxies = proxies
+        self.scopes = scopes
+        self.api_version = api_version
         self.url = url
         self.response_type = response_type
         self.path_parameters = path_parameters
@@ -158,7 +157,7 @@ class MSGraphTrigger(BaseTrigger):
                 "conn_id": self.conn_id,
                 "timeout": self.timeout,
                 "proxies": self.proxies,
-                "scopes": self.hook.scopes,
+                "scopes": self.scopes,
                 "api_version": self.api_version,
                 "serializer": f"{self.serializer.__class__.__module__}.{self.serializer.__class__.__name__}",
                 "url": self.url,
@@ -173,23 +172,23 @@ class MSGraphTrigger(BaseTrigger):
         )
 
     def get_conn(self) -> RequestAdapter:
-        return self.hook.get_conn()
-
-    @property
-    def conn_id(self) -> str:
-        return self.hook.conn_id
+        """
+        Initiate a new RequestAdapter connection.
 
-    @property
-    def timeout(self) -> float | None:
-        return self.hook.timeout
-
-    @property
-    def proxies(self) -> dict | None:
-        return self.hook.proxies
+        .. warning::
+           This method is deprecated.
+        """
+        return self.hook.get_conn()
 
-    @property
-    def api_version(self) -> APIVersion | str:
-        return self.hook.api_version
+    @cached_property
+    def hook(self) -> KiotaRequestAdapterHook:
+        return KiotaRequestAdapterHook(
+            conn_id=self.conn_id,
+            timeout=self.timeout,
+            proxies=self.proxies,
+            scopes=self.scopes,
+            api_version=self.api_version,
+        )
 
     async def run(self) -> AsyncIterator[TriggerEvent]:
         """Make a series of asynchronous HTTP calls via a KiotaRequestAdapterHook."""

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

@@ -20,6 +20,7 @@ from __future__ import annotations
 import asyncio
 import time
 from collections.abc import AsyncIterator
+from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
 import tenacity
@@ -32,10 +33,56 @@ from airflow.providers.microsoft.azure.hooks.powerbi import (
 from airflow.triggers.base import BaseTrigger, TriggerEvent
 
 if TYPE_CHECKING:
+    from kiota_abstractions.request_adapter import RequestAdapter
     from msgraph_core import APIVersion
 
 
-class PowerBITrigger(BaseTrigger):
+class BasePowerBITrigger(BaseTrigger):
+    """
+    Base class for all PowerBI related triggers.
+
+    :param conn_id: The connection Id to connect to PowerBI.
+    :param timeout: The HTTP timeout being used by the `KiotaRequestAdapter` (default is None).
+        When no timeout is specified or set to None then there is no HTTP timeout on each request.
+    :param proxies: A dict defining the HTTP proxies to be used (default is None).
+    :param api_version: The API version of the Microsoft Graph API to be used (default is v1).
+        You can pass an enum named APIVersion which has 2 possible members v1 and beta,
+        or you can pass a string as `v1.0` or `beta`.
+    """
+
+    def __init__(
+        self,
+        conn_id: str,
+        timeout: float = 60 * 60 * 24 * 7,
+        proxies: dict | None = None,
+        api_version: APIVersion | str | None = None,
+    ):
+        super().__init__()
+        self.conn_id = conn_id
+        self.timeout = timeout
+        self.proxies = proxies
+        self.api_version = api_version
+
+    def get_conn(self) -> RequestAdapter:
+        """
+        Initiate a new RequestAdapter connection.
+
+        .. warning::
+           This method is deprecated.
+        """
+        return self.hook.get_conn()
+
+    @cached_property
+    def hook(self) -> PowerBIHook:
+        return PowerBIHook(
+            conn_id=self.conn_id,
+            timeout=self.timeout,
+            proxies=self.proxies,
+            api_version=self.api_version,
+        )
+
+
+class PowerBITrigger(BasePowerBITrigger):
     """
     Triggers when Power BI dataset refresh is completed.
 
@@ -69,11 +116,9 @@ class PowerBITrigger(BaseTrigger):
         wait_for_termination: bool = True,
         request_body: dict[str, Any] | None = None,
     ):
-        super().__init__()
-        self.hook = PowerBIHook(conn_id=conn_id, proxies=proxies, api_version=api_version, timeout=timeout)
+        super().__init__(conn_id=conn_id, timeout=timeout, proxies=proxies, api_version=api_version)
         self.dataset_id = dataset_id
         self.dataset_refresh_id = dataset_refresh_id
-        self.timeout = timeout
         self.group_id = group_id
         self.check_interval = check_interval
         self.wait_for_termination = wait_for_termination
@@ -82,7 +127,7 @@ class PowerBITrigger(BaseTrigger):
     def serialize(self):
         """Serialize the trigger instance."""
         return (
-            "airflow.providers.microsoft.azure.triggers.powerbi.PowerBITrigger",
+            f"{self.__class__.__module__}.{self.__class__.__name__}",
             {
                 "conn_id": self.conn_id,
                 "proxies": self.proxies,
@@ -97,18 +142,6 @@ class PowerBITrigger(BaseTrigger):
             },
         )
 
-    @property
-    def conn_id(self) -> str:
-        return self.hook.conn_id
-
-    @property
-    def proxies(self) -> dict | None:
-        return self.hook.proxies
-
-    @property
-    def api_version(self) -> APIVersion | str:
-        return self.hook.api_version
-
     async def run(self) -> AsyncIterator[TriggerEvent]:
         """Make async connection to the PowerBI and polls for the dataset refresh status."""
         if not self.dataset_refresh_id:
@@ -236,7 +269,7 @@ class PowerBITrigger(BaseTrigger):
             )
 
 
-class PowerBIWorkspaceListTrigger(BaseTrigger):
+class PowerBIWorkspaceListTrigger(BasePowerBITrigger):
     """
     Triggers a call to the API to request the available workspace IDs.
 
@@ -257,15 +290,13 @@ class PowerBIWorkspaceListTrigger(BaseTrigger):
         proxies: dict | None = None,
         api_version: APIVersion | str | None = None,
     ):
-        super().__init__()
-        self.hook = PowerBIHook(conn_id=conn_id, proxies=proxies, api_version=api_version, timeout=timeout)
-        self.timeout = timeout
+        super().__init__(conn_id=conn_id, timeout=timeout, proxies=proxies, api_version=api_version)
         self.workspace_ids = workspace_ids
 
     def serialize(self):
         """Serialize the trigger instance."""
         return (
-            "airflow.providers.microsoft.azure.triggers.powerbi.PowerBIWorkspaceListTrigger",
+            f"{self.__class__.__module__}.{self.__class__.__name__}",
             {
                 "conn_id": self.conn_id,
                 "proxies": self.proxies,
@@ -275,18 +306,6 @@ class PowerBIWorkspaceListTrigger(BaseTrigger):
             },
         )
 
-    @property
-    def conn_id(self) -> str:
-        return self.hook.conn_id
-
-    @property
-    def proxies(self) -> dict | None:
-        return self.hook.proxies
-
-    @property
-    def api_version(self) -> APIVersion | str:
-        return self.hook.api_version
-
     async def run(self) -> AsyncIterator[TriggerEvent]:
         """Make async connection to the PowerBI and polls for the list of workspace IDs."""
         # Trigger the API to get the workspace list
@@ -313,7 +332,7 @@ class PowerBIWorkspaceListTrigger(BaseTrigger):
         return
 
 
-class PowerBIDatasetListTrigger(BaseTrigger):
+class PowerBIDatasetListTrigger(BasePowerBITrigger):
     """
     Triggers a call to the API to request the available dataset IDs.
 
@@ -336,16 +355,14 @@ class PowerBIDatasetListTrigger(BaseTrigger):
         proxies: dict | None = None,
         api_version: APIVersion | str | None = None,
     ):
-        super().__init__()
-        self.hook = PowerBIHook(conn_id=conn_id, proxies=proxies, api_version=api_version, timeout=timeout)
-        self.timeout = timeout
+        super().__init__(conn_id=conn_id, timeout=timeout, proxies=proxies, api_version=api_version)
         self.group_id = group_id
         self.dataset_ids = dataset_ids
 
     def serialize(self):
         """Serialize the trigger instance."""
         return (
-            "airflow.providers.microsoft.azure.triggers.powerbi.PowerBIDatasetListTrigger",
+            f"{self.__class__.__module__}.{self.__class__.__name__}",
             {
                 "conn_id": self.conn_id,
                 "proxies": self.proxies,
@@ -356,18 +373,6 @@ class PowerBIDatasetListTrigger(BaseTrigger):
             },
         )
 
-    @property
-    def conn_id(self) -> str:
-        return self.hook.conn_id
-
-    @property
-    def proxies(self) -> dict | None:
-        return self.hook.proxies
-
-    @property
-    def api_version(self) -> APIVersion | str:
-        return self.hook.api_version
-
     async def run(self) -> AsyncIterator[TriggerEvent]:
         """Make async connection to the PowerBI and polls for the list of dataset IDs."""
         # Trigger the API to get the dataset list

{apache_airflow_providers_microsoft_azure-12.6.1.dist-info → apache_airflow_providers_microsoft_azure-12.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: apache-airflow-providers-microsoft-azure
-Version: 12.6.1
+Version: 12.7.0
 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>
@@ -42,22 +42,20 @@ Requires-Dist: azure-kusto-data>=4.1.0,!=4.6.0
 Requires-Dist: azure-mgmt-datafactory>=2.0.0
 Requires-Dist: azure-mgmt-containerregistry>=8.0.0
 Requires-Dist: azure-mgmt-containerinstance>=10.1.0
-Requires-Dist: flask-appbuilder>=4.0.0
 Requires-Dist: msgraph-core>=1.3.3
 Requires-Dist: microsoft-kiota-http>=1.9.4,<2.0.0
 Requires-Dist: microsoft-kiota-serialization-json>=1.9.4
 Requires-Dist: microsoft-kiota-serialization-text>=1.9.4
 Requires-Dist: microsoft-kiota-abstractions>=1.9.4,<2.0.0
 Requires-Dist: microsoft-kiota-authentication-azure>=1.9.4,<2.0.0
-Requires-Dist: msal-extensions>=1.1.0
-Requires-Dist: portalocker>=2.8.1
+Requires-Dist: msal-extensions>=1.3.0
 Requires-Dist: apache-airflow-providers-amazon ; extra == "amazon"
 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.6.1/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.6.1
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.7.0/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.7.0
 Project-URL: Mastodon, https://fosstodon.org/@airflow
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
@@ -92,9 +90,8 @@ Provides-Extra: sftp
 
 Package ``apache-airflow-providers-microsoft-azure``
 
-Release: ``12.6.1``
+Release: ``12.7.0``
 
-Release Date: ``|PypiReleaseDate|``
 
 `Microsoft Azure <https://azure.microsoft.com/>`__
 
@@ -106,12 +103,12 @@ 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.6.1/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.7.0/>`_.
 
 Installation
 ------------
 
-You can install this package on top of an existing Airflow 2 installation (see ``Requirements`` below
+You can install this package on top of an existing Airflow installation (see ``Requirements`` below
 for the minimum Airflow version supported) via
 ``pip install apache-airflow-providers-microsoft-azure``
 
@@ -145,15 +142,13 @@ PIP package                               Version required
 ``azure-mgmt-datafactory``                ``>=2.0.0``
 ``azure-mgmt-containerregistry``          ``>=8.0.0``
 ``azure-mgmt-containerinstance``          ``>=10.1.0``
-``flask-appbuilder``                      ``>=4.0.0``
 ``msgraph-core``                          ``>=1.3.3``
 ``microsoft-kiota-http``                  ``>=1.9.4,<2.0.0``
 ``microsoft-kiota-serialization-json``    ``>=1.9.4``
 ``microsoft-kiota-serialization-text``    ``>=1.9.4``
 ``microsoft-kiota-abstractions``          ``>=1.9.4,<2.0.0``
 ``microsoft-kiota-authentication-azure``  ``>=1.9.4,<2.0.0``
-``msal-extensions``                       ``>=1.1.0``
-``portalocker``                           ``>=2.8.1``
+``msal-extensions``                       ``>=1.3.0``
 ========================================  ===================
 
 Cross provider package dependencies
@@ -179,5 +174,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.6.1/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/12.7.0/changelog.html>`_.
 

{apache_airflow_providers_microsoft_azure-12.6.1.dist-info → apache_airflow_providers_microsoft_azure-12.7.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 airflow/providers/microsoft/azure/LICENSE,sha256=gXPVwptPlW1TJ4HSuG5OMPg-a3h43OGMkZRR1rpwfJA,10850
-airflow/providers/microsoft/azure/__init__.py,sha256=1Q8KMcBHUsa-0RLQjg3JqcmUh5Woo5iVbEl52L7oAO8,1505
+airflow/providers/microsoft/azure/__init__.py,sha256=Fu_0IHkdL_yZvQy3f6BKNl14No6pKkMpt_qTK46I-Bs,1505
 airflow/providers/microsoft/azure/get_provider_info.py,sha256=AeIwe8L-5p1z7IVF9Yc_mmQyAe149r6YVCyRpjfc3u0,18946
 airflow/providers/microsoft/azure/utils.py,sha256=KU9vHQRUhqTbC30GvmuZbL8rPBAziemM1oaT4rZp6K8,9015
 airflow/providers/microsoft/azure/version_compat.py,sha256=pBBF2QxTvpmUMxTE4MfbLfZmydT8FgSxiMMYLAsTTIo,2388
@@ -17,10 +17,10 @@ airflow/providers/microsoft/azure/hooks/cosmos.py,sha256=gw90ZaoaPzTpE27eVM5zbJD
 airflow/providers/microsoft/azure/hooks/data_factory.py,sha256=oo1qcyTDDhJcclq0gE3eVJsFZ_DbuT1PXhTz0NXawds,45028
 airflow/providers/microsoft/azure/hooks/data_lake.py,sha256=1O6eRxe-4Hm5wkWH2zeS4nxc3CYUSFXOu8ySz-UtZJ8,23888
 airflow/providers/microsoft/azure/hooks/fileshare.py,sha256=x8hXYHE6qUjJDTa1qp47DS34eZBeBrwE4cZkk1i0u48,10790
-airflow/providers/microsoft/azure/hooks/msgraph.py,sha256=W7tASxPQnBQciIOYZi2hQo0-QtemMO_YTUCkkfH2pDk,21159
+airflow/providers/microsoft/azure/hooks/msgraph.py,sha256=QdQIBhbgFdu4-W0meOnPiWwKmFUzZRGxR8Kjepn3BV4,22247
 airflow/providers/microsoft/azure/hooks/powerbi.py,sha256=_Z-PWDcZxwSmyP9uXBEEYNRrSIlGPFVUtDrtVk3KKxE,9778
 airflow/providers/microsoft/azure/hooks/synapse.py,sha256=PC7whM9SOuc29hu9mU1DOvi5xmevGLS9Gadql2u0KYM,16077
-airflow/providers/microsoft/azure/hooks/wasb.py,sha256=EJhGrTGvzabZh72g3LjjKscpt9_jX51rJPczyWuFolY,31995
+airflow/providers/microsoft/azure/hooks/wasb.py,sha256=mjYMNWNuPVgW2EGUwmHS1Af3bKx1cho4-cGAz5xgZXQ,32882
 airflow/providers/microsoft/azure/log/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/microsoft/azure/log/wasb_task_handler.py,sha256=s0BCDf7fritIvuwznFJTUEE03GqfecKy-stf1niWr3o,9926
 airflow/providers/microsoft/azure/operators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
@@ -31,12 +31,12 @@ airflow/providers/microsoft/azure/operators/batch.py,sha256=NYjaxYcFwficmJb7l3Vj
 airflow/providers/microsoft/azure/operators/container_instances.py,sha256=8ti1L50pQQcRVL_7noKo6IXoobi8hO0-A4SlgcbIhX0,18672
 airflow/providers/microsoft/azure/operators/cosmos.py,sha256=t7XWU4L5W7tr33J6lC3_mIrFANW_JE1QOYpSFydkBxs,2848
 airflow/providers/microsoft/azure/operators/data_factory.py,sha256=TQz7zQ3zBKOHhackZxMTdBFo66RenoPwcO8gXaLVXIs,12797
-airflow/providers/microsoft/azure/operators/msgraph.py,sha256=Mz0r3Gz3Pv_mg_myXEVcuZEnJ-GgjIwJmLqk2_yt8lI,14163
+airflow/providers/microsoft/azure/operators/msgraph.py,sha256=d8PU8NqXq5_4YA0_aYlqpUfh58ZGp95hmR7yaTMiSJk,14157
 airflow/providers/microsoft/azure/operators/powerbi.py,sha256=qcIMFKemQLgiJKQBxciBl1cT8nOwYP-7IPzPosUAAo8,11863
-airflow/providers/microsoft/azure/operators/synapse.py,sha256=l-I72OIkiwpTJefP478Sxak_FnwcRSj_wIi4eDplGxQ,12742
+airflow/providers/microsoft/azure/operators/synapse.py,sha256=boHHnenDSaTehFnuf-upPeBA9ej5Mnx3u6HQyVXH_5g,12747
 airflow/providers/microsoft/azure/operators/wasb_delete_blob.py,sha256=Rigi5xFXkHFNcX4-VnA4fFxJlKHlevdsCExX6VJWCts,2748
 airflow/providers/microsoft/azure/secrets/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/microsoft/azure/secrets/key_vault.py,sha256=EmtGfyBtfefGu1ZTtZ5WswIleOx-nx8wst7xfcua2rI,8962
+airflow/providers/microsoft/azure/secrets/key_vault.py,sha256=zbgjqiQo-cWu1N6QD7gFXRGSbmHw4xgAN-TiCovKYk4,9051
 airflow/providers/microsoft/azure/sensors/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
 airflow/providers/microsoft/azure/sensors/cosmos.py,sha256=Py3n-LOqaH2XuTxQVDZQv4d4RMREqBrmhWZ4NIbLr9Y,2851
 airflow/providers/microsoft/azure/sensors/data_factory.py,sha256=u_8vcnVmjzuCrOJbTeexxytfpXlkz0bSMkuWYmyZktI,5214
@@ -50,10 +50,10 @@ airflow/providers/microsoft/azure/transfers/s3_to_wasb.py,sha256=jQgicPw4uGyRSss
 airflow/providers/microsoft/azure/transfers/sftp_to_wasb.py,sha256=l6kdtxq3SvCUo_Hx60t1YiX4bP8lF3WmbglICHk24vo,8409
 airflow/providers/microsoft/azure/triggers/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/microsoft/azure/triggers/data_factory.py,sha256=U3vY_pj4yORxE7X6YR7CP3Jl73K9euCdUczy3dLmikU,11165
-airflow/providers/microsoft/azure/triggers/msgraph.py,sha256=l7A50JoBebiKhhsxILtLvuoulyIn59BVdjTvdAezdpk,8704
-airflow/providers/microsoft/azure/triggers/powerbi.py,sha256=TD2VYR3yj8JwRMR6QpqWM_KuBsS9qbghqV_2aBKjCus,15798
+airflow/providers/microsoft/azure/triggers/msgraph.py,sha256=iz9WRgI0R-LfGsPMiByvGZxhFA2pvIDlh5c03IuzZxU,8771
+airflow/providers/microsoft/azure/triggers/powerbi.py,sha256=z9LJ-PfgWstt7c2eIJX5LOGUicfdMv2EELhCsrCJ6Nk,16371
 airflow/providers/microsoft/azure/triggers/wasb.py,sha256=RF-C6iqDEs6_pWireCWZXqxcqWK-sFJ695Okdd_EJOA,7456
-apache_airflow_providers_microsoft_azure-12.6.1.dist-info/entry_points.txt,sha256=6iWHenOoUC3YZBb3OKn6g0HlJsV58Ba56i8USmQrcJI,111
-apache_airflow_providers_microsoft_azure-12.6.1.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-apache_airflow_providers_microsoft_azure-12.6.1.dist-info/METADATA,sha256=2p8_6jS9mhDZkUKi8j3gRguMk2rNQUjmOa7cdU5dv3Q,8863
-apache_airflow_providers_microsoft_azure-12.6.1.dist-info/RECORD,,
+apache_airflow_providers_microsoft_azure-12.7.0.dist-info/entry_points.txt,sha256=6iWHenOoUC3YZBb3OKn6g0HlJsV58Ba56i8USmQrcJI,111
+apache_airflow_providers_microsoft_azure-12.7.0.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+apache_airflow_providers_microsoft_azure-12.7.0.dist-info/METADATA,sha256=Lq3YJ08cpjOA4JWHfDmrWLNFV51KVeFJy_Aa8SW-hz8,8644
+apache_airflow_providers_microsoft_azure-12.7.0.dist-info/RECORD,,