Add database migration scripts
This commit is contained in:
parent
98d5351d87
commit
9a697362cb
4 changed files with 58 additions and 11 deletions
43
src/backend/langflow/alembic/versions/1ef9c4f3765d_.py
Normal file
43
src/backend/langflow/alembic/versions/1ef9c4f3765d_.py
Normal file
|
|
@ -0,0 +1,43 @@
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
Revision ID: 1ef9c4f3765d
|
||||||
|
Revises: fd531f8868b1
|
||||||
|
Create Date: 2023-12-04 15:00:27.968998
|
||||||
|
|
||||||
|
"""
|
||||||
|
from typing import Sequence, Union
|
||||||
|
|
||||||
|
import sqlalchemy as sa
|
||||||
|
import sqlmodel
|
||||||
|
from alembic import op
|
||||||
|
|
||||||
|
# revision identifiers, used by Alembic.
|
||||||
|
revision: str = '1ef9c4f3765d'
|
||||||
|
down_revision: Union[str, None] = 'fd531f8868b1'
|
||||||
|
branch_labels: Union[str, Sequence[str], None] = None
|
||||||
|
depends_on: Union[str, Sequence[str], None] = None
|
||||||
|
|
||||||
|
|
||||||
|
def upgrade() -> None:
|
||||||
|
# ### commands auto generated by Alembic - please adjust! ###
|
||||||
|
try:
|
||||||
|
with op.batch_alter_table('apikey', schema=None) as batch_op:
|
||||||
|
batch_op.alter_column('name',
|
||||||
|
existing_type=sqlmodel.sql.sqltypes.AutoString(),
|
||||||
|
nullable=True)
|
||||||
|
except Exception as e:
|
||||||
|
pass
|
||||||
|
# ### end Alembic commands ###
|
||||||
|
|
||||||
|
|
||||||
|
def downgrade() -> None:
|
||||||
|
# ### commands auto generated by Alembic - please adjust! ###
|
||||||
|
try:
|
||||||
|
with op.batch_alter_table('apikey', schema=None) as batch_op:
|
||||||
|
batch_op.alter_column('name',
|
||||||
|
existing_type=sa.VARCHAR(),
|
||||||
|
nullable=False)
|
||||||
|
except Exception as e:
|
||||||
|
pass
|
||||||
|
# ### end Alembic commands ###
|
||||||
|
|
@ -7,9 +7,9 @@ Create Date: 2023-08-27 19:49:02.681355
|
||||||
"""
|
"""
|
||||||
from typing import Sequence, Union
|
from typing import Sequence, Union
|
||||||
|
|
||||||
from alembic import op
|
|
||||||
import sqlalchemy as sa
|
import sqlalchemy as sa
|
||||||
import sqlmodel
|
import sqlmodel
|
||||||
|
from alembic import op
|
||||||
from sqlalchemy.engine.reflection import Inspector
|
from sqlalchemy.engine.reflection import Inspector
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
# revision identifiers, used by Alembic.
|
||||||
|
|
|
||||||
|
|
@ -20,12 +20,15 @@ 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! ###
|
||||||
|
try:
|
||||||
with op.batch_alter_table('component', schema=None) as batch_op:
|
with op.batch_alter_table('component', schema=None) as batch_op:
|
||||||
batch_op.drop_index('ix_component_frontend_node_id')
|
batch_op.drop_index('ix_component_frontend_node_id')
|
||||||
batch_op.drop_index('ix_component_name')
|
batch_op.drop_index('ix_component_name')
|
||||||
|
|
||||||
op.drop_table('component')
|
op.drop_table('component')
|
||||||
op.drop_table('flowstyle')
|
op.drop_table('flowstyle')
|
||||||
|
except Exception as e:
|
||||||
|
print(e)
|
||||||
|
pass
|
||||||
with op.batch_alter_table('apikey', schema=None) as batch_op:
|
with op.batch_alter_table('apikey', schema=None) as batch_op:
|
||||||
batch_op.alter_column('name',
|
batch_op.alter_column('name',
|
||||||
existing_type=sa.VARCHAR(),
|
existing_type=sa.VARCHAR(),
|
||||||
|
|
|
||||||
|
|
@ -7,10 +7,9 @@ Create Date: 2023-10-04 10:18:25.640458
|
||||||
"""
|
"""
|
||||||
from typing import Sequence, Union
|
from typing import Sequence, Union
|
||||||
|
|
||||||
from alembic import op
|
|
||||||
import sqlalchemy as sa
|
import sqlalchemy as sa
|
||||||
|
from alembic import op
|
||||||
from sqlalchemy import exc
|
from sqlalchemy import exc
|
||||||
import sqlmodel # noqa: F401
|
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
# revision identifiers, used by Alembic.
|
||||||
revision: str = "eb5866d51fd2"
|
revision: str = "eb5866d51fd2"
|
||||||
|
|
@ -28,14 +27,16 @@ def upgrade() -> None:
|
||||||
batch_op.drop_index("ix_component_frontend_node_id")
|
batch_op.drop_index("ix_component_frontend_node_id")
|
||||||
batch_op.drop_index("ix_component_name")
|
batch_op.drop_index("ix_component_name")
|
||||||
except exc.SQLAlchemyError:
|
except exc.SQLAlchemyError:
|
||||||
connection.execute("ROLLBACK")
|
# connection.execute(text("ROLLBACK"))
|
||||||
|
pass
|
||||||
except Exception:
|
except Exception:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
try:
|
try:
|
||||||
op.drop_table("component")
|
op.drop_table("component")
|
||||||
except exc.SQLAlchemyError:
|
except exc.SQLAlchemyError:
|
||||||
connection.execute("ROLLBACK")
|
# connection.execute(text("ROLLBACK"))
|
||||||
|
pass
|
||||||
except Exception:
|
except Exception:
|
||||||
pass
|
pass
|
||||||
# ### end Alembic commands ###
|
# ### end Alembic commands ###
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue