apache-airflow-providers-microsoft-azure 6.1.1__py3-none-any.whl → 6.1.2rc1__py3-none-any.whl

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

@@ -37,21 +37,20 @@ from airflow.providers.microsoft.azure.utils import _ensure_prefixes, get_field
 
 
 class AzureDataLakeHook(BaseHook):
-    """
-    This module contains integration with Azure Data Lake.
-
-    AzureDataLakeHook communicates via a REST API compatible with WebHDFS. Make sure that a
-    Airflow connection of type `azure_data_lake` exists. Authorization can be done by supplying a
-    login (=Client ID), password (=Client Secret) and extra fields tenant (Tenant) and account_name
-    (Account Name)(see connection `azure_data_lake_default` for an example).
+    """Integration with Azure Data Lake.
 
-    Interacts with Azure Data Lake.
+    AzureDataLakeHook communicates via a REST API compatible with WebHDFS. Make
+    sure that a Airflow connection of type ``azure_data_lake`` exists.
+    Authorization can be done by supplying a *login* (=Client ID), *password*
+    (=Client Secret), and extra fields *tenant* (Tenant) and *account_name*
+    (Account Name). See connection ``azure_data_lake_default`` for an example.
 
-    Client ID and client secret should be in user and password parameters.
+    Client ID and secret should be in user and password parameters.
     Tenant and account name should be extra field as
-    {"tenant": "<TENANT>", "account_name": "ACCOUNT_NAME"}.
+    ``{"tenant": "<TENANT>", "account_name": "ACCOUNT_NAME"}``.
 
-    :param azure_data_lake_conn_id: Reference to the :ref:`Azure Data Lake connection<howto/connection:adl>`.
+    :param azure_data_lake_conn_id: Reference to
+        :ref:`Azure Data Lake connection<howto/connection:adl>`.
     """
 
     conn_name_attr = "azure_data_lake_conn_id"
@@ -61,7 +60,7 @@ class AzureDataLakeHook(BaseHook):
 
     @staticmethod
     def get_connection_form_widgets() -> dict[str, Any]:
-        """Returns connection widgets to add to connection form"""
+        """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import StringField
@@ -76,7 +75,7 @@ class AzureDataLakeHook(BaseHook):
     @staticmethod
     @_ensure_prefixes(conn_type="azure_data_lake")
     def get_ui_field_behaviour() -> dict[str, Any]:
-        """Returns custom field behaviour"""
+        """Returns custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "host", "extra"],
             "relabeling": {
@@ -118,8 +117,7 @@ class AzureDataLakeHook(BaseHook):
         return self._conn
 
     def check_for_file(self, file_path: str) -> bool:
-        """
-        Check if a file exists on Azure Data Lake.
+        """Check if a file exists on Azure Data Lake.
 
         :param file_path: Path and name of the file.
         :return: True if the file exists, False otherwise.
@@ -140,8 +138,7 @@ class AzureDataLakeHook(BaseHook):
         blocksize: int = 4194304,
         **kwargs,
     ) -> None:
-        """
-        Upload a file to Azure Data Lake.
+        """Upload a file to Azure Data Lake.
 
         :param local_path: local path. Can be single file, directory (in which case,
             upload recursively) or glob pattern. Recursive glob patterns using `**`
@@ -181,8 +178,7 @@ class AzureDataLakeHook(BaseHook):
         blocksize: int = 4194304,
         **kwargs,
     ) -> None:
-        """
-        Download a file from Azure Blob Storage.
+        """Download a file from Azure Blob Storage.
 
         :param local_path: local path. If downloading a single file, will write to this
             specific file, unless it is an existing directory, in which case a file is
@@ -214,8 +210,7 @@ class AzureDataLakeHook(BaseHook):
         )
 
     def list(self, path: str) -> list:
-        """
-        List files in Azure Data Lake Storage
+        """List files in Azure Data Lake Storage.
 
         :param path: full path/globstring to use to list files in ADLS
         """
