wip

migrations/versions/79972ec5f79d_int.py

@@ -0,0 +1,46 @@
+"""int
+
+Revision ID: 79972ec5f79d
+Revises: 
+Create Date: 2023-06-22 15:02:20.292322
+
+"""
+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 = '79972ec5f79d'
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('hero',
+    sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('secret_name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_table('pet',
+    sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+    sa.Column('birthday', sa.DateTime(), nullable=True),
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.PrimaryKeyConstraint('id')
+    )
+    # ### 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_table('pet')
+    op.drop_table('hero')
+    # ### end Alembic commands ###