add renames to second migration

getsentry · Jul 17, 2024 · 0b94109 · 0b94109
1 parent 625f7f7
commit 0b94109
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/migrations/versions/2597db647e9a_drop_old_grouping_records_table.py b/src/migrations/versions/2597db647e9a_drop_old_grouping_records_table.py
@@ -16,8 +16,21 @@
 
 
 def upgrade():
+    op.execute("ALTER TABLE IF EXISTS grouping_records RENAME to grouping_records_old;")
+
+    op.execute("ALTER TABLE grouping_records_new RENAME TO grouping_records;")
+    for i in range(100):
+        op.execute(f"ALTER TABLE grouping_records_new_p{i} RENAME TO grouping_records_p{i};")
+
     op.execute("DROP TABLE IF EXISTS grouping_records_old")
 
 
 def downgrade():
-    pass
+    op.execute("ALTER TABLE IF EXISTS grouping_records RENAME TO grouping_records_new;")
+
+    for i in range(100):
+        op.execute(f"ALTER TABLE grouping_records_p{i} RENAME TO grouping_records_new_p{i};")
+
+    op.execute("ALTER TABLE grouping_records_old RENAME TO grouping_records;")
+
+    op.execute("DROP TABLE IF EXISTS grouping_records_new")