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

🐛 fix(alembic): add exception handling to prevent migration failure if columns already exist

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-18 18:18:07 -03:00
commit a65086c8e4
1 changed files with 17 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

30
src/backend/langflow/alembic/versions/595c9c2a2ad4_changes_for_the_store.py
View file

@ -21,25 +21,29 @@ depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None: def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
with op.batch_alter_table("flow", schema=None) as batch_op: try:
batch_op.add_column(sa.Column("is_component", sa.Boolean(), nullable=False)) with op.batch_alter_table("flow", schema=None) as batch_op:
batch_op.add_column(sa.Column("is_component", sa.Boolean(), nullable=False))
with op.batch_alter_table("user", schema=None) as batch_op: with op.batch_alter_table("user", schema=None) as batch_op:
batch_op.add_column( batch_op.add_column(
sa.Column( sa.Column(
"store_api_key", sqlmodel.sql.sqltypes.AutoString(), nullable=True "store_api_key", sqlmodel.sql.sqltypes.AutoString(), nullable=True
)
) )
) except Exception as exc:
pass
# ### end Alembic commands ### # ### end Alembic commands ###
def downgrade() -> None: def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
with op.batch_alter_table("user", schema=None) as batch_op: try:
batch_op.drop_column("store_api_key") with op.batch_alter_table("user", schema=None) as batch_op:
batch_op.drop_column("store_api_key")
with op.batch_alter_table("flow", schema=None) as batch_op:
batch_op.drop_column("is_component")
with op.batch_alter_table("flow", schema=None) as batch_op:
batch_op.drop_column("is_component")
except Exception as exc:
pass
# ### end Alembic commands ### # ### end Alembic commands ###