mirror of
https://github.com/StefBuwalda/cal_counter.git
synced 2025-10-30 11:19:59 +00:00
Stef
c7395b07d9
The logout route was relocated from the main app to the auth blueprint for better organization. The logout link in the base template was updated to reference the new route location.
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
from flask import Blueprint, request, render_template, redirect, url_for
|
|
from flask_login import current_user, login_user, logout_user
|
|
from forms import LoginForm, ChangePasswordForm
|
|
from models import User
|
|
from application.utils import default_return
|
|
from application import db
|
|
|
|
bp = Blueprint(
|
|
"auth",
|
|
__name__,
|
|
template_folder="templates",
|
|
)
|
|
|
|
|
|
@bp.route("/login", methods=["GET", "POST"])
|
|
def login():
|
|
if current_user.is_authenticated:
|
|
return default_return()
|
|
|
|
form = LoginForm()
|
|
if form.validate_on_submit():
|
|
user = User.query.filter_by(username=form.username.data).first()
|
|
if user and user.check_password(password=form.password.data):
|
|
# User found and password correct
|
|
next_page = request.args.get("next") # Get next page if given
|
|
login_user(user) # Log in the user
|
|
return default_return(next_page=next_page)
|
|
else:
|
|
pass
|
|
# invalid user
|
|
return render_template("login.html", form=form)
|
|
|
|
|
|
@bp.route("/change_password", methods=["GET", "POST"])
|
|
def change_password():
|
|
if not current_user.is_authenticated:
|
|
return redirect(url_for("auth.login"))
|
|
|
|
form = ChangePasswordForm()
|
|
if form.validate_on_submit():
|
|
cur_check = current_user.check_password(
|
|
password=form.current_password.data
|
|
)
|
|
eq_check = form.new_password.data == form.confirm_password.data
|
|
if cur_check and eq_check:
|
|
current_user.change_password(form.new_password.data)
|
|
current_user.set_pw_change(False)
|
|
db.session.commit()
|
|
return default_return()
|
|
return render_template("change_password.html", form=form)
|
|
|
|
|
|
@bp.route("/logout")
|
|
def logout():
|
|
if not current_user.is_authenticated:
|
|
return redirect(url_for("auth.login"))
|
|
|
|
logout_user()
|
|
return redirect(url_for("index"))
|