52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
|
from flask import render_template, redirect, url_for, flash
|
||
|
|
from flask_login import login_required, current_user
|
||
|
|
from flask_wtf import FlaskForm
|
||
|
|
from wtforms import StringField, validators
|
||
|
|
|
||
|
|
from app.config import EMAIL_DOMAIN
|
||
|
|
from app.dashboard.base import dashboard_bp
|
||
|
|
from app.extensions import db
|
||
|
|
from app.log import LOG
|
||
|
|
from app.models import GenEmail
|
||
|
|
from app.utils import convert_to_id
|
||
|
|
|
||
|
|
|
||
|
|
class CustomAliasForm(FlaskForm):
|
||
|
|
email = StringField("Email", validators=[validators.DataRequired()])
|
||
|
|
|
||
|
|
|
||
|
|
@dashboard_bp.route("/custom_alias", methods=["GET", "POST"])
|
||
|
|
@login_required
|
||
|
|
def custom_alias():
|
||
|
|
# check if user has the right to create custom alias
|
||
|
|
if not current_user.can_create_custom_email():
|
||
|
|
# notify admin
|
||
|
|
LOG.error("user %s tries to create custom alias", current_user)
|
||
|
|
flash("ony premium user can choose custom alias", "warning")
|
||
|
|
return redirect(url_for("dashboard.index"))
|
||
|
|
|
||
|
|
form = CustomAliasForm()
|
||
|
|
error = ""
|
||
|
|
|
||
|
|
if form.validate_on_submit():
|
||
|
|
email = form.email.data
|
||
|
|
email = convert_to_id(email)
|
||
|
|
if len(email) < 6:
|
||
|
|
error = "email must be at least 6 letters"
|
||
|
|
else:
|
||
|
|
full_email = f"{email}@{EMAIL_DOMAIN}"
|
||
|
|
# check if email already exists
|
||
|
|
if GenEmail.get_by(email=full_email):
|
||
|
|
error = "email already chosen, please choose another one"
|
||
|
|
else:
|
||
|
|
# create the new alias
|
||
|
|
LOG.d("create custom alias %s for user %s", full_email, current_user)
|
||
|
|
GenEmail.create(email=full_email, user_id=current_user.id)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
flash(f"Email alias {full_email} has been created", "success")
|
||
|
|
|
||
|
|
return redirect(url_for("dashboard.index"))
|
||
|
|
|
||
|
|
return render_template("dashboard/custom_alias.html", form=form, error=error)
|