48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
from datetime import datetime as dt, timezone as tz
|
|
from flask import request, redirect, url_for, flash, render_template, Blueprint, jsonify
|
|
from ..models.user import User, Tenant
|
|
from ..extensions import db, bcrypt
|
|
from .auth_forms import LoginForm
|
|
from flask_jwt_extended import (create_access_token, create_refresh_token, set_access_cookies, set_refresh_cookies,
|
|
unset_jwt_cookies)
|
|
|
|
auth_bp = Blueprint('auth_bp', __name__)
|
|
|
|
|
|
@auth_bp.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
if request.method == 'POST':
|
|
email = request.form.get('email')
|
|
password = request.form.get('password')
|
|
remember_me = True if request.form.get('remember_me') else False
|
|
|
|
user = User.query.filter_by(email=email).first()
|
|
if user:
|
|
if bcrypt.check_password_hash(user.password, password):
|
|
response = jsonify({'msg': 'Login Successful'})
|
|
flash('Logged in successfully!', category='success')
|
|
access_token = create_access_token(
|
|
identity=user.id,
|
|
additional_claims={'tenant': user.tenant_id})
|
|
refresh_token = create_refresh_token(
|
|
identity=user.id,
|
|
additional_claims={'tenant': user.tenant_id})
|
|
set_access_cookies(response, access_token)
|
|
set_refresh_cookies(response, refresh_token)
|
|
|
|
return redirect(url_for('user_bp.user'))
|
|
else:
|
|
flash('Incorrect password, try again.', category='error')
|
|
else:
|
|
flash('Email does not exist.', category='error')
|
|
|
|
form = LoginForm()
|
|
return render_template('login.html', form=form)
|
|
|
|
|
|
@auth_bp.route('/logout', methods=['POST'])
|
|
def logout():
|
|
response = jsonify({'msg': 'Logout Successful'})
|
|
unset_jwt_cookies(response)
|
|
return redirect(url_for('/'))
|