24 lines
935 B
Python
24 lines
935 B
Python
from flask import request, Blueprint, render_template, session, current_app, redirect, url_for
|
|
from .auth_model import auth_model
|
|
|
|
auth_bp = Blueprint('auth_bp', __name__, template_folder='templates')
|
|
|
|
@auth_bp.route('/', methods=['GET', 'POST'])
|
|
def auth():
|
|
if request.method == 'GET':
|
|
return render_template('auth.html')
|
|
else:
|
|
data = request.form.to_dict()
|
|
data['table'] = 'internal_users' if 'internal' in data else 'external_users'
|
|
auth_data = auth_model(data)
|
|
if auth_data.status:
|
|
session.update({
|
|
'login': auth_data.result[0]['login'],
|
|
'role': auth_data.result[0]['user_role'],
|
|
'db_config': current_app.config['db_config'],
|
|
'permanent': True
|
|
})
|
|
return redirect(url_for('index'))
|
|
else:
|
|
return render_template('error.html', error_message=auth_data.error_message)
|