add User.otp_secret, enable_otp column

This commit is contained in:
Son NK 2019-12-27 15:20:10 +01:00 committed by Son Nguyen Kim
parent efd887917f
commit 80facfe122
2 changed files with 36 additions and 0 deletions

View File

@ -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

View File

@ -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 ###