如何在气流日期中获取执行日期作为变量？

Question

我希望能够通过运行回填来测试DAG行为。 我有一个BranchPythonOperator，它使用dag运行的日期来决定要去哪个分支。

我用

datetime.today（）

但是我要寻找的是执行日期

today = datetime.today()

def check_working_day(today, **kwargs):
    cal = France()
    print(today)
    if (cal.is_working_day(today)):
        return "is_file_available"
    else:
        return "not_holiday"

今天= datetime.today（）

应该类似于

今天=执行日期_dag

Answer 1

您正在寻找ds或execution_date，它们在** kwargs中可用。看看macro reference和PythonOperator Documentation.

这应该有效：

def check_working_day(execution_date, **kwargs):
    cal = France()
    print(execution_date)
    if (cal.is_working_day(execution_date)):
        return "is_file_available"
    else:
        return "not_holiday"

您只需要确保您的PythonOperator设置了provide_context=True。