on_failure_callback not working in airflow DAG()

user_15010774 0
Viewed 2748

I want to call two different functions for dag failure and success. For that i want to use on_failure_callback and on_success_callback in DAG() function.

As per my requirement, this callbacks should be on dag level and not task level. That's why i am writing this callbacks insiled DAG() functions while declaring dag variable.

but this callback function are not being called. Same function if i call on task level , then working fine.

This is my code:

def success():
      print("successful")
    
dag = DAG(dag_id='callback_test',schedule_interval=None,default_args=default_args,on_success_callback=success)
    
    def fun1(**kwargs):
        print("function called")
    
    task1 = PythonOperator(
        task_id='task1',
        provide_context=True,
        python_callable=fun1,
        dag=dag
        )
    
    task1
airflow
2 Answers
user_72194600

If you want to pass agrument, you can do it by creating higher order function just like crearing decorator returning actual callback function by outer function.

user_83745790
Related