From 8cb4fe9f5f05f0e1a9b84c5e15c258bfbec0e2e9 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 16 Jan 2024 23:13:09 -0300 Subject: [PATCH] Add primary key and unique constraints to user, apikey, and flow tables --- .../alembic/versions/260dbcc8b680_adds_tables.py | 14 +++++++------- .../versions/2ac71eb9c3ae_adds_credential_table.py | 2 +- ...7d2162acc8b2_adds_updated_at_and_folder_cols.py | 6 +++--- .../eb5866d51fd2_change_columns_to_be_nullable.py | 6 +++--- 4 files changed, 14 insertions(+), 14 deletions(-) diff --git a/src/backend/langflow/alembic/versions/260dbcc8b680_adds_tables.py b/src/backend/langflow/alembic/versions/260dbcc8b680_adds_tables.py index 729e1898f..48c56e90d 100644 --- a/src/backend/langflow/alembic/versions/260dbcc8b680_adds_tables.py +++ b/src/backend/langflow/alembic/versions/260dbcc8b680_adds_tables.py @@ -60,8 +60,8 @@ def upgrade() -> None: sa.Column("create_at", sa.DateTime(), nullable=False), sa.Column("updated_at", sa.DateTime(), nullable=False), sa.Column("last_login_at", sa.DateTime(), nullable=True), - sa.PrimaryKeyConstraint("id"), - sa.UniqueConstraint("id"), + sa.PrimaryKeyConstraint("id", name="pk_user"), + sa.UniqueConstraint("id", name="uq_user_id"), ) with op.batch_alter_table("user", schema=None) as batch_op: batch_op.create_index( @@ -83,8 +83,8 @@ def upgrade() -> None: ["user_id"], ["user.id"], ), - sa.PrimaryKeyConstraint("id"), - sa.UniqueConstraint("id"), + sa.PrimaryKeyConstraint("id", name="pk_apikey"), + sa.UniqueConstraint("id", name="uq_apikey_id"), ) with op.batch_alter_table("apikey", schema=None) as batch_op: batch_op.create_index( @@ -106,8 +106,8 @@ def upgrade() -> None: ["user_id"], ["user.id"], ), - sa.PrimaryKeyConstraint("id"), - sa.UniqueConstraint("id"), + sa.PrimaryKeyConstraint("id", name="pk_flow"), + sa.UniqueConstraint("id", name="uq_flow_id"), ) # Conditionally create indices for 'flow' table # if _alembic_tmp_flow exists, then we need to drop it first @@ -145,7 +145,7 @@ def upgrade() -> None: def downgrade() -> None: # ### commands auto generated by Alembic - please adjust! ### - conn = op.get_bind() + conn = op.get_bind() inspector = Inspector.from_engine(conn) # type: ignore # List existing tables existing_tables = inspector.get_table_names() diff --git a/src/backend/langflow/alembic/versions/2ac71eb9c3ae_adds_credential_table.py b/src/backend/langflow/alembic/versions/2ac71eb9c3ae_adds_credential_table.py index 3f974dc04..351b3d8f1 100644 --- a/src/backend/langflow/alembic/versions/2ac71eb9c3ae_adds_credential_table.py +++ b/src/backend/langflow/alembic/versions/2ac71eb9c3ae_adds_credential_table.py @@ -29,7 +29,7 @@ def upgrade() -> None: sa.Column('id', sqlmodel.sql.sqltypes.GUID(), nullable=False), sa.Column('created_at', sa.DateTime(), nullable=False), sa.Column('updated_at', sa.DateTime(), nullable=True), - sa.PrimaryKeyConstraint('id') + sa.PrimaryKeyConstraint('id', name=op.f('pk_credential')), ) except Exception: pass diff --git a/src/backend/langflow/alembic/versions/7d2162acc8b2_adds_updated_at_and_folder_cols.py b/src/backend/langflow/alembic/versions/7d2162acc8b2_adds_updated_at_and_folder_cols.py index f8280053b..ddf8c67f2 100644 --- a/src/backend/langflow/alembic/versions/7d2162acc8b2_adds_updated_at_and_folder_cols.py +++ b/src/backend/langflow/alembic/versions/7d2162acc8b2_adds_updated_at_and_folder_cols.py @@ -67,8 +67,8 @@ def downgrade() -> None: sa.Column('flow_id', sa.CHAR(length=32), nullable=True), sa.Column('id', sa.CHAR(length=32), nullable=False), sa.ForeignKeyConstraint(['flow_id'], ['flow.id'], ), - sa.PrimaryKeyConstraint('id'), - sa.UniqueConstraint('id') + sa.PrimaryKeyConstraint('id', name=op.f('pk_flowstyle')) + sa.UniqueConstraint('id', name=op.f('uq_flowstyle_id')) ) op.create_table('component', sa.Column('id', sa.CHAR(length=32), nullable=False), @@ -81,7 +81,7 @@ def downgrade() -> None: sa.Column('is_read_only', sa.BOOLEAN(), nullable=False), sa.Column('create_at', sa.DATETIME(), nullable=False), sa.Column('update_at', sa.DATETIME(), nullable=False), - sa.PrimaryKeyConstraint('id') + sa.PrimaryKeyConstraint('id', name=op.f('pk_component')) ) with op.batch_alter_table('component', schema=None) as batch_op: diff --git a/src/backend/langflow/alembic/versions/eb5866d51fd2_change_columns_to_be_nullable.py b/src/backend/langflow/alembic/versions/eb5866d51fd2_change_columns_to_be_nullable.py index 080602358..a92f3d7b2 100644 --- a/src/backend/langflow/alembic/versions/eb5866d51fd2_change_columns_to_be_nullable.py +++ b/src/backend/langflow/alembic/versions/eb5866d51fd2_change_columns_to_be_nullable.py @@ -57,7 +57,7 @@ def downgrade() -> None: sa.Column("is_read_only", sa.BOOLEAN(), nullable=False), sa.Column("create_at", sa.DATETIME(), nullable=False), sa.Column("update_at", sa.DATETIME(), nullable=False), - sa.PrimaryKeyConstraint("id"), + sa.PrimaryKeyConstraint("id", name="pk_component"), ) with op.batch_alter_table("component", schema=None) as batch_op: batch_op.create_index("ix_component_name", ["name"], unique=False) @@ -78,8 +78,8 @@ def downgrade() -> None: ["flow_id"], ["flow.id"], ), - sa.PrimaryKeyConstraint("id"), - sa.UniqueConstraint("id"), + sa.PrimaryKeyConstraint("id", name="pk_flowstyle"), + sa.UniqueConstraint("id", name="uq_flowstyle_id"), ) except Exception: pass