New-password endpoints are implemented.

There should be a better way to validate the token.
This commit is contained in:
Rebecca Law
2016-01-06 17:37:07 +00:00
parent f94966154d
commit b5901a1ac7
9 changed files with 104 additions and 36 deletions

View File

@@ -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