from flask import render_template, request, redirect, url_for, flash
from flask_login import login_user, logout_user
from app.models import User

from . import bp


@bp.route('/login')
def login():
    return render_template('login.html', title="Login")


@bp.route('/login', methods=['POST'])
def login_post():
    user_login = request.form.get('login')
    user_password = request.form.get('password')
    user_remember = request.form.get('remember')
    user = User.query.filter_by(login=user_login).one_or_none()
    print("user login: {}, user password: {}".format(user.login, user.password))
    if user and user.password == user_password:
        # login code goes here
        login_user(user)
        flash("Connection Réussie !")
        return redirect(url_for('main.index'))
    else:
        flash("Mauvais login ou mot de passe.")
        return redirect(url_for('auth.login'))


@bp.route('/logout')
def logout():
    logout_user()
    return redirect(url_for('main.index'))