有没有办法在气流中调用任务中的任务?

Question

我有以下逻辑：

def function_1():
if condition_1 == True:
function_2()
else:
function_3()
function_2()

有没有办法在气流中调用任务中的任务？

Answer 1

您可以使用 BranchPythonOperator 实现相同的目标，如下所示。

def function_1(**kwargs):
if condition_1 == True :
return 'function_2'
else:
return 'function_3'
check_task = BranchPythonOperator(
task_id='check_task',
python_callable=function_1, # defined above method holds the branching condition
provide_context=True,
dag=dag
)
function_2_task = BashOperator(
task_id= 'function_2',
bash_command="echo function_2 task executed",
dag=dag
)
function_3_task = BashOperator(
task_id= 'function_3',
bash_command="echo function_3 task executed",
dag=dag
)
function_2_task.set_upstream(check_task)
function_3_task.set_upstream(check_task)
function_3_task.set_upstream(function_2_task)

任务图视图