def pause_dag(self, dag_id, execution_date): """ Sets the pause flag on this dag/execution """ try: AIRFLOW_DB.pause_dag_run(dag_id=dag_id, execution_date=execution_date) except AirflowStateError as state_error: raise ApiError(title='Unable to pause action', description=state_error.message, status=falcon.HTTP_409)
def _get_tasks_db(dag_id, execution_date): """ Wrapper for call to the airflow db to get all tasks for a dag run :returns: a list of task dictionaries """ return AIRFLOW_DB.get_tasks_by_id(dag_id=dag_id, execution_date=execution_date)
def get_dag_run_db(self, dag_id, execution_date): """ Wrapper for call to the airflow db to get a dag_run :returns: a dag run dictionaries """ return AIRFLOW_DB.get_dag_runs_by_id( dag_id=dag_id, execution_date=execution_date)
def _get_dag_run_like_id_db(self, dag_id, execution_date): """ Wrapper for call to the airflow database to get a single action by id """ return AIRFLOW_DB.get_dag_runs_like_id(dag_id=dag_id, execution_date=execution_date)
def get_all_tasks_db(self): """ Wrapper for call to the airflow db to get all tasks :returns: a list of task dictionaries """ return AIRFLOW_DB.get_all_tasks()
def get_all_dag_runs_db(self): """ Wrapper for call to the airflow db to get all dag runs :returns: a list of dictionaries representing dag runs in airflow """ return AIRFLOW_DB.get_all_dag_runs()
def _get_all_dag_runs_db(self): """ Wrapper for call to the airflow database to get all actions :returns: a list of dag_runs dictionaries """ return AIRFLOW_DB.get_all_dag_runs()