@@ -225,8 +220,7 @@ class AzureDataLakeHook(BaseHook):
             return self.get_conn().walk(path)
 
     def remove(self, path: str, recursive: bool = False, ignore_not_found: bool = True) -> None:
-        """
-        Remove files in Azure Data Lake Storage
+        """Remove files in Azure Data Lake Storage.
 
         :param path: A directory or file to remove in ADLS
         :param recursive: Whether to loop into directories in the location and remove the files
@@ -242,13 +236,14 @@ class AzureDataLakeHook(BaseHook):
 
 
 class AzureDataLakeStorageV2Hook(BaseHook):
-    """
-    This Hook interacts with ADLS gen2 storage account it mainly helps to create and manage
-    directories and files in storage accounts that have a hierarchical namespace. Using Adls_v2 connection
-    details create DataLakeServiceClient object
+    """Interact with a ADLS gen2 storage account.
+
+    It mainly helps to create and manage directories and files in storage
+    accounts that have a hierarchical namespace. Using Adls_v2 connection
+    details create DataLakeServiceClient object.
 
-    Due to Wasb is marked as legacy and and retirement of the (ADLS1) it would be nice to
-    implement ADLS gen2 hook for interacting with the storage account.
+    Due to Wasb is marked as legacy and and retirement of the (ADLS1), it would
+    be nice to implement ADLS gen2 hook for interacting with the storage account.
 
     .. seealso::
         https://learn.microsoft.com/en-us/azure/storage/blobs/data-lake-storage-directory-file-acl-python
@@ -264,7 +259,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
 
     @staticmethod
     def get_connection_form_widgets() -> dict[str, Any]:
-        """Returns connection widgets to add to connection form"""
+        """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3PasswordFieldWidget, BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import PasswordField, StringField
@@ -280,7 +275,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
 
     @staticmethod
     def get_ui_field_behaviour() -> dict[str, Any]:
-        """Returns custom field behaviour"""
+        """Returns custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port"],
             "relabeling": {
@@ -342,9 +337,9 @@ class AzureDataLakeStorageV2Hook(BaseHook):
         return extra_dict.get(f"{prefix}{field_name}") or None
 
     def create_file_system(self, file_system_name: str) -> None:
-        """
-        A container acts as a file system for your files. Creates a new file system under
-        the specified account.
+        """Create a new file system under the specified account.
+
+        A container acts as a file system for your files.
 
         If the file system with the same name already exists, a ResourceExistsError will
         be raised. This method returns a client with which to interact with the newly
@@ -360,8 +355,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
             raise
 
     def get_file_system(self, file_system: FileSystemProperties | str) -> FileSystemClient:
-        """
-        Get a client to interact with the specified file system
+        """Get a client to interact with the specified file system.
 
         :param file_system: This can either be the name of the file system
             or an instance of FileSystemProperties.
@@ -379,8 +373,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
     def create_directory(
         self, file_system_name: FileSystemProperties | str, directory_name: str, **kwargs
     ) -> DataLakeDirectoryClient:
-        """
-        Create a directory under the specified file system.
+        """Create a directory under the specified file system.
 
         :param file_system_name: Name of the file system or instance of FileSystemProperties.
         :param directory_name: Name of the directory which needs to be created in the file system.
@@ -393,8 +386,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
         file_system_name: FileSystemProperties | str,
         directory_name: DirectoryProperties | str,
     ) -> DataLakeDirectoryClient:
-        """
-        Get the specific directory under the specified file system.
+        """Get the specific directory under the specified file system.
 
         :param file_system_name: Name of the file system or instance of FileSystemProperties.
         :param directory_name: Name of the directory or instance of DirectoryProperties which needs to be
@@ -413,8 +405,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
             raise
 
     def create_file(self, file_system_name: FileSystemProperties | str, file_name: str) -> DataLakeFileClient:
