Fix SQLA deprecations in Airflow core (#39211)

apache · Apr 25, 2024 · 6cb9def · 6cb9def
1 parent d08f893
commit 6cb9def
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/airflow/utils/db_cleanup.py b/airflow/utils/db_cleanup.py
@@ -196,8 +196,8 @@ def _do_delete(*, query, orm_model, skip_archive, session):
     session.execute(delete)
     session.commit()
     if skip_archive:
-        metadata.bind = session.get_bind()
-        target_table.drop()
+        bind = session.get_bind()
+        target_table.drop(bind=bind)
     session.commit()
     print("Finished Performing Delete")
 

diff --git a/airflow/www/views.py b/airflow/www/views.py
@@ -846,9 +846,7 @@ def index(self):
 
             is_paused_count = dict(
                 session.execute(
-                    all_dags.with_only_columns([DagModel.is_paused, func.count()]).group_by(
-                        DagModel.is_paused
-                    )
+                    all_dags.with_only_columns(DagModel.is_paused, func.count()).group_by(DagModel.is_paused)
                 ).all()
             )
 
@@ -3294,7 +3292,7 @@ def next_run_datasets(self, dag_id):
             latest_run = dag_model.get_last_dagrun(session=session)
 
             events = [
-                dict(info)
+                dict(info._mapping)
                 for info in session.execute(
                     select(
                         DatasetModel.id,
@@ -3456,7 +3454,7 @@ def datasets_summary(self):
             count_query = count_query.where(*filters)
 
             query = session.execute(query)
-            datasets = [dict(dataset) for dataset in query]
+            datasets = [dict(dataset._mapping) for dataset in query]
             data = {"datasets": datasets, "total_entries": session.scalar(count_query)}
 
             return (