From 20d6fbae485b0b384964cabf95f1b42345d76e0f Mon Sep 17 00:00:00 2001 From: Stefan Auditor Date: Sun, 15 Apr 2018 11:23:58 +0200 Subject: [PATCH] Add enabled flag to user model --- core/admin/mailu/models.py | 1 + .../migrations/versions/49d77a93118e_.py | 24 +++++++++++++++++++ 2 files changed, 25 insertions(+) create mode 100644 core/admin/migrations/versions/49d77a93118e_.py diff --git a/core/admin/mailu/models.py b/core/admin/mailu/models.py index 63817cdc..8e7939af 100644 --- a/core/admin/mailu/models.py +++ b/core/admin/mailu/models.py @@ -225,6 +225,7 @@ class User(Base, Email): password = db.Column(db.String(255), nullable=False) quota_bytes = db.Column(db.Integer(), nullable=False, default=10**9) global_admin = db.Column(db.Boolean(), nullable=False, default=False) + enabled = db.Column(db.Boolean(), nullable=False, default=True) # Features enable_imap = db.Column(db.Boolean(), nullable=False, default=True) diff --git a/core/admin/migrations/versions/49d77a93118e_.py b/core/admin/migrations/versions/49d77a93118e_.py new file mode 100644 index 00000000..98dd9e34 --- /dev/null +++ b/core/admin/migrations/versions/49d77a93118e_.py @@ -0,0 +1,24 @@ +""" Add enabled flag to user model + +Revision ID: 49d77a93118e +Revises: 423155f8fc15 +Create Date: 2018-04-15 11:17:32.306088 + +""" + +# revision identifiers, used by Alembic. +revision = '49d77a93118e' +down_revision = '423155f8fc15' + +from alembic import op +import sqlalchemy as sa + + +def upgrade(): + with op.batch_alter_table('user') as batch: + batch.add_column(sa.Column('enabled', sa.Boolean(), nullable=False, server_default=sa.sql.expression.true())) + + +def downgrade(): + with op.batch_alter_table('user') as batch: + batch.drop_column('user', 'enabled')