52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
from flask import request, render_template, redirect, url_for
|
|
from flask_login import login_user
|
|
from flask_wtf import FlaskForm
|
|
from wtforms import StringField, validators
|
|
|
|
from app.auth.base import auth_bp
|
|
from app.log import LOG
|
|
from app.models import User
|
|
|
|
|
|
class LoginForm(FlaskForm):
|
|
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 form.validate_on_submit():
|
|
user = User.filter_by(email=form.email.data).first()
|
|
|
|
if not user:
|
|
return render_template(
|
|
"auth/login.html", form=form, error="Email not exist in our system"
|
|
)
|
|
|
|
if not user.check_password(form.password.data):
|
|
return render_template("auth/login.html", form=form, error="Wrong password")
|
|
|
|
if not user.activated:
|
|
return render_template(
|
|
"auth/login.html",
|
|
form=form,
|
|
show_resend_activation=True,
|
|
error="Please check your inbox for the activation email. You can also have this email re-sent",
|
|
)
|
|
|
|
LOG.debug("log user %s in", user)
|
|
login_user(user)
|
|
|
|
# User comes to login page from another page
|
|
if "next" in request.args:
|
|
next_url = request.args.get("next")
|
|
LOG.debug("redirect user to %s", next_url)
|
|
return redirect(next_url)
|
|
else:
|
|
LOG.debug("redirect user to dashboard")
|
|
return redirect(url_for("dashboard.index"))
|
|
|
|
return render_template("auth/login.html", form=form)
|