Можно ли принудительно пометить успешное выполнение любой задачи в группе доступности базы данных через определенный промежуток времени программно в Airflow?

Question

У меня есть DAG с одной задачей, которая извлекает данные из API. Я хочу, чтобы эта задача извлекала данные только за определенный промежуток времени и помечала себя как УСПЕХ, чтобы после этого запускались задачи.

Обратите внимание, что нижеприведенные задачи зависят от задач, которые я хочу отметить УСПЕХ. Я знаю, что могу пометить задачу SUCCESS вручную из CLI или UI, но я хочу сделать это автоматически.

Возможно ли это сделать программно, используя python в Airflow?

Answer 1

Вы можете установить статус задачи, используя код python, например:

def set_task_status(**kwargs):
  execution_date = kwargs['execution_date']
  ti = TaskInstance(HiveOperatorTest, execution_date)
  ti.set_state(State.SUCCESS)