Source code for airflow.example_dags.example_setup_teardown_taskflow

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
"""Example DAG demonstrating the usage of setup and teardown tasks."""
from __future__ import annotations

import pendulum

from airflow.decorators import setup, task, task_group, teardown
from airflow.models.dag import DAG
from airflow.settings import _ENABLE_AIP_52

if _ENABLE_AIP_52:
    with DAG(
        start_date=pendulum.datetime(2021, 1, 1, tz="UTC"),
    ) as dag:
        # You can use the setup and teardown decorators to add setup and teardown tasks at the DAG level
[docs] def root_setup(): print("Hello from root_setup")
@teardown @task def root_teardown(): print("Goodbye from root_teardown") @task def normal(): print("I am just a normal task") @task_group def section_1(): # You can also have setup and teardown tasks at the task group level @setup @task def my_setup(): print("I set up") @teardown @task def my_teardown(): print("I tear down") @task def hello(): print("I say hello") my_setup() hello() my_teardown() root_setup() normal() >> section_1() root_teardown()

Was this entry helpful?