8832d7a1b5fb_refine_databasemodel.py

"""refine databasemodel

Revision ID: 8832d7a1b5fb
Revises: 1dc4ac9953cb
Create Date: 2022-03-18 09:43:55.350402

"""
import os

from alembic import op

here = os.path.dirname(os.path.realpath(__file__))

# revision identifiers, used by Alembic.
revision = "8832d7a1b5fb"
down_revision = "959c5af94dc2"
branch_labels = None
depends_on = None

revision_dir = f"{here}/{revision}"


# idea from https://github.com/tbobm/alembic-sequeled
def process_migration(script_name: str):
    filename = f"{revision_dir}/{script_name}.sql"

    query = "\n".join(open(filename))
    if len(query) > 0:
        op.execute(query)


def upgrade():
    process_migration("upgrade")


def downgrade():
    process_migration("downgrade")