yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-10-09 15:56:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Clean DagRun info (#4877)

Browse Source
This commit is contained in:
Pere Miquel Brull 2022-05-11 16:02:06 +02:00 committed by GitHub
parent 21ab0ef6f5
commit d8383fd0e5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
openmetadata-airflow-apis/src/openmetadata/operations/delete.py
View File

@ -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: