浏览代码

Allow the same mailbox used by different user

Son NK 5 年之前
父节点
当前提交
c85ea1538e
共有 2 个文件被更改,包括 3 次插入3 次删除
  1. 2 2
      app/dashboard/views/custom_alias.py
  2. 1 1
      app/email_utils.py

+ 2 - 2
app/dashboard/views/custom_alias.py

@@ -64,7 +64,7 @@ def custom_alias():
 
         # check if mailbox is not tempered with
         if mailbox_email != current_user.email:
-            mailbox = Mailbox.get_by(email=mailbox_email)
+            mailbox = Mailbox.get_by(email=mailbox_email, user_id=current_user.id)
             if not mailbox or mailbox.user_id != current_user.id:
                 flash("Something went wrong, please retry", "warning")
                 return redirect(url_for("dashboard.custom_alias"))
@@ -91,7 +91,7 @@ def custom_alias():
                     "warning",
                 )
             else:
-                mailbox = Mailbox.get_by(email=mailbox_email)
+                mailbox = Mailbox.get_by(email=mailbox_email, user_id=current_user.id)
 
                 alias = Alias.create(
                     user_id=current_user.id,

+ 1 - 1
app/email_utils.py

@@ -376,7 +376,7 @@ def email_already_used(email: str) -> bool:
 
 
 def mailbox_already_used(email: str, user) -> bool:
-    if Mailbox.get_by(email=email):
+    if Mailbox.get_by(email=email, user_id=user.id):
         return True
 
     # support the case user wants to re-add their real email as mailbox