diff --git a/app/models.py b/app/models.py index 0accd422..dfb95de8 100644 --- a/app/models.py +++ b/app/models.py @@ -95,6 +95,11 @@ class User(db.Model, ModelMixin, UserMixin): profile_picture_id = db.Column(db.ForeignKey(File.id), nullable=True) + otp_secret = db.Column(db.String(16), nullable=True) + enable_otp = db.Column( + db.Boolean, nullable=False, default=False, server_default="0" + ) + profile_picture = db.relationship(File) @classmethod diff --git a/migrations/versions/d4e4488a0032_.py b/migrations/versions/d4e4488a0032_.py new file mode 100644 index 00000000..ec988e55 --- /dev/null +++ b/migrations/versions/d4e4488a0032_.py @@ -0,0 +1,31 @@ +"""empty message + +Revision ID: d4e4488a0032 +Revises: 9e1b06b9df13 +Create Date: 2019-12-27 15:19:11.060497 + +""" +import sqlalchemy_utils +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision = 'd4e4488a0032' +down_revision = '9e1b06b9df13' +branch_labels = None +depends_on = None + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.add_column('users', sa.Column('enable_otp', sa.Boolean(), server_default='0', nullable=False)) + op.add_column('users', sa.Column('otp_secret', sa.String(length=16), nullable=True)) + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.drop_column('users', 'otp_secret') + op.drop_column('users', 'enable_otp') + # ### end Alembic commands ###