reorder how tables are created in the downgrade path

apache · Jun 3, 2024 · 316cd3c · 316cd3c
1 parent 8b64967
commit 316cd3c
Showing 1 changed file with 52 additions and 52 deletions.
diff --git a/superset/migrations/versions/2024-05-24_11-31_02f4f7811799_remove_sl__tables.py b/superset/migrations/versions/2024-05-24_11-31_02f4f7811799_remove_sl__tables.py
@@ -41,20 +41,6 @@ def upgrade():
 
 
 def downgrade():
-    op.create_table(
-        "sl_dataset_users",
-        sa.Column("dataset_id", sa.INTEGER(), nullable=False),
-        sa.Column("user_id", sa.INTEGER(), nullable=False),
-        sa.ForeignKeyConstraint(
-            ["dataset_id"],
-            ["sl_datasets.id"],
-        ),
-        sa.ForeignKeyConstraint(
-            ["user_id"],
-            ["ab_user.id"],
-        ),
-        sa.PrimaryKeyConstraint("dataset_id", "user_id"),
-    )
     op.create_table(
         "sl_datasets",
         sa.Column("uuid", sa.NUMERIC(precision=16), nullable=True),
@@ -116,18 +102,56 @@ def downgrade():
         sa.UniqueConstraint("uuid"),
     )
     op.create_table(
-        "sl_dataset_columns",
-        sa.Column("dataset_id", sa.INTEGER(), nullable=False),
-        sa.Column("column_id", sa.INTEGER(), nullable=False),
+        "sl_columns",
+        sa.Column("uuid", sa.NUMERIC(precision=16), nullable=True),
+        sa.Column("created_on", sa.DATETIME(), nullable=True),
+        sa.Column("changed_on", sa.DATETIME(), nullable=True),
+        sa.Column("id", sa.INTEGER(), nullable=False),
+        sa.Column("is_aggregation", sa.BOOLEAN(), nullable=False),
+        sa.Column("is_additive", sa.BOOLEAN(), nullable=False),
+        sa.Column("is_dimensional", sa.BOOLEAN(), nullable=False),
+        sa.Column("is_filterable", sa.BOOLEAN(), nullable=False),
+        sa.Column("is_increase_desired", sa.BOOLEAN(), nullable=False),
+        sa.Column("is_managed_externally", sa.BOOLEAN(), nullable=False),
+        sa.Column("is_partition", sa.BOOLEAN(), nullable=False),
+        sa.Column("is_physical", sa.BOOLEAN(), nullable=False),
+        sa.Column("is_temporal", sa.BOOLEAN(), nullable=False),
+        sa.Column("is_spatial", sa.BOOLEAN(), nullable=False),
+        sa.Column("name", sa.TEXT(), nullable=True),
+        sa.Column("type", sa.TEXT(), nullable=True),
+        sa.Column("unit", sa.TEXT(), nullable=True),
+        sa.Column("expression", sa.TEXT(), nullable=True),
+        sa.Column("description", sa.TEXT(), nullable=True),
+        sa.Column("warning_text", sa.TEXT(), nullable=True),
+        sa.Column("external_url", sa.TEXT(), nullable=True),
+        sa.Column("extra_json", sa.TEXT(), nullable=True),
+        sa.Column("created_by_fk", sa.INTEGER(), nullable=True),
+        sa.Column("changed_by_fk", sa.INTEGER(), nullable=True),
+        sa.Column("advanced_data_type", sa.TEXT(), nullable=True),
         sa.ForeignKeyConstraint(
-            ["column_id"],
-            ["sl_columns.id"],
+            ["changed_by_fk"],
+            ["ab_user.id"],
+        ),
+        sa.ForeignKeyConstraint(
+            ["created_by_fk"],
+            ["ab_user.id"],
         ),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("uuid"),
+    )
+    op.create_table(
+        "sl_dataset_users",
+        sa.Column("dataset_id", sa.INTEGER(), nullable=False),
+        sa.Column("user_id", sa.INTEGER(), nullable=False),
         sa.ForeignKeyConstraint(
             ["dataset_id"],
             ["sl_datasets.id"],
         ),
-        sa.PrimaryKeyConstraint("dataset_id", "column_id"),
+        sa.ForeignKeyConstraint(
+            ["user_id"],
+            ["ab_user.id"],
+        ),
+        sa.PrimaryKeyConstraint("dataset_id", "user_id"),
     )
     op.create_table(
         "sl_dataset_tables",
@@ -158,40 +182,16 @@ def downgrade():
         sa.PrimaryKeyConstraint("table_id", "column_id"),
     )
     op.create_table(
-        "sl_columns",
-        sa.Column("uuid", sa.NUMERIC(precision=16), nullable=True),
-        sa.Column("created_on", sa.DATETIME(), nullable=True),
-        sa.Column("changed_on", sa.DATETIME(), nullable=True),
-        sa.Column("id", sa.INTEGER(), nullable=False),
-        sa.Column("is_aggregation", sa.BOOLEAN(), nullable=False),
-        sa.Column("is_additive", sa.BOOLEAN(), nullable=False),
-        sa.Column("is_dimensional", sa.BOOLEAN(), nullable=False),
-        sa.Column("is_filterable", sa.BOOLEAN(), nullable=False),
-        sa.Column("is_increase_desired", sa.BOOLEAN(), nullable=False),
-        sa.Column("is_managed_externally", sa.BOOLEAN(), nullable=False),
-        sa.Column("is_partition", sa.BOOLEAN(), nullable=False),
-        sa.Column("is_physical", sa.BOOLEAN(), nullable=False),
-        sa.Column("is_temporal", sa.BOOLEAN(), nullable=False),
-        sa.Column("is_spatial", sa.BOOLEAN(), nullable=False),
-        sa.Column("name", sa.TEXT(), nullable=True),
-        sa.Column("type", sa.TEXT(), nullable=True),
-        sa.Column("unit", sa.TEXT(), nullable=True),
-        sa.Column("expression", sa.TEXT(), nullable=True),
-        sa.Column("description", sa.TEXT(), nullable=True),
-        sa.Column("warning_text", sa.TEXT(), nullable=True),
-        sa.Column("external_url", sa.TEXT(), nullable=True),
-        sa.Column("extra_json", sa.TEXT(), nullable=True),
-        sa.Column("created_by_fk", sa.INTEGER(), nullable=True),
-        sa.Column("changed_by_fk", sa.INTEGER(), nullable=True),
-        sa.Column("advanced_data_type", sa.TEXT(), nullable=True),
+        "sl_dataset_columns",
+        sa.Column("dataset_id", sa.INTEGER(), nullable=False),
+        sa.Column("column_id", sa.INTEGER(), nullable=False),
         sa.ForeignKeyConstraint(
-            ["changed_by_fk"],
-            ["ab_user.id"],
+            ["column_id"],
+            ["sl_columns.id"],
         ),
         sa.ForeignKeyConstraint(
-            ["created_by_fk"],
-            ["ab_user.id"],
+            ["dataset_id"],
+            ["sl_datasets.id"],
         ),
-        sa.PrimaryKeyConstraint("id"),
-        sa.UniqueConstraint("uuid"),
+        sa.PrimaryKeyConstraint("dataset_id", "column_id"),
     )