airflow.providers.databricks.utils.mixins¶

Classes¶

`GetHookHasFields`	Protocol for get_hook method.
`HandleExecutionHasFields`	Protocol for _handle_execution method.
`HandleDeferrableExecutionHasFields`	Protocol for _handle_deferrable_execution method.
`ExecuteCompleteHasFields`	Protocol for execute_complete method.
`OnKillHasFields`	Protocol for on_kill method.
`DatabricksSQLStatementsMixin`	Mixin class to be used by both the DatabricksSqlStatementsOperator, and the DatabricksSqlStatementSensor.

Module Contents¶

class airflow.providers.databricks.utils.mixins.GetHookHasFields[source]¶

Bases: Protocol

Protocol for get_hook method.

databricks_conn_id: str[source]¶

databricks_retry_args: dict | None[source]¶

databricks_retry_delay: int[source]¶

databricks_retry_limit: int[source]¶

class airflow.providers.databricks.utils.mixins.HandleExecutionHasFields[source]¶

Bases: Protocol

Protocol for _handle_execution method.

log: logging.Logger[source]¶

polling_period_seconds: int[source]¶

task_id: str[source]¶

timeout: int[source]¶

statement_id: str[source]¶

class airflow.providers.databricks.utils.mixins.HandleDeferrableExecutionHasFields[source]¶

Bases: Protocol

Protocol for _handle_deferrable_execution method.

databricks_conn_id: str[source]¶

databricks_retry_args: dict[Any, Any] | None[source]¶

databricks_retry_delay: int[source]¶

databricks_retry_limit: int[source]¶

defer: Any[source]¶

log: logging.Logger[source]¶

polling_period_seconds: int[source]¶

statement_id: str[source]¶

task_id: str[source]¶

timeout: int[source]¶

class airflow.providers.databricks.utils.mixins.ExecuteCompleteHasFields[source]¶

Bases: Protocol

Protocol for execute_complete method.

statement_id: str[source]¶

log: logging.Logger[source]¶

class airflow.providers.databricks.utils.mixins.OnKillHasFields[source]¶

Bases: Protocol

Protocol for on_kill method.

log: logging.Logger[source]¶

statement_id: str[source]¶

task_id: str[source]¶

class airflow.providers.databricks.utils.mixins.DatabricksSQLStatementsMixin[source]¶

Mixin class to be used by both the DatabricksSqlStatementsOperator, and the DatabricksSqlStatementSensor.

_handle_operator_execution (renamed to _handle_execution)

_handle_deferrable_operator_execution (renamed to _handle_deferrable_execution)

execute_complete

on_kill

execute_complete(context, event)[source]¶

on_kill()[source]¶