-        """
-        Creates a file under the file system
+        """Create a file under the file system.
 
         :param file_system_name: Name of the file system or instance of FileSystemProperties.
         :param file_name: Name of the file which needs to be created in the file system.
@@ -430,8 +421,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
         overwrite: bool = False,
         **kwargs: Any,
     ) -> None:
-        """
-        Create a file with data in the file system
+        """Create a file with data in the file system.
 
         :param file_system_name: Name of the file system or instance of FileSystemProperties.
         :param file_name: Name of the file to be created with name.
@@ -451,9 +441,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
         overwrite: bool = False,
         **kwargs: Any,
     ) -> None:
-        """
-        Create a new file and return the file client to be interacted with and then
-        upload data to a file
+        """Upload data to a file.
 
         :param file_system_name: Name of the file system or instance of FileSystemProperties.
         :param directory_name: Name of the directory.
@@ -469,8 +457,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
     def list_files_directory(
         self, file_system_name: FileSystemProperties | str, directory_name: str
     ) -> list[str]:
-        """
-        Get the list of files or directories under the specified file system
+        """List files or directories under the specified file system.
 
         :param file_system_name: Name of the file system or instance of FileSystemProperties.
         :param directory_name: Name of the directory.
@@ -484,8 +471,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
     def list_file_system(
         self, prefix: str | None = None, include_metadata: bool = False, **kwargs: Any
     ) -> list[str]:
-        """
-        Get the list the file systems under the specified account.
+        """List file systems under the specified account.
 
         :param prefix:
             Filters the results to return only file systems whose names
@@ -502,8 +488,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
         return file_system_list
 
     def delete_file_system(self, file_system_name: FileSystemProperties | str) -> None:
-        """
-        Deletes the file system
+        """Delete the file system.
 
         :param file_system_name: Name of the file system or instance of FileSystemProperties.
         """
@@ -517,8 +502,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
             raise
 
     def delete_directory(self, file_system_name: FileSystemProperties | str, directory_name: str) -> None:
-        """
-        Deletes specified directory in file system
+        """Delete the specified directory in a file system.
 
         :param file_system_name: Name of the file system or instance of FileSystemProperties.
         :param directory_name: Name of the directory.

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

@@ -76,7 +76,7 @@ class AzureFileShareHook(BaseHook):
 
     @staticmethod
     def get_connection_form_widgets() -> dict[str, Any]:
-        """Returns connection widgets to add to connection form"""
+        """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3PasswordFieldWidget, BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import PasswordField, StringField
@@ -94,7 +94,7 @@ class AzureFileShareHook(BaseHook):
     @staticmethod
     @_ensure_prefixes(conn_type="azure_fileshare")
     def get_ui_field_behaviour() -> dict[str, Any]:
-        """Returns custom field behaviour"""
+        """Returns custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "host", "extra"],
             "relabeling": {

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

@@ -62,7 +62,7 @@ class AzureSynapseHook(BaseHook):
 
     @staticmethod
     def get_connection_form_widgets() -> dict[str, Any]:
-        """Returns connection widgets to add to connection form"""
+        """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import StringField
@@ -74,7 +74,7 @@ class AzureSynapseHook(BaseHook):
 
     @staticmethod
     def get_ui_field_behaviour() -> dict[str, Any]:
-        """Returns custom field behaviour"""
+        """Returns custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "extra"],
             "relabeling": {"login": "Client ID", "password": "Secret", "host": "Synapse Workspace URL"},

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

@@ -103,7 +103,7 @@ class WasbHook(BaseHook):
 
     @staticmethod
     def get_connection_form_widgets() -> dict[str, Any]:
-        """Returns connection widgets to add to connection form"""
+        """Returns connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3PasswordFieldWidget, BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import PasswordField, StringField
@@ -124,7 +124,7 @@ class WasbHook(BaseHook):
     @staticmethod
     @_ensure_prefixes(conn_type="wasb")
     def get_ui_field_behaviour() -> dict[str, Any]:
