add GenEmail.note column

This commit is contained in:
Son NK 2020-02-05 15:45:29 +07:00
parent ec248dcae6
commit ebadcfb175
2 changed files with 33 additions and 2 deletions

View File

@ -476,10 +476,12 @@ class GenEmail(db.Model, ModelMixin):
db.ForeignKey("directory.id", ondelete="cascade"), nullable=True
)
note = db.Column(db.Text, default=None, nullable=True)
user = db.relationship(User)
@classmethod
def create_new(cls, user_id, prefix):
def create_new(cls, user_id, prefix, note=None):
if not prefix:
raise Exception("alias prefix cannot be empty")
@ -491,7 +493,7 @@ class GenEmail(db.Model, ModelMixin):
if not cls.get_by(email=email):
break
return GenEmail.create(user_id=user_id, email=email)
return GenEmail.create(user_id=user_id, email=email, note=note)
@classmethod
def create_new_random(

View File

@ -0,0 +1,29 @@
"""empty message
Revision ID: b9f849432543
Revises: 9c976df9b9c4
Create Date: 2020-02-05 15:16:16.912369
"""
import sqlalchemy_utils
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision = 'b9f849432543'
down_revision = '9c976df9b9c4'
branch_labels = None
depends_on = None
def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.add_column('gen_email', sa.Column('note', sa.Text(), nullable=True))
# ### end Alembic commands ###
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.drop_column('gen_email', 'note')
# ### end Alembic commands ###