diff --git a/data/forest/migrations/versions/c91b20e3d138_.py b/data/forest/migrations/versions/c91b20e3d138_.py new file mode 100644 index 0000000..aa8a11b --- /dev/null +++ b/data/forest/migrations/versions/c91b20e3d138_.py @@ -0,0 +1,32 @@ +"""empty message + +Revision ID: c91b20e3d138 +Revises: +Create Date: 2024-04-08 15:29:03.170183 + +""" +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision = 'c91b20e3d138' +down_revision = None +branch_labels = None +depends_on = None + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + with op.batch_alter_table('users', schema=None) as batch_op: + batch_op.add_column(sa.Column('setup', sa.Boolean(), nullable=True)) + + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + with op.batch_alter_table('users', schema=None) as batch_op: + batch_op.drop_column('setup') + + # ### end Alembic commands ### diff --git a/src/forest/models.py b/src/forest/models.py index c4ac321..5d94fbb 100644 --- a/src/forest/models.py +++ b/src/forest/models.py @@ -61,6 +61,7 @@ class User(db.Model, UserMixin): email = db.Column(db.String(128), unique=True, nullable=False) active = db.Column(db.Boolean(), default=True, nullable=False) confirmed = db.Column(db.Boolean, default=False) + setup = db.Column(db.Boolean, default=False) role_id = db.Column(db.ForeignKey('roles.id')) #FK password_hash = db.Column(db.String)