only drop fks for non-sqllite dbs

apache · Aug 15, 2024 · 403551b · 403551b
1 parent 99b1898
commit 403551b
Show file tree

Hide file tree

Showing 6 changed files with 18 additions and 6 deletions.
diff --git a/superset/migrations/versions/2024-08-13_15-17_39549add7bfc_remove_sl_table_columns_table.py b/superset/migrations/versions/2024-08-13_15-17_39549add7bfc_remove_sl_table_columns_table.py
@@ -33,7 +33,9 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_table_columns")
+    connection = op.get_bind()
+    if connection.dialect.name != "sqlite":
+        drop_fks_for_table("sl_table_columns")
     op.drop_table("sl_table_columns")
 
 

diff --git a/superset/migrations/versions/2024-08-13_15-23_38f4144e8558_remove_sl_dataset_tables.py b/superset/migrations/versions/2024-08-13_15-23_38f4144e8558_remove_sl_dataset_tables.py
@@ -33,7 +33,9 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_dataset_tables")
+    connection = op.get_bind()
+    if connection.dialect.name != "sqlite":
+        drop_fks_for_table("sl_dataset_tables")
     op.drop_table("sl_dataset_tables")
 
 

diff --git a/superset/migrations/versions/2024-08-13_15-27_e53fd48cc078_remove_sl_dataset_users.py b/superset/migrations/versions/2024-08-13_15-27_e53fd48cc078_remove_sl_dataset_users.py
@@ -33,7 +33,9 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_dataset_users")
+    connection = op.get_bind()
+    if connection.dialect.name != "sqlite":
+        drop_fks_for_table("sl_dataset_users")
     op.drop_table("sl_dataset_users")
 
 

diff --git a/superset/migrations/versions/2024-08-13_15-29_a6b32d2d07b1_remove_sl_columns.py b/superset/migrations/versions/2024-08-13_15-29_a6b32d2d07b1_remove_sl_columns.py
@@ -33,7 +33,9 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_columns")
+    connection = op.get_bind()
+    if connection.dialect.name != "sqlite":
+        drop_fks_for_table("sl_columns")
     op.drop_table("sl_columns")
 
 

diff --git a/superset/migrations/versions/2024-08-13_15-31_007a1abffe7e_remove_sl_tables.py b/superset/migrations/versions/2024-08-13_15-31_007a1abffe7e_remove_sl_tables.py
@@ -33,7 +33,9 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_tables")
+    connection = op.get_bind()
+    if connection.dialect.name != "sqlite":
+        drop_fks_for_table("sl_tables")
     op.drop_table("sl_tables")
 
 

diff --git a/superset/migrations/versions/2024-08-13_15-33_48cbb571fa3a_remove_sl_datasets.py b/superset/migrations/versions/2024-08-13_15-33_48cbb571fa3a_remove_sl_datasets.py
@@ -33,7 +33,9 @@
 
 
 def upgrade():
-    drop_fks_for_table("sl_datasets")
+    connection = op.get_bind()
+    if connection.dialect.name != "sqlite":
+        drop_fks_for_table("sl_datasets")
     op.drop_table("sl_datasets")