Files
WebTech/application/auth/views.py

40 lines
1.2 KiB
Python

from flask import Blueprint, render_template, redirect, url_for
# from application import db
from application.auth.models import User
from application.auth.forms import LoginForm
from flask_login import login_required, login_user, logout_user # type: ignore
from werkzeug.security import check_password_hash
auth_blueprint = Blueprint("auth", __name__, template_folder="templates")
# Routes
@auth_blueprint.route("/login", methods=["GET", "POST"])
def login():
login_form = LoginForm()
feedback = None
if login_form.validate_on_submit(): # type: ignore
username = login_form.username.data
password = login_form.password.data
user = User.query.filter_by(username=username).first() # type: ignore
if user and check_password_hash(
user.password, password # type: ignore
):
login_user(user) # type: ignore
return redirect("/")
else:
feedback = "Username or password is incorrect"
return render_template("login.html", form=login_form, feedback=feedback)
@auth_blueprint.route("/logout")
@login_required
def logout():
logout_user()
return redirect(url_for("index"))