blob: 53a05a54667b8eee58830d6dbac8c631be148747 [file] [log] [blame]
:mod:`airflow.contrib.sensors.aws_athena_sensor`
================================================
.. py:module:: airflow.contrib.sensors.aws_athena_sensor
Module Contents
---------------
.. py:class:: AthenaSensor(query_execution_id, max_retires=None, aws_conn_id='aws_default', sleep_time=10, *args, **kwargs)
Bases: :class:`airflow.sensors.base_sensor_operator.BaseSensorOperator`
Asks for the state of the Query until it reaches a failure state or success state.
If it fails, failing the task.
:param query_execution_id: query_execution_id to check the state of
:type query_execution_id: str
:param max_retires: Number of times to poll for query state before
returning the current state, defaults to None
:type max_retires: int
:param aws_conn_id: aws connection to use, defaults to 'aws_default'
:type aws_conn_id: str
:param sleep_time: Time to wait between two consecutive call to
check query status on athena, defaults to 10
:type sleep_time: int
.. attribute:: INTERMEDIATE_STATES
:annotation: = ['QUEUED', 'RUNNING']
.. attribute:: FAILURE_STATES
:annotation: = ['FAILED', 'CANCELLED']
.. attribute:: SUCCESS_STATES
:annotation: = ['SUCCEEDED']
.. attribute:: template_fields
:annotation: = ['query_execution_id']
.. attribute:: template_ext
:annotation: = []
.. attribute:: ui_color
:annotation: = #66c3ff
.. method:: poke(self, context)
.. method:: get_hook(self)