Airflow 在 BashOperator 上获取/设置参数

Question

我正在尝试将一个值设置为参数，然后在 bashOperations 上使用它。它失败了，因为我不知道如何使用 task_instance.xcom_pull。请帮我获取/设置 bash 字符串中的参数。

dag = DAG(
    dag_id = 'stackover',
    schedule_interval='01 6 * * *',
    default_args=default_args,
    description='stackover',
    tags=['stackover'],
    catchup=False
)

week_op = BashOperator(
    task_id="week_op",
    bash_command='echo "{{ params.week }}"',
    params = {'week' : ' $(date +%V -d \"1 week ago\") ',},
    provide_context=True,
    xcom_push=True,
    dag=dag
)


start_op = BashOperator(
    task_id="start_op",
    bash_command=' {{ params.query }}"',
    params = {'query' : 'echo {{ (task_instance.xcom_pull(task_ids=\'week_op\')) }} ',},
    provide_context=True,
    xcom_push=True,
    dag=dag
)

week_op >> start_op

提前致谢

Answer 1

你可以直接在bash_command中使用Jinja作为：

start_op = BashOperator(
    task_id="start_op",
    bash_command=' echo {{ ti.xcom_pull("week_op") }}',
    provide_context=True,
    xcom_push=True,
    dag=dag
)