浏览代码

Handle the case where a deleted mailbox verification link is clicked

Son NK 5 年之前
父节点
当前提交
e9d03d1d4b
共有 1 个文件被更改,包括 4 次插入0 次删除
  1. 4 0
      app/dashboard/views/mailbox.py

+ 4 - 0
app/dashboard/views/mailbox.py

@@ -150,6 +150,10 @@ def mailbox_verify():
         return redirect(url_for("dashboard.mailbox_route"))
         return redirect(url_for("dashboard.mailbox_route"))
     else:
     else:
         mailbox = Mailbox.get(r_id)
         mailbox = Mailbox.get(r_id)
+        if not mailbox:
+            flash("Invalid link", "error")
+            return redirect(url_for("dashboard.mailbox_route"))
+
         mailbox.verified = True
         mailbox.verified = True
         db.session.commit()
         db.session.commit()