要在Airflow的Jinja模板中获取参数,否则使用执行日期(ds_nodash
),可以使用以下代码示例:
from airflow import DAG
from airflow.operators.python_operator import PythonOperator
from datetime import datetime
from jinja2 import Template
def my_task(**kwargs):
template = Template("{{ var.value.my_parameter or execution_date.strftime('%Y%m%d') }}")
rendered_template = template.render(**kwargs)
print(rendered_template)
with DAG('my_dag', start_date=datetime(2021, 1, 1), schedule_interval='@daily') as dag:
task = PythonOperator(
task_id='my_task',
python_callable=my_task,
op_kwargs={'my_parameter': '{{var.value.my_parameter}}'},
provide_context=True
)
在上面的示例中,我们定义了一个名为my_task
的Python函数,它接受my_parameter
作为参数。在任务中,我们使用Jinja2
模板来获取参数值,如果参数未定义,则使用执行日期(ds_nodash
)。参数值从var.value.my_parameter
中获取,var.value
是Airflow中用于获取变量值的语法。
在PythonOperator
中,我们将my_parameter
作为op_kwargs
传递给任务,并设置provide_context=True
以提供上下文。这样,我们可以在任务中使用kwargs
来访问执行上下文中的变量和执行日期等信息。
最后,我们使用print
语句将渲染后的模板值打印出来。你可以根据需要修改任务的实现。