From d8383fd0e5f2bc255ca11b7c7f637e74e74025b1 Mon Sep 17 00:00:00 2001 From: Pere Miquel Brull Date: Wed, 11 May 2022 16:02:06 +0200 Subject: [PATCH] Clean DagRun info (#4877) --- .../src/openmetadata/operations/delete.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/openmetadata-airflow-apis/src/openmetadata/operations/delete.py b/openmetadata-airflow-apis/src/openmetadata/operations/delete.py index 5d7c5236764..f52ae83a33d 100644 --- a/openmetadata-airflow-apis/src/openmetadata/operations/delete.py +++ b/openmetadata-airflow-apis/src/openmetadata/operations/delete.py @@ -15,7 +15,7 @@ import os from pathlib import Path from airflow import settings -from airflow.models import DagModel +from airflow.models import DagModel, DagRun from flask import Response from openmetadata.api.config import AIRFLOW_DAGS_FOLDER, DAG_GENERATED_CONFIGS from openmetadata.api.response import ApiResponse @@ -27,7 +27,7 @@ def delete_dag_id(dag_id: str) -> Response: We clean: - py file in AIRFLOW_DAGS_FOLDER - config file in DAG_GENERATED_CONFIGS - - DagModel entries in airflow db (Not the DagRun) + - DagModel and DagRun entries in airflow db :param dag_id: DAG to delete :return: API Response """ @@ -50,6 +50,7 @@ def delete_dag_id(dag_id: str) -> Response: deleted_dags = ( session.query(DagModel).filter(DagModel.dag_id == dag_id).delete() ) + session.query(DagRun).filter(DagRun.dag_id == dag_id).delete() session.commit() if deleted_dags > 0 and deleted_file and deleted_config: