D401 Support in Microsoft providers (apache#37327)

Signed-off-by: Vincent Kling <[email protected]>

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

@@ -45,7 +45,7 @@
 
 class AzureDataExplorerHook(BaseHook):
     """
-    Interacts with Azure Data Explorer (Kusto).
+    Interact with Azure Data Explorer (Kusto).
 
     **Cluster**:
 
@@ -85,7 +85,7 @@ class AzureDataExplorerHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return 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
@@ -103,7 +103,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "extra"],
             "relabeling": {

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

@@ -48,7 +48,7 @@ class BaseAzureServiceBusHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import PasswordField, StringField
@@ -62,7 +62,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["port", "host", "extra", "login", "password"],
             "relabeling": {"schema": "Connection String"},

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

@@ -50,7 +50,7 @@ class AzureBaseHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import StringField
@@ -62,7 +62,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         import json
 
         return {
@@ -93,7 +93,7 @@ def __init__(self, sdk_client: Any, conn_id: str = "azure_default"):
 
     def get_conn(self) -> Any:
         """
-        Authenticates the resource using the connection id passed during init.
+        Authenticate the resource using the connection id passed during init.
 
         :return: the authenticated client.
         """

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

@@ -53,7 +53,7 @@ class AzureBatchHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import StringField
@@ -64,7 +64,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "host", "extra"],
             "relabeling": {
@@ -136,7 +136,7 @@ def configure_pool(
         **kwargs,
     ) -> PoolAddParameter:
         """
-        Configures a pool.
+        Configure a pool.
 
         :param pool_id: A string that uniquely identifies the Pool within the Account
 
@@ -221,7 +221,7 @@ def configure_pool(
 
     def create_pool(self, pool: PoolAddParameter) -> None:
         """
-        Creates a pool if not already existing.
+        Create a pool if not already existing.
 
         :param pool: the pool object to create
 
@@ -296,7 +296,7 @@ def configure_job(
         **kwargs,
     ) -> JobAddParameter:
         """
-        Configures a job for use in the pool.
+        Configure a job for use in the pool.
 
         :param job_id: A string that uniquely identifies the job within the account
         :param pool_id: A string that identifies the pool
@@ -312,7 +312,7 @@ def configure_job(
 
     def create_job(self, job: JobAddParameter) -> None:
         """
-        Creates a job in the pool.
+        Create a job in the pool.
 
         :param job: The job object to create
         """
@@ -334,7 +334,7 @@ def configure_task(
         **kwargs,
     ) -> TaskAddParameter:
         """
-        Creates a task.
+        Create a task.
 
         :param task_id: A string that identifies the task to create
         :param command_line: The command line of the Task.

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

@@ -66,7 +66,7 @@ def connection(self):
 
     def get_conn(self) -> Any:
         """
-        Authenticates the resource using the connection id passed during init.
+        Authenticate the resource using the connection id passed during init.
 
         :return: the authenticated client.
         """

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

@@ -49,7 +49,7 @@ class AzureContainerRegistryHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import StringField
@@ -67,7 +67,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "extra"],
             "relabeling": {

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

@@ -46,7 +46,7 @@ class AzureContainerVolumeHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return 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
@@ -67,7 +67,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "host", "extra"],
             "relabeling": {

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

@@ -44,7 +44,7 @@
 
 class AzureCosmosDBHook(BaseHook):
     """
-    Interacts with Azure CosmosDB.
+    Interact with Azure CosmosDB.
 
     login should be the endpoint uri, password should be the master key
     optionally, you can use the following extras to default these values
@@ -62,7 +62,7 @@ class AzureCosmosDBHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import StringField
@@ -86,7 +86,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "host", "extra"],
             "relabeling": {
@@ -180,7 +180,7 @@ def __get_collection_name(self, collection_name: str | None = None) -> str:
         return coll_name
 
     def does_collection_exist(self, collection_name: str, database_name: str) -> bool:
-        """Checks if a collection exists in CosmosDB."""
+        """Check if a collection exists in CosmosDB."""
         if collection_name is None:
             raise AirflowBadRequest("Collection name cannot be None.")
 
@@ -205,7 +205,7 @@ def create_collection(
         database_name: str | None = None,
         partition_key: str | None = None,
     ) -> None:
-        """Creates a new collection in the CosmosDB database."""
+        """Create a new collection in the CosmosDB database."""
         if collection_name is None:
             raise AirflowBadRequest("Collection name cannot be None.")
 
@@ -229,7 +229,7 @@ def create_collection(
             )
 
     def does_database_exist(self, database_name: str) -> bool:
-        """Checks if a database exists in CosmosDB."""
+        """Check if a database exists in CosmosDB."""
         if database_name is None:
             raise AirflowBadRequest("Database name cannot be None.")
 
@@ -247,7 +247,7 @@ def does_database_exist(self, database_name: str) -> bool:
         return True
 
     def create_database(self, database_name: str) -> None:
-        """Creates a new database in CosmosDB."""
+        """Create a new database in CosmosDB."""
         if database_name is None:
             raise AirflowBadRequest("Database name cannot be None.")
 
@@ -267,14 +267,14 @@ def create_database(self, database_name: str) -> None:
             self.get_conn().create_database(database_name)
 
     def delete_database(self, database_name: str) -> None:
-        """Deletes an existing database in CosmosDB."""
+        """Delete an existing database in CosmosDB."""
         if database_name is None:
             raise AirflowBadRequest("Database name cannot be None.")
 
         self.get_conn().delete_database(database_name)
 
     def delete_collection(self, collection_name: str, database_name: str | None = None) -> None:
-        """Deletes an existing collection in the CosmosDB database."""
+        """Delete an existing collection in the CosmosDB database."""
         if collection_name is None:
             raise AirflowBadRequest("Collection name cannot be None.")
 

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

@@ -159,7 +159,7 @@ class AzureDataFactoryHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import StringField
@@ -175,7 +175,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "host", "extra"],
             "relabeling": {
@@ -817,7 +817,7 @@ def wait_for_pipeline_run_status(
         timeout: int = 60 * 60 * 24 * 7,
     ) -> bool:
         """
-        Waits for a pipeline run to match an expected status.
+        Wait for a pipeline run to match an expected status.
 
         :param run_id: The pipeline run identifier.
         :param expected_statuses: The desired status(es) to check against a pipeline run's current status.

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

@@ -68,7 +68,7 @@ class AzureDataLakeHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import StringField
@@ -82,7 +82,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "host", "extra"],
             "relabeling": {
@@ -277,7 +277,7 @@ class AzureDataLakeStorageV2Hook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return 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
@@ -293,7 +293,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port"],
             "relabeling": {

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

@@ -45,7 +45,7 @@ class AzureFileShareHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return 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
@@ -59,7 +59,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "host", "extra"],
             "relabeling": {

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

@@ -72,7 +72,7 @@ class AzureSynapseHook(BaseHook):
     @classmethod
     @add_managed_identity_connection_widgets
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import StringField
@@ -84,7 +84,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "extra"],
             "relabeling": {
@@ -179,7 +179,7 @@ def wait_for_job_run_status(
         timeout: int = 60 * 60 * 24 * 7,
     ) -> bool:
         """
-        Waits for a job run to match an expected status.
+        Wait for a job run to match an expected status.
 
         :param job_id: The job run identifier.
         :param expected_statuses: The desired status(es) to check against a job run's current status.
@@ -255,7 +255,7 @@ class AzureSynapsePipelineHook(BaseHook):
 
     @classmethod
     def get_connection_form_widgets(cls) -> dict[str, Any]:
-        """Returns connection widgets to add to connection form."""
+        """Return connection widgets to add to connection form."""
         from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
         from flask_babel import lazy_gettext
         from wtforms import StringField
@@ -266,7 +266,7 @@ def get_connection_form_widgets(cls) -> dict[str, Any]:
 
     @classmethod
     def get_ui_field_behaviour(cls) -> dict[str, Any]:
-        """Returns custom field behaviour."""
+        """Return custom field behaviour."""
         return {
             "hidden_fields": ["schema", "port", "extra"],
             "relabeling": {"login": "Client ID", "password": "Secret", "host": "Synapse Workspace URL"},
@@ -362,7 +362,7 @@ def wait_for_pipeline_run_status(
         timeout: int = 60 * 60 * 24 * 7,
     ) -> bool:
         """
-        Waits for a pipeline run to match an expected status.
+        Wait for a pipeline run to match an expected status.
 
         :param run_id: The pipeline run identifier.
         :param expected_statuses: The desired status(es) to check against a pipeline run's current status.