Update foreign key migration

Update foreign key migration
This commit is contained in:
pablonyx
2025-01-31 08:45:19 -08:00
committed by GitHub

View File

@@ -16,16 +16,18 @@ depends_on = None
def upgrade() -> None: def upgrade() -> None:
# First drop the existing FK constraints # Safely drop constraints if exists
op.drop_constraint( op.execute(
"inputprompt__user_input_prompt_id_fkey", """
"inputprompt__user", ALTER TABLE inputprompt__user
type_="foreignkey", DROP CONSTRAINT IF EXISTS inputprompt__user_input_prompt_id_fkey
"""
) )
op.drop_constraint( op.execute(
"inputprompt__user_user_id_fkey", """
"inputprompt__user", ALTER TABLE inputprompt__user
type_="foreignkey", DROP CONSTRAINT IF EXISTS inputprompt__user_user_id_fkey
"""
) )
# Recreate with ON DELETE CASCADE # Recreate with ON DELETE CASCADE
@@ -37,10 +39,11 @@ def upgrade() -> None:
["id"], ["id"],
ondelete="CASCADE", ondelete="CASCADE",
) )
op.create_foreign_key( op.create_foreign_key(
"inputprompt__user_user_id_fkey", "inputprompt__user_user_id_fkey",
"inputprompt__user", "inputprompt__user",
'"user"', "user",
["user_id"], ["user_id"],
["id"], ["id"],
ondelete="CASCADE", ondelete="CASCADE",
@@ -71,7 +74,7 @@ def downgrade() -> None:
op.create_foreign_key( op.create_foreign_key(
"inputprompt__user_user_id_fkey", "inputprompt__user_user_id_fkey",
"inputprompt__user", "inputprompt__user",
'"user"', "user",
["user_id"], ["user_id"],
["id"], ["id"],
) )