-        """Returns custom field behaviour"""
+        """Returns custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "extra"],
             "relabeling": {
@@ -221,7 +221,7 @@ class WasbHook(BaseHook):
 
     def _get_container_client(self, container_name: str) -> ContainerClient:
         """
-        Instantiates a container client
+        Instantiates a container client.
 
         :param container_name: The name of the container
         :return: ContainerClient
@@ -230,7 +230,7 @@ class WasbHook(BaseHook):
 
     def _get_blob_client(self, container_name: str, blob_name: str) -> BlobClient:
         """
-        Instantiates a blob client
+        Instantiates a blob client.
 
         :param container_name: The name of the blob container
         :param blob_name: The name of the blob. This needs not be existing
@@ -273,7 +273,7 @@ class WasbHook(BaseHook):
         **kwargs,
     ) -> list:
         """
-        List blobs in a given container
+        List blobs in a given container.
 
         :param container_name: The name of the container
         :param prefix: Filters the results to return only blobs whose names
@@ -400,7 +400,7 @@ class WasbHook(BaseHook):
         self, container_name, blob_name, offset: int | None = None, length: int | None = None, **kwargs
     ) -> StorageStreamDownloader:
         """
-        Downloads a blob to the StorageStreamDownloader
+        Downloads a blob to the StorageStreamDownloader.
 
         :param container_name: The name of the container containing the blob
         :param blob_name: The name of the blob to download
@@ -413,7 +413,7 @@ class WasbHook(BaseHook):
 
     def create_container(self, container_name: str) -> None:
         """
-        Create container object if not already existing
+        Create container object if not already existing.
 
         :param container_name: The name of the container to create
         """
@@ -444,7 +444,7 @@ class WasbHook(BaseHook):
 
     def delete_container(self, container_name: str) -> None:
         """
-        Delete a container object
+        Delete a container object.
 
         :param container_name: The name of the container
         """

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

@@ -19,20 +19,21 @@ from __future__ import annotations
 
 import os
 import shutil
+from functools import cached_property
 from pathlib import Path
 from typing import TYPE_CHECKING, Any
 
 from azure.core.exceptions import HttpResponseError
 from packaging.version import Version
 
-from airflow.compat.functools import cached_property
 from airflow.configuration import conf
 from airflow.utils.log.file_task_handler import FileTaskHandler
 from airflow.utils.log.logging_mixin import LoggingMixin
 
 
 def get_default_delete_local_copy():
-    """Load delete_local_logs conf if Airflow version > 2.6 and return False if not
+    """Load delete_local_logs conf if Airflow version > 2.6 and return False if not.
+
     TODO: delete this function when min airflow version >= 2.6
     """
     from airflow.version import version
@@ -128,7 +129,7 @@ class WasbTaskHandler(FileTaskHandler, LoggingMixin):
         # Mark closed so we don't double write if close is called twice
         self.closed = True
 
-    def _read_remote_logs(self, ti, try_number, metadata=None):
+    def _read_remote_logs(self, ti, try_number, metadata=None) -> tuple[list[str], list[str]]:
         messages = []
         logs = []
         worker_log_relative_path = self._render_filename(ti, try_number)
@@ -192,7 +193,7 @@ class WasbTaskHandler(FileTaskHandler, LoggingMixin):
 
     def wasb_log_exists(self, remote_log_location: str) -> bool:
         """
-        Check if remote_log_location exists in remote storage
+        Check if remote_log_location exists in remote storage.
 
         :param remote_log_location: log's location in remote storage
         :return: True if location exists else False

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

@@ -64,7 +64,7 @@ class ADLSDeleteOperator(BaseOperator):
 
 class ADLSListOperator(BaseOperator):
     """
-    List all files from the specified path
+    List all files from the specified path.
 
     This operator returns a python list with the names of files which can be used by
      `xcom` in the downstream tasks.

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

