mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 02:42:26 -05:00
48b7a7dc37
Still need to figure out how to override the load_user method, currently it is not working.
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
from datetime import datetime
|
|
|
|
from flask import render_template, redirect, jsonify
|
|
from flask_login import login_user
|
|
|
|
from app.main import main
|
|
from app.main.forms import LoginForm
|
|
from app.main.dao import users_dao
|
|
from app.models import Users
|
|
from app.main.encryption import encrypt
|
|
|
|
|
|
@main.route("/sign-in", methods=(['GET']))
|
|
def render_sign_in():
|
|
return render_template('signin.html', form=LoginForm())
|
|
|
|
|
|
@main.route('/sign-in', methods=(['POST']))
|
|
def process_sign_in():
|
|
form = LoginForm()
|
|
if form.validate_on_submit():
|
|
user = users_dao.get_user_by_email(form.email_address.data)
|
|
if user is None:
|
|
return jsonify(authorization=False), 404
|
|
if user.password == encrypt(form.password.data):
|
|
login_user(user)
|
|
else:
|
|
return jsonify(authorization=False), 404
|
|
else:
|
|
return jsonify(form.errors), 404
|
|
return redirect('/two-factor')
|
|
|
|
|
|
@main.route('/temp-create-users', methods=(['GET']))
|
|
def render_create_user():
|
|
return render_template('temp-create-users.html', form=LoginForm())
|
|
|
|
|
|
@main.route('/temp-create-users', methods=(['POST']))
|
|
def create_user_for_test():
|
|
form = LoginForm()
|
|
if form.validate_on_submit():
|
|
user = Users(email_address=form.email_address.data,
|
|
name=form.email_address.data,
|
|
password=form.password.data,
|
|
created_at=datetime.now(),
|
|
mobile_number='+447651234534',
|
|
role_id=1)
|
|
users_dao.insert_user(user)
|
|
|
|
return redirect('/sign-in')
|
|
else:
|
|
print(form.errors)
|
|
return redirect(form.errors), 400
|