highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-07-28 05:43:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update foreign key migration

Browse Source 
Update foreign key migration
This commit is contained in:
pablonyx
2025-01-31 08:45:19 -08:00
committed by GitHub
parent 17b280e59e ffe8ac168f
commit 3e0d24a3f6
1 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
backend/alembic/versions/33ea50e88f24_foreign_key_input_prompts.py
View File

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