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

🔥 chore(adds_tables.py): remove 'flowstyle' table and related indices if it exists during upgrade to ensure data consistency

Browse source 
🔥 chore(adds_tables.py): remove 'flowstyle' table and related indices if it exists during downgrade to ensure data consistency
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-28 14:45:30 -03:00
commit 9604ace084
1 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/langflow/alembic/versions/260dbcc8b680_adds_tables.py
View file

@ -26,6 +26,15 @@ def upgrade() -> None:
inspector = Inspector.from_engine(conn) inspector = Inspector.from_engine(conn)
# List existing tables # List existing tables
existing_tables = inspector.get_table_names() existing_tables = inspector.get_table_names()
# Drop 'flowstyle' table if it exists
# and other related indices
if "flowstyle" in existing_tables:
op.drop_table("flowstyle")
if "ix_flowstyle_flow_id" in [
index["name"] for index in inspector.get_indexes("flowstyle")
]:
op.drop_index("ix_flowstyle_flow_id", table_name="flowstyle")
existing_indices_flow = [] existing_indices_flow = []
existing_fks_flow = [] existing_fks_flow = []
if "flow" in existing_tables: if "flow" in existing_tables:
@ -159,4 +168,10 @@ def downgrade() -> None:
batch_op.drop_index(batch_op.f("ix_user_username")) batch_op.drop_index(batch_op.f("ix_user_username"))
op.drop_table("user") op.drop_table("user")
if "flowstyle" in existing_tables:
op.drop_table("flowstyle")
if "component" in existing_tables:
op.drop_table("component")
# ### end Alembic commands ### # ### end Alembic commands ###