wip

migrations/versions/c79214cdc7b3_add_hero_size.py

@@ -0,0 +1,34 @@
+"""add hero.size
+
+Revision ID: c79214cdc7b3
+Revises: 3555f61aaa79
+Create Date: 2023-06-28 11:39:02.606001
+
+"""
+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
+
+
+
+# revision identifiers, used by Alembic.
+revision = 'c79214cdc7b3'
+down_revision = '3555f61aaa79'
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('hero', sa.Column('size', sa.Integer(), nullable=True))
+    # ### end Alembic commands ###
+    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:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('hero', 'size')
+    # ### end Alembic commands ###