scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-09-06 08:18:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: catch edge case for no pks and check on downgrade

Browse source
This commit is contained in:
Ryan 2026-03-20 15:34:13 +13:00
parent 7f7cf8fb29
commit fcc99d9378
No known key found for this signature in database
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
backend/open_webui/migrations/versions/c3d4e5f6a7b8_add_missing_primary_keys.py
View file

@ -33,7 +33,7 @@ def upgrade() -> None:
inspector = Inspector.from_engine(conn)
for table, column in TABLES:
pk = inspector.get_pk_constraint(table)
pk = inspector.get_pk_constraint(table) or {}
if column in (pk.get("constrained_columns") or []):
continue
@ -46,6 +46,13 @@ def upgrade() -> None:
def downgrade() -> None:
conn = op.get_bind()
inspector = Inspector.from_engine(conn)
for table, column in TABLES:
pk = inspector.get_pk_constraint(table) or {}
if column not in (pk.get("constrained_columns") or []):
continue
with op.batch_alter_table(table, schema=None) as batch_op:
batch_op.drop_constraint(f"pk_{column}", type_="primary")