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

Refactor Alembic migration scripts to set name and value columns as not nullable

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-30 14:46:28 -03:00
commit 354a7b5985
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/base/langflow/alembic/versions/c153816fd85f_set_name_and_value_to_not_nullable.py
View file

@ -26,10 +26,10 @@ def upgrade() -> None:
columns = inspector.get_columns("variable") columns = inspector.get_columns("variable")
with op.batch_alter_table("variable", schema=None) as batch_op: with op.batch_alter_table("variable", schema=None) as batch_op:
name_column = [column for column in columns if column["name"] == "name"][0] name_column = [column for column in columns if column["name"] == "name"][0]
if name_column and name_column[0]["nullable"]: if name_column and name_column["nullable"]:
batch_op.alter_column("name", existing_type=sa.VARCHAR(), nullable=False) batch_op.alter_column("name", existing_type=sa.VARCHAR(), nullable=False)
value_column = [column for column in columns if column["name"] == "value"][0] value_column = [column for column in columns if column["name"] == "value"][0]
if value_column and value_column[0]["nullable"]: if value_column and value_column["nullable"]:
batch_op.alter_column("value", existing_type=sa.VARCHAR(), nullable=False) batch_op.alter_column("value", existing_type=sa.VARCHAR(), nullable=False)
@ -43,10 +43,10 @@ def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
with op.batch_alter_table("variable", schema=None) as batch_op: with op.batch_alter_table("variable", schema=None) as batch_op:
name_column = [column for column in columns if column["name"] == "name"][0] name_column = [column for column in columns if column["name"] == "name"][0]
if name_column and not name_column[0]["nullable"]: if name_column and not name_column["nullable"]:
batch_op.alter_column("name", existing_type=sa.VARCHAR(), nullable=True) batch_op.alter_column("name", existing_type=sa.VARCHAR(), nullable=True)
value_column = [column for column in columns if column["name"] == "value"][0] value_column = [column for column in columns if column["name"] == "value"][0]
if value_column and not value_column[0]["nullable"]: if value_column and not value_column["nullable"]:
batch_op.alter_column("name", existing_type=sa.VARCHAR(), nullable=True) batch_op.alter_column("name", existing_type=sa.VARCHAR(), nullable=True)
# ### end Alembic commands ### # ### end Alembic commands ###