- add __tablename__ for all models - use sa and orm instead of db - rollback all changes in tests - remove session in @app.teardown_appcontext
35 lines
918 B
Python
35 lines
918 B
Python
from flask import request, flash, render_template, redirect, url_for
|
|
from flask_login import login_user
|
|
|
|
from app.auth.base import auth_bp
|
|
from app.db import Session
|
|
from app.models import EmailChange
|
|
|
|
|
|
@auth_bp.route("/change_email", methods=["GET", "POST"])
|
|
def change_email():
|
|
code = request.args.get("code")
|
|
|
|
email_change: EmailChange = EmailChange.get_by(code=code)
|
|
|
|
if not email_change:
|
|
return render_template("auth/change_email.html")
|
|
|
|
if email_change.is_expired():
|
|
# delete the expired email
|
|
EmailChange.delete(email_change.id)
|
|
Session.commit()
|
|
return render_template("auth/change_email.html")
|
|
|
|
user = email_change.user
|
|
user.email = email_change.new_email
|
|
|
|
EmailChange.delete(email_change.id)
|
|
Session.commit()
|
|
|
|
flash("Your new email has been updated", "success")
|
|
|
|
login_user(user)
|
|
|
|
return redirect(url_for("dashboard.index"))
|