Add database migration scripts

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-04 15:35:29 -03:00
commit 9a697362cb
4 changed files with 58 additions and 11 deletions

View 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 ###

View file

@ -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.

View file

@ -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(),

View file

@ -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 ###