yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-10-10 16:25:37 +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 pathlib import Path
from airflow import settings from airflow import settings
from airflow.models import DagModel from airflow.models import DagModel, DagRun
from flask import Response from flask import Response
from openmetadata.api.config import AIRFLOW_DAGS_FOLDER, DAG_GENERATED_CONFIGS from openmetadata.api.config import AIRFLOW_DAGS_FOLDER, DAG_GENERATED_CONFIGS
from openmetadata.api.response import ApiResponse from openmetadata.api.response import ApiResponse
@ -27,7 +27,7 @@ def delete_dag_id(dag_id: str) -> Response:
We clean: We clean:
- py file in AIRFLOW_DAGS_FOLDER - py file in AIRFLOW_DAGS_FOLDER
- config file in DAG_GENERATED_CONFIGS - 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 :param dag_id: DAG to delete
:return: API Response :return: API Response
""" """
@ -50,6 +50,7 @@ def delete_dag_id(dag_id: str) -> Response:
deleted_dags = ( deleted_dags = (
session.query(DagModel).filter(DagModel.dag_id == dag_id).delete() session.query(DagModel).filter(DagModel.dag_id == dag_id).delete()
) )
session.query(DagRun).filter(DagRun.dag_id == dag_id).delete()
session.commit() session.commit()
if deleted_dags > 0 and deleted_file and deleted_config: if deleted_dags > 0 and deleted_file and deleted_config: