only user with can_use_multiple_mailbox enabled can use this feature

app/dashboard/views/mailbox.py

@@ -27,6 +27,10 @@ class NewMailboxForm(FlaskForm):
 @dashboard_bp.route("/mailbox", methods=["GET", "POST"])
 @login_required
 def mailbox_route():
+    if not current_user.can_use_multiple_mailbox:
+        flash("You don't have access to this page, redirect to home page", "warning")
+        return redirect(url_for("dashboard.index"))
+
     mailboxes = Mailbox.query.filter_by(user_id=current_user.id).all()
 
     new_mailbox_form = NewMailboxForm()
@@ -48,9 +52,10 @@ def mailbox_route():
             return redirect(url_for("dashboard.mailbox_route"))
 
         elif request.form.get("form-name") == "create":
-            if not current_user.is_premium():
-                flash("Only premium plan can add additional mailbox", "warning")
-                return redirect(url_for("dashboard.mailbox_route"))
+            # todo: only premium user can add additional mailbox?
+            # if not current_user.is_premium():
+            #     flash("Only premium plan can add additional mailbox", "warning")
+            #     return redirect(url_for("dashboard.mailbox_route"))
 
             if new_mailbox_form.validate():
                 mailbox_email = new_mailbox_form.email.data.lower()

templates/menu.html

@@ -38,6 +38,7 @@
     </a>
   </li>
 
+  {% if current_user.can_use_multiple_mailbox %}
   <li class="nav-item">
     <a href="{{ url_for('dashboard.mailbox_route') }}"
        class="nav-link {{ 'active' if active_page == 'mailbox' }}">
@@ -45,6 +46,7 @@
       <span class="badge badge-info" style="font-size: .5rem; top: 5px">Beta</span>
     </a>
   </li>
+  {% endif %}
 
   <!--
   <li class="nav-item">