From 80a6168e87e806513790bb2bcdd643f108f29fae Mon Sep 17 00:00:00 2001
From: Son NK <nguyenkims@hotmail.com>
Date: Sun, 23 Feb 2020 14:39:51 +0700
Subject: [PATCH] add convert_user_full_mailbox()

---
 shell.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/shell.py b/shell.py
index e594a133..1f12c28d 100644
--- a/shell.py
+++ b/shell.py
@@ -40,6 +40,23 @@ def send_safari_extension_newsletter():
         )
 
 
+def convert_user_full_mailbox(user):
+    # create a default mailbox
+    default_mb = Mailbox.get_by(user_id=user.id, email=user.email)
+    if not default_mb:
+        LOG.d("create default mailbox for user %s", user)
+        default_mb = Mailbox.create(user_id=user.id, email=user.email, verified=True)
+        db.session.commit()
+
+    # assign existing alias to this mailbox
+    for gen_email in GenEmail.query.filter_by(user_id=user.id):
+        gen_email.mailbox_id = default_mb.id
+
+    # finally set user to full_mailbox
+    user.full_mailbox = True
+    db.session.commit()
+
+
 app = create_app()
 
 with app.app_context():