PCv5/app/routes/login.py

34 lines
1.3 KiB
Python

from flask import redirect, url_for, request, flash
from flask_login import login_user, logout_user, login_required, current_user
from app import app
from app.forms import LoginForm
from app.models.users import Member
from app.utils.render import render
from app.routes import * # For open redirections
@app.route('/login', methods=['GET', 'POST'])
def login():
form = LoginForm()
if form.validate_on_submit():
member = Member.query.filter_by(name=form.username.data).first()
if member is None or not member.check_password(form.password.data):
flash('Pseudo ou mot de passe invalide', 'error')
return redirect(request.referrer)
login_user(member, remember=form.remember_me.data)
# TODO: petit mot personnalisé pour l'utilisateur ?
flash('Connexion réussie', 'info')
r = request.args.get('next')
if r:
return redirect(r)
# TODO: faire en sorte qu'on soit redirigé vers la page actuelle si on se loggue dans la sidebar
# TODO: mais pas rediriger vers login si on est déjà loggué
return redirect(url_for('index'))
return render('login.html', form=form)
@app.route('/logout')
@login_required
def logout():
logout_user()
flash('Déconnexion réussie', 'info')
return redirect(url_for('index'))