From 6b84e549c0d5772be6a643583d1ab8f385fd1a2b Mon Sep 17 00:00:00 2001 From: Kirilishin Aleksei Date: Sun, 3 Dec 2023 01:31:14 +0400 Subject: [PATCH] 34058: Fix UI Grid error when DAG has been removed. --- airflow/www/views.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/airflow/www/views.py b/airflow/www/views.py index 649440865c35c..6ba23de9df1b2 100644 --- a/airflow/www/views.py +++ b/airflow/www/views.py @@ -3011,6 +3011,9 @@ def legacy_graph(self): def graph(self, dag_id: str, session: Session = NEW_SESSION): """Redirect to the replacement - grid + graph. Kept for backwards compatibility.""" dag = get_airflow_app().dag_bag.get_dag(dag_id, session=session) + if not dag: + flash(f'DAG "{dag_id}" seems to be missing from DagBag.', "error") + return redirect(url_for("Airflow.index")) dt_nr_dr_data = get_date_time_num_runs_dag_runs_form_data(request, session, dag) dttm = dt_nr_dr_data["dttm"] dag_run = dag.get_dagrun(execution_date=dttm)