airflow.providers.amazon.aws.sensors.athena

Module Contents

class airflow.providers.amazon.aws.sensors.athena.AthenaSensor(*, query_execution_id: str, max_retries: Optional[int] = None, aws_conn_id: str = 'aws_default', sleep_time: int = 10, **kwargs)[source]

Bases: airflow.sensors.base.BaseSensorOperator

Asks for the state of the Query until it reaches a failure state or success state. If the query fails, the task will fail.

Parameters
  • query_execution_id (str) -- query_execution_id to check the state of

  • max_retries (int) -- Number of times to poll for query state before returning the current state, defaults to None

  • aws_conn_id (str) -- aws connection to use, defaults to 'aws_default'

  • sleep_time (int) -- Time in seconds to wait between two consecutive call to check query status on athena, defaults to 10

INTERMEDIATE_STATES = ['QUEUED', 'RUNNING'][source]
FAILURE_STATES = ['FAILED', 'CANCELLED'][source]
SUCCESS_STATES = ['SUCCEEDED'][source]
template_fields = ['query_execution_id'][source]
template_ext = [][source]
ui_color = #66c3ff[source]
poke(self, context: dict)[source]
hook(self)[source]

Create and return an AWSAthenaHook

Was this entry helpful?