@@ -15,7 +15,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-"""This module contains Azure Data Explorer operators"""
+"""This module contains Azure Data Explorer operators."""
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Sequence
@@ -62,14 +62,16 @@ class AzureDataExplorerQueryOperator(BaseOperator):
         self.azure_data_explorer_conn_id = azure_data_explorer_conn_id
 
     def get_hook(self) -> AzureDataExplorerHook:
-        """Returns new instance of AzureDataExplorerHook"""
+        """Returns new instance of AzureDataExplorerHook."""
         return AzureDataExplorerHook(self.azure_data_explorer_conn_id)
 
     def execute(self, context: Context) -> KustoResultTable | str:
         """
         Run KQL Query on Azure Data Explorer (Kusto).
-        Returns `PrimaryResult` of Query v2 HTTP response contents
-        (https://docs.microsoft.com/en-us/azure/kusto/api/rest/response2)
+
+        Returns `PrimaryResult` of Query v2 HTTP response contents.
+
+        https://docs.microsoft.com/en-us/azure/kusto/api/rest/response2
         """
         hook = self.get_hook()
         response = hook.run_query(self.query, self.database, self.options)

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

@@ -29,8 +29,7 @@ if TYPE_CHECKING:
 
 
 class AzureServiceBusCreateQueueOperator(BaseOperator):
-    """
-    Creates a Azure Service Bus queue under a Service Bus Namespace by using ServiceBusAdministrationClient
+    """Create a Azure Service Bus queue under a Service Bus Namespace.
 
     .. seealso::
         For more information on how to use this operator, take a look at the guide:
@@ -68,7 +67,7 @@ class AzureServiceBusCreateQueueOperator(BaseOperator):
         self.azure_service_bus_conn_id = azure_service_bus_conn_id
 
     def execute(self, context: Context) -> None:
-        """Creates Queue in Azure Service Bus namespace, by connecting to Service Bus Admin client in hook"""
+        """Creates Queue in Azure Service Bus namespace, by connecting to Service Bus Admin client in hook."""
         hook = AdminClientHook(azure_service_bus_conn_id=self.azure_service_bus_conn_id)
 
         # create queue with name
@@ -82,8 +81,7 @@ class AzureServiceBusCreateQueueOperator(BaseOperator):
 
 
 class AzureServiceBusSendMessageOperator(BaseOperator):
-    """
-    Send Message or batch message to the Service Bus queue
+    """Send Message or batch message to the Service Bus queue.
 
     .. seealso::
         For more information on how to use this operator, take a look at the guide:
@@ -116,10 +114,7 @@ class AzureServiceBusSendMessageOperator(BaseOperator):
         self.azure_service_bus_conn_id = azure_service_bus_conn_id
 
     def execute(self, context: Context) -> None:
-        """
-        Sends Message to the specific queue in Service Bus namespace, by
-        connecting to Service Bus  client
-        """
+        """Sends Message to the specific queue in Service Bus namespace."""
         # Create the hook
         hook = MessageHook(azure_service_bus_conn_id=self.azure_service_bus_conn_id)
 
@@ -128,8 +123,7 @@ class AzureServiceBusSendMessageOperator(BaseOperator):
 
 
 class AzureServiceBusReceiveMessageOperator(BaseOperator):
-    """
-    Receive a batch of messages at once in a specified Queue name
+    """Receive a batch of messages at once in a specified Queue name.
 
     .. seealso::
         For more information on how to use this operator, take a look at the guide:
@@ -163,7 +157,7 @@ class AzureServiceBusReceiveMessageOperator(BaseOperator):
     def execute(self, context: Context) -> None:
         """
         Receive Message in specific queue in Service Bus namespace,
-        by connecting to Service Bus client
+        by connecting to Service Bus client.
         """
         # Create the hook
         hook = MessageHook(azure_service_bus_conn_id=self.azure_service_bus_conn_id)
@@ -175,8 +169,7 @@ class AzureServiceBusReceiveMessageOperator(BaseOperator):
 
 
 class AzureServiceBusDeleteQueueOperator(BaseOperator):
