How to clean airflow dags

AirflowCleaning.py

import sqlalchemy
import sys


def print_clear_dag(dag_id):
    """Clear all information for a DAG from airflow postgres database"""
    list_tables = ['xcom', 'task_instance', 'sla_miss', 'log', 'job', 'dag_run', 'dag',
                   'dag_stats', 'task_fail']
    for table in list_tables:
        queries = ["DELETE FROM {} WHERE dag_id='{}'".format(table, dag_id),
                   "DELETE FROM {} WHERE dag_id LIKE '{}'".format(table, dag_id)]
        for query in queries:
            print("{};".format(query))

def clear_dag(dag_id):
    """Clear all information for a DAG from airflow postgres database"""
    list_tables = ['xcom', 'task_instance', 'sla_miss', 'log', 'job', 'dag_run', 'dag',
                   'dag_stats', 'task_fail']
    uri = 'mysql+mysqldb://  /airflow'
    engine = sqlalchemy.create_engine(uri)
    with engine.connect() as conn:
        for table in list_tables:
            queries = ["DELETE FROM {} WHERE dag_id='{}'".format(table, dag_id),
                       "DELETE FROM {} WHERE dag_id LIKE '{}'".format(table, dag_id)]
            for query in queries:
                print("\t: running {}".format(query))
                conn.execute(query)


print("Deleting {}".format(sys.argv[1]))
print_clear_dag(sys.argv[1])