Remove dag parsing from db init command (#22531)

apache · Mar 31, 2022 · 8079b4c · 8079b4c
1 parent e7650b8
commit 8079b4c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/airflow/utils/db.py b/airflow/utils/db.py
@@ -644,13 +644,6 @@ def initdb(session: Session = NEW_SESSION):
 
     with create_global_lock(session=session, lock=DBLocks.MIGRATIONS):
 
-        dagbag = DagBag()
-        # Save DAGs in the ORM
-        dagbag.sync_to_db(session=session)
-
-        # Deactivate the unknown ones
-        DAG.deactivate_unknown_dags(dagbag.dags.keys(), session=session)
-
         from flask_appbuilder.models.sqla import Base
 
         Base.metadata.create_all(settings.engine)
@@ -1285,6 +1278,17 @@ def resetdb(session: Session = NEW_SESSION):
     initdb(session=session)
 
 
+@provide_session
+def bootstrap_dagbag(session: Session = NEW_SESSION):
+
+    dagbag = DagBag()
+    # Save DAGs in the ORM
+    dagbag.sync_to_db(session=session)
+
+    # Deactivate the unknown ones
+    DAG.deactivate_unknown_dags(dagbag.dags.keys(), session=session)
+
+
 @provide_session
 def downgrade(*, to_revision, from_revision=None, show_sql_only=False, session: Session = NEW_SESSION):
     """

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -169,6 +169,7 @@ def initial_db_init():
     from airflow.utils import db
 
     db.resetdb()
+    db.bootstrap_dagbag()
 
 
 @pytest.fixture(autouse=True, scope="session")