From 354a7b5985a307af7bae19d1cf23f1fc7b8f541f Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 30 Apr 2024 14:46:28 -0300 Subject: [PATCH] Refactor Alembic migration scripts to set name and value columns as not nullable --- .../c153816fd85f_set_name_and_value_to_not_nullable.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/base/langflow/alembic/versions/c153816fd85f_set_name_and_value_to_not_nullable.py b/src/backend/base/langflow/alembic/versions/c153816fd85f_set_name_and_value_to_not_nullable.py index 26cf872ab..cec3f6081 100644 --- a/src/backend/base/langflow/alembic/versions/c153816fd85f_set_name_and_value_to_not_nullable.py +++ b/src/backend/base/langflow/alembic/versions/c153816fd85f_set_name_and_value_to_not_nullable.py @@ -26,10 +26,10 @@ def upgrade() -> None: columns = inspector.get_columns("variable") with op.batch_alter_table("variable", schema=None) as batch_op: 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) 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) @@ -43,10 +43,10 @@ def downgrade() -> None: # ### commands auto generated by Alembic - please adjust! ### with op.batch_alter_table("variable", schema=None) as batch_op: 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) 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) # ### end Alembic commands ###