make sure that user cannot use any suffix

This commit is contained in:
Son NK 2019-12-31 17:11:42 +01:00
parent 8202be7bae
commit 652e623111
2 changed files with 13 additions and 1 deletions

View File

@ -6,7 +6,7 @@ from app.dashboard.base import dashboard_bp
from app.extensions import db from app.extensions import db
from app.log import LOG from app.log import LOG
from app.models import GenEmail, DeletedAlias, CustomDomain from app.models import GenEmail, DeletedAlias, CustomDomain
from app.utils import convert_to_id, random_word from app.utils import convert_to_id, random_word, word_exist
@dashboard_bp.route("/custom_alias", methods=["GET", "POST"]) @dashboard_bp.route("/custom_alias", methods=["GET", "POST"])
@ -27,6 +27,14 @@ def custom_alias():
email_prefix = convert_to_id(email_prefix) email_prefix = convert_to_id(email_prefix)
email_suffix = request.form.get("email-suffix") email_suffix = request.form.get("email-suffix")
# verify email_suffix
if not word_exist(email_suffix):
flash(
"nice try :). The suffix is there so no one can take all the *nice* aliases though",
"warning",
)
return redirect(url_for("dashboard.custom_alias"))
if not email_prefix: if not email_prefix:
error = "alias prefix cannot be empty" error = "alias prefix cannot be empty"
else: else:

View File

@ -16,6 +16,10 @@ def random_word():
return random.choice(_words) return random.choice(_words)
def word_exist(word):
return word in _words
def random_words(): def random_words():
"""Generate a random words. Used to generate user-facing string, for ex email addresses""" """Generate a random words. Used to generate user-facing string, for ex email addresses"""
nb_words = random.randint(2, 3) nb_words = random.randint(2, 3)