-    """
-    Deletes the Queue in the Azure Service Bus namespace
+    """Delete the Queue in the Azure Service Bus namespace.
 
     .. seealso::
         For more information on how to use this operator, take a look at the guide:
@@ -202,7 +195,7 @@ class AzureServiceBusDeleteQueueOperator(BaseOperator):
         self.azure_service_bus_conn_id = azure_service_bus_conn_id
 
     def execute(self, context: Context) -> None:
-        """Delete Queue in Service Bus namespace, by connecting to Service Bus Admin client"""
+        """Delete Queue in Service Bus namespace, by connecting to Service Bus Admin client."""
         # Create the hook
         hook = AdminClientHook(azure_service_bus_conn_id=self.azure_service_bus_conn_id)
 
@@ -211,8 +204,7 @@ class AzureServiceBusDeleteQueueOperator(BaseOperator):
 
 
 class AzureServiceBusTopicCreateOperator(BaseOperator):
-    """
-    Create an Azure Service Bus Topic under a Service Bus Namespace by using ServiceBusAdministrationClient
+    """Create an Azure Service Bus Topic under a Service Bus Namespace.
 
     .. seealso::
         For more information on how to use this operator, take a look at the guide:
@@ -296,7 +288,7 @@ class AzureServiceBusTopicCreateOperator(BaseOperator):
         self.max_message_size_in_kilobytes = max_message_size_in_kilobytes
 
     def execute(self, context: Context) -> str:
-        """Creates Topic in Service Bus namespace, by connecting to Service Bus Admin client"""
+        """Creates Topic in Service Bus namespace, by connecting to Service Bus Admin client."""
         if self.topic_name is None:
             raise TypeError("Topic name cannot be None.")
 
@@ -330,9 +322,7 @@ class AzureServiceBusTopicCreateOperator(BaseOperator):
 
 
 class AzureServiceBusSubscriptionCreateOperator(BaseOperator):
-    """
-    Create an Azure Service Bus Topic Subscription under a Service Bus Namespace
-    by using ServiceBusAdministrationClient
+    """Create an Azure Service Bus Topic Subscription under a Service Bus Namespace.
 
     .. seealso::
         For more information on how to use this operator, take a look at the guide:
@@ -409,7 +399,7 @@ class AzureServiceBusSubscriptionCreateOperator(BaseOperator):
         self.azure_service_bus_conn_id = azure_service_bus_conn_id
 
     def execute(self, context: Context) -> None:
-        """Creates Subscription in Service Bus namespace, by connecting to Service Bus Admin client"""
+        """Creates Subscription in Service Bus namespace, by connecting to Service Bus Admin client."""
         if self.subscription_name is None:
             raise TypeError("Subscription name cannot be None.")
         if self.topic_name is None:
@@ -438,9 +428,7 @@ class AzureServiceBusSubscriptionCreateOperator(BaseOperator):
 
 
 class AzureServiceBusUpdateSubscriptionOperator(BaseOperator):
-    """
-    Update an Azure ServiceBus Topic Subscription under a ServiceBus Namespace
-    by using ServiceBusAdministrationClient
+    """Update an Azure ServiceBus Topic Subscription under a ServiceBus Namespace.
 
     .. seealso::
         For more information on how to use this operator, take a look at the guide:
@@ -481,7 +469,7 @@ class AzureServiceBusUpdateSubscriptionOperator(BaseOperator):
         self.azure_service_bus_conn_id = azure_service_bus_conn_id
 
     def execute(self, context: Context) -> None:
-        """Updates Subscription properties, by connecting to Service Bus Admin client"""
+        """Updates Subscription properties, by connecting to Service Bus Admin client."""
         hook = AdminClientHook(azure_service_bus_conn_id=self.azure_service_bus_conn_id)
 
         with hook.get_conn() as service_mgmt_conn:
