add alembic migrations

This commit is contained in:
Jakub Miazek
2022-06-04 15:15:04 +02:00
parent bffa7db7d5
commit 8ef59168d4
7 changed files with 45 additions and 65 deletions

View File

@@ -5,7 +5,6 @@ import sys
from alembic import context
from sqlalchemy.ext.asyncio import create_async_engine
parent_dir = os.path.abspath(os.path.join(os.getcwd()))
sys.path.append(parent_dir)
@@ -14,27 +13,11 @@ from the_app.models.base import Base as app_base
target_metadata = app_base.metadata
def run_migrations_offline():
"""Run migrations in 'offline' mode.
This configures the context with just a URL
and not an Engine, though an Engine is acceptable
here as well. By skipping the Engine creation
we don't even need a DBAPI to be available.
Calls to context.execute() here emit the given string to the
script output.
"""
url = f"postgresql+asyncpg://user:secret@db:5432/devdb"
context.configure(url=url, target_metadata=target_metadata, literal_binds=True)
with context.begin_transaction():
context.run_migrations()
def do_run_migrations(connection):
context.configure(connection=connection, target_metadata=target_metadata)
context.configure(connection=connection,
target_metadata=target_metadata,
include_schemas=True,
version_table_schema=target_metadata.schema)
with context.begin_transaction():
context.run_migrations()
@@ -54,7 +37,4 @@ async def run_migrations_online():
await connection.run_sync(do_run_migrations)
if context.is_offline_mode():
run_migrations_offline()
else:
asyncio.run(run_migrations_online())
asyncio.run(run_migrations_online())

View File

@@ -1,8 +1,8 @@
"""init tables
"""initialize happy_cog and shakespeare tables
Revision ID: 421df2c2140f
Revision ID: 957232a5b7ee
Revises:
Create Date: 2022-05-22 13:02:56.908424
Create Date: 2022-06-04 13:02:45.839649
"""
from alembic import op
@@ -10,7 +10,7 @@ import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic.
revision = '421df2c2140f'
revision = '957232a5b7ee'
down_revision = None
branch_labels = None
depends_on = None
@@ -18,6 +18,24 @@ depends_on = None
def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.create_table('nonsense',
sa.Column('id', postgresql.UUID(as_uuid=True), autoincrement=True, nullable=True),
sa.Column('name', sa.String(), nullable=False),
sa.Column('description', sa.String(), nullable=False),
sa.PrimaryKeyConstraint('name'),
sa.UniqueConstraint('id'),
sa.UniqueConstraint('name'),
schema='happy_hog'
)
op.create_table('stuff',
sa.Column('id', postgresql.UUID(as_uuid=True), autoincrement=True, nullable=True),
sa.Column('name', sa.String(), nullable=False),
sa.Column('description', sa.String(), nullable=False),
sa.PrimaryKeyConstraint('name'),
sa.UniqueConstraint('id'),
sa.UniqueConstraint('name'),
schema='happy_hog'
)
op.create_table('character',
sa.Column('id', sa.String(length=32), nullable=False),
sa.Column('name', sa.String(length=64), nullable=False),
@@ -49,24 +67,6 @@ def upgrade():
sa.PrimaryKeyConstraint('id', name='work_pkey'),
schema='shakespeare'
)
op.create_table('nonsense',
sa.Column('id', postgresql.UUID(as_uuid=True), autoincrement=True, nullable=True),
sa.Column('name', sa.String(), nullable=False),
sa.Column('description', sa.String(), nullable=False),
sa.PrimaryKeyConstraint('name'),
sa.UniqueConstraint('id'),
sa.UniqueConstraint('name'),
schema='the_others'
)
op.create_table('stuff',
sa.Column('id', postgresql.UUID(as_uuid=True), autoincrement=True, nullable=True),
sa.Column('name', sa.String(), nullable=False),
sa.Column('description', sa.String(), nullable=False),
sa.PrimaryKeyConstraint('name'),
sa.UniqueConstraint('id'),
sa.UniqueConstraint('name'),
schema='the_others'
)
op.create_table('chapter',
sa.Column('id', sa.Integer(), nullable=False),
sa.Column('work_id', sa.String(length=32), nullable=False),
@@ -113,9 +113,9 @@ def downgrade():
op.drop_table('paragraph', schema='shakespeare')
op.drop_table('character_work', schema='shakespeare')
op.drop_table('chapter', schema='shakespeare')
op.drop_table('stuff', schema='the_others')
op.drop_table('nonsense', schema='the_others')
op.drop_table('work', schema='shakespeare')
op.drop_table('wordform', schema='shakespeare')
op.drop_table('character', schema='shakespeare')
op.drop_table('stuff', schema='happy_hog')
op.drop_table('nonsense', schema='happy_hog')
# ### end Alembic commands ###