From 4815313ac6f36154e6e59b0ee3ca87c04a36bb7c Mon Sep 17 00:00:00 2001
From: morpheus65535 <louis_vezina@hotmail.com>
Date: Wed, 8 May 2024 22:29:31 -0400
Subject: [PATCH] Fixed db migrations dropping tables content because of
 ForeignKey constraints. #2489

---
 migrations/env.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/migrations/env.py b/migrations/env.py
index d706218a2..beddf9710 100644
--- a/migrations/env.py
+++ b/migrations/env.py
@@ -1,6 +1,7 @@
 from flask import current_app
 
 from alembic import context
+from sqlalchemy import text
 
 import logging
 
@@ -95,8 +96,22 @@ def run_migrations_online():
         )
 
         with context.begin_transaction():
+            bind = context.get_bind()
+
+            if bind.engine.name == 'sqlite':
+                bind.execute(text("PRAGMA foreign_keys=OFF;"))
+            elif bind.engine.name == 'postgresql':
+                bind.execute(text("SET CONSTRAINTS ALL DEFERRED;"))
+
             context.run_migrations()
 
+            if bind.engine.name == 'sqlite':
+                bind.execute(text("PRAGMA foreign_keys=ON;"))
+            elif bind.engine.name == 'postgresql':
+                bind.execute(text("SET CONSTRAINTS ALL IMMEDIATE;"))
+
+            bind.close()
+
 
 if context.is_offline_mode():
     run_migrations_offline()