@@ -499,8 +487,7 @@ class AzureServiceBusUpdateSubscriptionOperator(BaseOperator):
 
 
 class ASBReceiveSubscriptionMessageOperator(BaseOperator):
-    """
-    Receive a Batch messages from a Service Bus Subscription under specific Topic.
+    """Receive a Batch messages from a Service Bus Subscription under specific Topic.
 
     .. seealso::
         For more information on how to use this operator, take a look at the guide:
@@ -542,7 +529,7 @@ class ASBReceiveSubscriptionMessageOperator(BaseOperator):
     def execute(self, context: Context) -> None:
         """
         Receive Message in specific queue in Service Bus namespace,
-        by connecting to Service Bus client
+        by connecting to Service Bus client.
         """
         # Create the hook
         hook = MessageHook(azure_service_bus_conn_id=self.azure_service_bus_conn_id)
@@ -554,8 +541,7 @@ class ASBReceiveSubscriptionMessageOperator(BaseOperator):
 
 
 class AzureServiceBusSubscriptionDeleteOperator(BaseOperator):
-    """
-    Deletes the topic subscription in the Azure ServiceBus namespace
+    """Delete the topic subscription in the Azure ServiceBus namespace.
 
     .. seealso::
         For more information on how to use this operator, take a look at the guide:
@@ -584,7 +570,7 @@ class AzureServiceBusSubscriptionDeleteOperator(BaseOperator):
         self.azure_service_bus_conn_id = azure_service_bus_conn_id
 
     def execute(self, context: Context) -> None:
-        """Delete topic subscription in Service Bus namespace, by connecting to Service Bus Admin client"""
+        """Delete topic subscription in Service Bus namespace, by connecting to Service Bus Admin client."""
         # Create the hook
         hook = AdminClientHook(azure_service_bus_conn_id=self.azure_service_bus_conn_id)
 
@@ -593,8 +579,7 @@ class AzureServiceBusSubscriptionDeleteOperator(BaseOperator):
 
 
 class AzureServiceBusTopicDeleteOperator(BaseOperator):
-    """
-    Deletes the topic in the Azure Service Bus namespace
+    """Delete the topic in the Azure Service Bus namespace.
 
     .. seealso::
         For more information on how to use this operator, take a look at the guide:
@@ -620,7 +605,7 @@ class AzureServiceBusTopicDeleteOperator(BaseOperator):
         self.azure_service_bus_conn_id = azure_service_bus_conn_id
 
     def execute(self, context: Context) -> None:
-        """Delete topic in Service Bus namespace, by connecting to Service Bus Admin client"""
+        """Delete topic in Service Bus namespace, by connecting to Service Bus Admin client."""
         if self.topic_name is None:
             raise TypeError("Topic name cannot be None.")
         hook = AdminClientHook(azure_service_bus_conn_id=self.azure_service_bus_conn_id)

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

@@ -31,7 +31,7 @@ if TYPE_CHECKING:
 
 class AzureBatchOperator(BaseOperator):
     """
-    Executes a job on Azure Batch Service
+    Executes a job on Azure Batch Service.
 
     :param batch_pool_id: A string that uniquely identifies the Pool within the Account.
     :param batch_pool_vm_size: The size of virtual machines in the Pool
@@ -316,7 +316,7 @@ class AzureBatchOperator(BaseOperator):
 
     def clean_up(self, pool_id: str | None = None, job_id: str | None = None) -> None:
         """
-        Delete the given pool and job in the batch account
+        Delete the given pool and job in the batch account.
 
         :param pool_id: The id of the pool to delete
         :param job_id: The id of the job to delete

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

@@ -60,7 +60,7 @@ DEFAULT_CPU = 1.0
 
 class AzureContainerInstancesOperator(BaseOperator):
     """
-    Start a container on Azure Container Instances
+    Start a container on Azure Container Instances.
 
     :param ci_conn_id: connection id of a service principal which will be used
         to start the container instance