Airflow配置Setup/Teardown资源时遇AttributeError问题求助

错误原因分析 #

• PythonOperator 报错 object has no attribute 'as_teardown'：
  Airflow的Setup/Teardown特性是2.4.0版本正式引入的，若你的Airflow版本低于此，这些方法不存在。执行airflow version确认版本，低于2.4.0需升级。
• @task装饰器报错 AttributeError: 'PlainXComArg' object has no attribute 'as_teardown'：
  你的代码存在两处问题：
  1. 错误将my_tear_down_task_obj1赋值为my_setup_task()，应为my_teardown_task()；
  2. 使用@task装饰器时，返回的是XComArg对象，无法直接调用as_setup()/as_teardown()，需通过装饰器参数或底层Task实例配置。

正确实现方式 #

方式1：PythonOperator（Airflow ≥2.4.0） #

版本符合要求后，修正代码如下：

from airflow.operators.python import PythonOperator
from time import sleep

def sleep_and_log(task_name):
    print(f"Executing {task_name}")
    sleep(1)

# 标记为setup任务
setup_task = PythonOperator(
    task_id="setup_task",
    python_callable=lambda: sleep_and_log("setup_task"),
).as_setup()

# 标记为teardown任务并关联setup
teardown_task = PythonOperator(
    task_id="teardown_task",
    python_callable=lambda: sleep_and_log("teardown_task"),
).as_teardown(setups=[setup_task])

# 示例下游任务（模拟失败场景）
def downstream_task_func():
    print("Running downstream task")
    raise Exception("Downstream task failed")

downstream_task = PythonOperator(
    task_id="downstream_task",
    python_callable=downstream_task_func,
)

# 设置依赖：setup成功后执行下游；teardown默认trigger_rule为all_done，setup完成后必执行
setup_task >> downstream_task

方式2：@task装饰器（Airflow ≥2.4.0） #

使用装饰器参数直接标记setup/teardown：

from airflow.decorators import task

# 用装饰器参数标记为setup任务
@task(setup=True)
def my_setup_task():
    print("Setting up resources!")

# 用装饰器参数标记为teardown任务
@task(teardown=True)
def my_teardown_task():
    print("Tearing down resources!")

# 下游任务（模拟失败）
@task
def downstream_task():
    print("Running downstream task")
    raise Exception("Downstream task failed")

# 实例化任务并设置依赖
setup_obj = my_setup_task()
downstream_obj = downstream_task()
teardown_obj = my_teardown_task()

# setup成功后执行下游；teardown关联setup，无论下游状态都会执行
setup_obj >> downstream_obj
setup_obj >> teardown_obj

需求满足关键配置 #

• 不跳过下游任务：下游任务默认trigger_rule为all_success，只要setup任务成功就会执行；若需忽略setup状态执行下游，可设置trigger_rule="all_done"。
• teardown执行后DAG保持失败状态：Airflow会以DAG中失败任务的状态作为最终状态，即使teardown任务成功执行，只要中间任务（如下游任务）失败，DAG最终状态仍为failed，无需额外配置。

内容的提问来源于stack exchange，提问作者Khilesh Chauhan