From 52d0c2719a0fc1b8d7be7d735cfbdfdce5da2056 Mon Sep 17 00:00:00 2001 From: Kevin Chung Date: Thu, 7 May 2020 11:23:48 -0400 Subject: [PATCH] Update migration --- ...8807ea_add_cascading_delete_to_dynamic_.py | 29 ++++++++++++++----- 1 file changed, 22 insertions(+), 7 deletions(-) diff --git a/CTFd/plugins/dynamic_challenges/migrations/b37fb68807ea_add_cascading_delete_to_dynamic_.py b/CTFd/plugins/dynamic_challenges/migrations/b37fb68807ea_add_cascading_delete_to_dynamic_.py index 6b4a35fc..b28d5708 100644 --- a/CTFd/plugins/dynamic_challenges/migrations/b37fb68807ea_add_cascading_delete_to_dynamic_.py +++ b/CTFd/plugins/dynamic_challenges/migrations/b37fb68807ea_add_cascading_delete_to_dynamic_.py @@ -13,10 +13,17 @@ depends_on = None def upgrade(op=None): - # ### commands auto generated by Alembic - please adjust! ### - op.drop_constraint( - "dynamic_challenge_ibfk_1", "dynamic_challenge", type_="foreignkey" - ) + bind = op.get_bind() + url = str(bind.engine.url) + if url.startswith("mysql"): + op.drop_constraint( + "dynamic_challenge_ibfk_1", "dynamic_challenge", type_="foreignkey" + ) + elif url.startswith("postgres"): + op.drop_constraint( + "dynamic_challenge_id_fkey", "dynamic_challenge", type_="foreignkey" + ) + op.create_foreign_key( None, "dynamic_challenge", "challenges", ["id"], ["id"], ondelete="CASCADE" ) @@ -24,7 +31,15 @@ def upgrade(op=None): def downgrade(op=None): - # ### commands auto generated by Alembic - please adjust! ### - op.drop_constraint(None, "dynamic_challenge", type_="foreignkey") + bind = op.get_bind() + url = str(bind.engine.url) + if url.startswith("mysql"): + op.drop_constraint( + "dynamic_challenge_ibfk_1", "dynamic_challenge", type_="foreignkey" + ) + elif url.startswith("postgres"): + op.drop_constraint( + "dynamic_challenge_id_fkey", "dynamic_challenge", type_="foreignkey" + ) + op.create_foreign_key(None, "dynamic_challenge", "challenges", ["id"], ["id"]) - # ### end Alembic commands ###