mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 10:53:28 -05:00
164bdad4f2
sms for verification. This may change again soon with story to split 2 factor pages, but for now is correct.
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
from flask import (
|
|
render_template,
|
|
redirect,
|
|
session,
|
|
url_for
|
|
)
|
|
|
|
from flask_login import login_user
|
|
|
|
from app.main import main
|
|
from app.main.dao import users_dao
|
|
from app.main.forms import (
|
|
VerifyForm,
|
|
VerifySmsForm
|
|
)
|
|
|
|
|
|
@main.route('/verify', methods=['GET', 'POST'])
|
|
def verify():
|
|
|
|
# TODO there needs to be a way to regenerate a session id
|
|
# or handle gracefully.
|
|
user_id = session['user_details']['id']
|
|
|
|
def _check_code(code, code_type):
|
|
return users_dao.check_verify_code(user_id, code, code_type)
|
|
|
|
if session.get('invited_user'):
|
|
form = VerifySmsForm(_check_code)
|
|
else:
|
|
form = VerifyForm(_check_code)
|
|
|
|
if form.validate_on_submit():
|
|
try:
|
|
user = users_dao.get_user_by_id(user_id)
|
|
activated_user = users_dao.activate_user(user)
|
|
login_user(activated_user)
|
|
return redirect(url_for('main.add_service', first='first'))
|
|
finally:
|
|
session.pop('user_details', None)
|
|
|
|
return render_template('views/verify.html', form=form)
|