jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🔧 chore(update_all_tables.py): add missing index and foreign key constraints to apikey table

Browse source 
🔧 chore(update_all_tables.py): add missing index and foreign key constraints to flow table
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-25 15:31:16 -03:00
commit 648bb25b3f
1 changed files with 10 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
src/backend/langflow/alembic/versions/d3749cf7ac7e_update_all_tables.py
View file

@ -27,19 +27,21 @@ def upgrade() -> None:
op.add_column(
"apikey", sa.Column("user_id", sqlmodel.sql.sqltypes.GUID(), nullable=False)
)
op.create_index(op.f("ix_apikey_user_id"), "apikey", ["user_id"], unique=False)
op.create_foreign_key(None, "apikey", "user", ["user_id"], ["id"])
with contextlib.suppress(sa.exc.OperationalError):
op.alter_column("apikey", "name", existing_type=sa.VARCHAR(), nullable=True)
with contextlib.suppress(sa.exc.OperationalError):
op.create_index(op.f("ix_apikey_name"), "apikey", ["name"], unique=False)
op.create_index(op.f("ix_apikey_user_id"), "apikey", ["user_id"], unique=False)
op.create_foreign_key(None, "apikey", "user", ["user_id"], ["id"])
with contextlib.suppress(sa.exc.OperationalError):
op.drop_column("apikey", "create_at")
op.add_column(
"flow", sa.Column("user_id", sqlmodel.sql.sqltypes.GUID(), nullable=False)
)
op.create_index(op.f("ix_flow_user_id"), "flow", ["user_id"], unique=False)
op.create_foreign_key(None, "flow", "user", ["user_id"], ["id"])
with contextlib.suppress(sa.exc.OperationalError):
op.add_column(
"flow", sa.Column("user_id", sqlmodel.sql.sqltypes.GUID(), nullable=False)
)
with contextlib.suppress(sa.exc.OperationalError):
op.create_index(op.f("ix_flow_user_id"), "flow", ["user_id"], unique=False)
op.create_foreign_key(None, "flow", "user", ["user_id"], ["id"])
# ### end Alembic commands ###