mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-09-08 04:38:24 -04:00
New-password endpoints are implemented.
There should be a better way to validate the token.
This commit is contained in:
@@ -1,16 +1,29 @@
|
||||
from flask import request
|
||||
from datetime import datetime
|
||||
|
||||
from flask import (Markup, render_template, url_for, redirect)
|
||||
|
||||
from app.main import main
|
||||
from app.main.dao import (password_reset_token_dao, users_dao)
|
||||
from app.main.forms import NewPasswordForm
|
||||
from app.main.views import send_sms_code
|
||||
|
||||
|
||||
@main.route('/new-password/<token>', methods=['GET', 'POST'])
|
||||
def new_password():
|
||||
# Validate token
|
||||
token = request.args.get('token')
|
||||
# get password token (better name)
|
||||
# is it expired
|
||||
# add NewPasswordForm
|
||||
# update password
|
||||
# create password_token table (id, token, user_id, expiry_date
|
||||
@main.route('/new-password/<path:token>', methods=['GET', 'POST'])
|
||||
def new_password(token):
|
||||
form = NewPasswordForm()
|
||||
if form.validate_on_submit():
|
||||
password_reset_token = password_reset_token_dao.get_token(str(Markup.escape(token)))
|
||||
if not valid_token(password_reset_token):
|
||||
form.new_password.errors.append('token is invalid') # Is there a better way
|
||||
return render_template('views/new-password.html', form=form)
|
||||
else:
|
||||
users_dao.update_password(password_reset_token.user_id, form.new_password.data)
|
||||
user = users_dao.get_user_by_id(password_reset_token.user_id)
|
||||
send_sms_code(user.id, user.mobile_number)
|
||||
return redirect(url_for('main.render_two_factor'))
|
||||
else:
|
||||
return render_template('views/new-password.html', toke=token, form=form)
|
||||
|
||||
return 'Got here'
|
||||
|
||||
def valid_token(token):
|
||||
return token and datetime.now() <= token.expiry_date
|
||||
|
||||
Reference in New Issue
Block a user