From ebadcfb175ffab027f00ed7540bd2322225c5e91 Mon Sep 17 00:00:00 2001
From: Son NK <nguyenkims@hotmail.com>
Date: Wed, 5 Feb 2020 15:45:29 +0700
Subject: [PATCH] add GenEmail.note column

---
 app/models.py                                 |  6 ++--
 .../versions/2020_020515_b9f849432543_.py     | 29 +++++++++++++++++++
 2 files changed, 33 insertions(+), 2 deletions(-)
 create mode 100644 migrations/versions/2020_020515_b9f849432543_.py

diff --git a/app/models.py b/app/models.py
index bc36729e..53efb870 100644
--- a/app/models.py
+++ b/app/models.py
@@ -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(
diff --git a/migrations/versions/2020_020515_b9f849432543_.py b/migrations/versions/2020_020515_b9f849432543_.py
new file mode 100644
index 00000000..735a7550
--- /dev/null
+++ b/migrations/versions/2020_020515_b9f849432543_.py
@@ -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 ###