Merge pull request #60 from astronomer/old_sqlalchemy

DAG History Migration - Compatability for sqlalchemy 1.4 and 2.0
astronomer · Oct 17, 2023 · dd9d85f · dd9d85f
2 parents e27215f + de68415
commit dd9d85f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/astronomer_starship/starship/services/local_airflow_client.py b/astronomer_starship/starship/services/local_airflow_client.py
@@ -98,9 +98,8 @@ def receive_dag(session: Session, data: list = None):
         except NoSuchTableError:
             table = Table(table_name, metadata_obj, autoload_with=engine)
 
-        with engine.connect() as connection:
-            connection.execute(insert(table).on_conflict_do_nothing(), data_list)
-            connection.commit()
+        with engine.begin() as txn:
+            txn.execute(insert(table).on_conflict_do_nothing(), data_list)
 
 
 @provide_session