airflow_end_date_testing

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

default_args = {
    'owner': 'examples',
    'depends_on_past': False,
    'start_date': datetime(2018, 2, 1),
    'email_on_failure': False,
    'email_on_retry': False,
    'retries': 2,
    'retry_delay': timedelta(minutes=5),
}

dag = DAG(
    'testing_end_dates',
    default_args=default_args,
    schedule_interval='@daily',
)

demo_task_1 = BashOperator(
    task_id='demo_task_1',
    bash_command='echo "Counting to 10" && for i in {1..10}; do sleep 1 && echo $i; done',
    dag=dag,
)

demo_task_2 = BashOperator(
    task_id='demo_task_2',
    bash_command='echo "I finished counting!"',
    end_date=datetime(2018, 2, 2),
    dag=dag,
)

demo_task_2.set_upstream(demo_task_1)