From 91d1b04d92b089d6bda0911927ff62b14428b24b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 17 Jan 2024 07:36:05 -0300 Subject: [PATCH] Add indexes to tables apikey, flow, and user --- .../alembic/versions/0b8757876a7c_.py | 69 ++++++++++++------- 1 file changed, 44 insertions(+), 25 deletions(-) diff --git a/src/backend/langflow/alembic/versions/0b8757876a7c_.py b/src/backend/langflow/alembic/versions/0b8757876a7c_.py index 6d6c22575..61b769694 100644 --- a/src/backend/langflow/alembic/versions/0b8757876a7c_.py +++ b/src/backend/langflow/alembic/versions/0b8757876a7c_.py @@ -18,35 +18,54 @@ depends_on: Union[str, Sequence[str], None] = None def upgrade() -> None: # ### commands auto generated by Alembic - please adjust! ### - with op.batch_alter_table('apikey', schema=None) as batch_op: - batch_op.create_index(batch_op.f('ix_apikey_api_key'), ['api_key'], unique=True) - batch_op.create_index(batch_op.f('ix_apikey_name'), ['name'], unique=False) - batch_op.create_index(batch_op.f('ix_apikey_user_id'), ['user_id'], unique=False) - - with op.batch_alter_table('flow', schema=None) as batch_op: - batch_op.create_index(batch_op.f('ix_flow_description'), ['description'], unique=False) - batch_op.create_index(batch_op.f('ix_flow_name'), ['name'], unique=False) - batch_op.create_index(batch_op.f('ix_flow_user_id'), ['user_id'], unique=False) - - with op.batch_alter_table('user', schema=None) as batch_op: - batch_op.create_index(batch_op.f('ix_user_username'), ['username'], unique=True) + try: + with op.batch_alter_table('apikey', schema=None) as batch_op: + batch_op.create_index(batch_op.f('ix_apikey_api_key'), ['api_key'], unique=True) + batch_op.create_index(batch_op.f('ix_apikey_name'), ['name'], unique=False) + batch_op.create_index(batch_op.f('ix_apikey_user_id'), ['user_id'], unique=False) + except Exception as e: + print(e) + pass + try: + with op.batch_alter_table('flow', schema=None) as batch_op: + batch_op.create_index(batch_op.f('ix_flow_description'), ['description'], unique=False) + batch_op.create_index(batch_op.f('ix_flow_name'), ['name'], unique=False) + batch_op.create_index(batch_op.f('ix_flow_user_id'), ['user_id'], unique=False) + except Exception as e: + print(e) + pass + try: + with op.batch_alter_table('user', schema=None) as batch_op: + batch_op.create_index(batch_op.f('ix_user_username'), ['username'], unique=True) + except Exception as e: + print(e) + pass # ### end Alembic commands ### def downgrade() -> None: # ### commands auto generated by Alembic - please adjust! ### - with op.batch_alter_table('user', schema=None) as batch_op: - batch_op.drop_index(batch_op.f('ix_user_username')) - - with op.batch_alter_table('flow', schema=None) as batch_op: - batch_op.drop_index(batch_op.f('ix_flow_user_id')) - batch_op.drop_index(batch_op.f('ix_flow_name')) - batch_op.drop_index(batch_op.f('ix_flow_description')) - - with op.batch_alter_table('apikey', schema=None) as batch_op: - batch_op.drop_index(batch_op.f('ix_apikey_user_id')) - batch_op.drop_index(batch_op.f('ix_apikey_name')) - batch_op.drop_index(batch_op.f('ix_apikey_api_key')) - + try: + with op.batch_alter_table('user', schema=None) as batch_op: + batch_op.drop_index(batch_op.f('ix_user_username')) + except Exception as e: + print(e) + pass + try: + with op.batch_alter_table('flow', schema=None) as batch_op: + batch_op.drop_index(batch_op.f('ix_flow_user_id')) + batch_op.drop_index(batch_op.f('ix_flow_name')) + batch_op.drop_index(batch_op.f('ix_flow_description')) + except Exception as e: + print(e) + pass + try: + with op.batch_alter_table('apikey', schema=None) as batch_op: + batch_op.drop_index(batch_op.f('ix_apikey_user_id')) + batch_op.drop_index(batch_op.f('ix_apikey_name')) + batch_op.drop_index(batch_op.f('ix_apikey_api_key')) + except Exception as e: + print(e) + pass # ### end Alembic commands ### \ No newline at end of file