: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)