aljaz/CTFd
1
0
Fork 0
mirror of https://github.com/aljazceru/CTFd.git synced 2026-02-01 20:34:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update migration

Browse Source
This commit is contained in:
Kevin Chung
2020-05-07 11:23:48 -04:00
parent c04235a5d1
commit 52d0c2719a
1 changed files with 22 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
CTFd/plugins/dynamic_challenges/migrations/b37fb68807ea_add_cascading_delete_to_dynamic_.py
View File

@@ -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 ###