小编Jac*_*son的帖子

Airflow - 从 XCOM 创建动态任务

我正在尝试从 XCOM 变量生成一组动态任务。在 XCOM 中,我存储了一个列表,我想使用列表中的每个元素来动态创建下游任务。

我的用例是我有一个上游操作符,它检查 sftp 服务器中的文件并返回匹配特定条件的文件名列表。我想为返回的每个文件名创建动态下游任务。

我已将其简化为以下内容,虽然它有效,但我觉得它不是惯用的气流解决方案。在我的用例中,我会编写一个从 python 运算符调用的 python 函数,该运算符从 xcom 中提取值并返回它,而不是使用 pusher 函数。

我知道虽然我可以创建一个结合了两者的自定义运算符,但我不认为创建一次性运算符是一种好习惯,我希望有另一种解决方案。

from airflow.operators.python_operator import PythonOperator
from airflow.operators.bash_operator import BashOperator
from airflow import DAG
from datetime import datetime, timedelta

default_args = {
    "owner": "test",
    "depends_on_past": False,
    "start_date": datetime(2018, 10, 27),
    "email": ["test@mctest.com"],
    "email_on_failure": False,
    "email_on_retry": False,
    "email_on_success": False,
    "retries": 0,
    "provide_context": True
}

dag = DAG("test",  default_args=default_args, schedule_interval="@daily", catchup=False)


def pusher(**context):
    return ['a', 'b', 'c', 'd', 'e']

pusher_task = PythonOperator(
    task_id='pusher_task',
    dag=dag,
    python_callable=pusher  
) …
Run Code Online (Sandbox Code Playgroud)

python airflow

11
推荐指数
1
解决办法
2721
查看次数

标签 统计

airflow ×1

python ×1