This commit is contained in:
Waylon Walker 2023-06-22 16:02:03 -05:00
parent 8480cd3d24
commit e86e432102
No known key found for this signature in database
GPG key ID: 66E2BF2B4190EFE4
11 changed files with 352 additions and 69 deletions

View file

@ -8,6 +8,9 @@ Create Date: ${create_date}
from alembic import op
import sqlalchemy as sa
import sqlmodel
from learn_sql_model.er_diagram import generate_er_diagram, generate_er_markdown
from learn_sql_model.config import get_config
${imports if imports else ""}
# revision identifiers, used by Alembic.
@ -19,6 +22,8 @@ depends_on = ${repr(depends_on)}
def upgrade() -> None:
${upgrades if upgrades else "pass"}
generate_er_diagram(f'migrations/versions/{revision}_er_diagram.png')
generate_er_markdown(f'migrations/versions/{revision}_er_diagram.md', f'migrations/versions/er_diagram_{revision}.png')
def downgrade() -> None: