fix(db): make migration statement conditional (#5625)

* fix: make migration statement conditional * Update src/dispatch/database/revisions/tenant/versions/2024-11-04_928b725d64f6.py Co-authored-by: David Whittaker <[email protected]> Signed-off-by: Rob Lambell <[email protected]> --------- Signed-off-by: Rob Lambell <[email protected]> Co-authored-by: David Whittaker <[email protected]>
Netflix · Dec 18, 2024 · 63f6bee · 63f6bee
1 parent 1150a71
commit 63f6bee
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/dispatch/database/revisions/tenant/versions/2024-11-04_928b725d64f6.py b/src/dispatch/database/revisions/tenant/versions/2024-11-04_928b725d64f6.py
@@ -52,7 +52,12 @@ def upgrade():
     if column_exists:
         op.drop_column("plugin_instance", "configuration")
 
-    op.drop_constraint("project_stable_priority_id_fkey", "project", type_="foreignkey")
+    foreign_keys = inspector.get_foreign_keys("project")
+
+    constraint_name = "project_stable_priority_id_fkey"
+    constraint_exists = any(fk["name"] == constraint_name for fk in foreign_keys)
+    if constraint_exists:
+        op.drop_constraint(constraint_name, "project", type_="foreignkey")
     # ### end Alembic commands ###