afni_dag.py

with DAG(
    dag_id='my_dag_name',
    schedule_interval=None,
    start_date=days_ago(2),
    catchup=False,
    ) as dag:

    preparation = PythonOperator(
        task_id='preparation',
        python_callable=prepare,
    )
    startup_task = PythonOperator(
        task_id='startup_task',
        python_callable=startup,
    )
    waiting_task = PythonOperator(
        task_id='waiting_task',
        python_callable=wait_for_update,
    )
    finalization = PythonOperator(
        task_id='continuation_task',
        python_callable=finalize,
    )
    preparation >> startup_task >> waiting_task >> finalization