app-MAIL-temp/app/auth/views/login.py

37 lines
1.1 KiB
Python

from flask import request, flash, render_template, redirect, url_for
from flask_login import login_user
from wtforms import Form, StringField, validators
from app.auth.base import auth_bp
from app.log import LOG
from app.models import User
class LoginForm(Form):
email = StringField("Email", validators=[validators.DataRequired()])
password = StringField("Password", validators=[validators.DataRequired()])
@auth_bp.route("/login", methods=["GET", "POST"])
def login():
form = LoginForm(request.form)
if request.method == "POST":
if form.validate():
user = User.query.filter_by(email=form.email.data).first()
if not user:
flash("No such email", "warning")
return render_template("auth/login.html", form=form)
if not user.check_password(form.password.data):
flash("Wrong password", "warning")
return render_template("auth/login.html", form=form)
LOG.debug("log user %s in", user)
login_user(user)
return redirect(url_for("dashboard.index"))
return render_template("auth/login.html", form=form)