109526520: render verify template with VerifyForm

This commit is contained in:
Rebecca Law
2015-12-04 16:21:01 +00:00
parent ab36653f24
commit 69da9f8f32
5 changed files with 33 additions and 7 deletions

View File

@@ -3,4 +3,4 @@ from flask import Blueprint
main = Blueprint('main', __name__)
from app.main.views import index, sign_in, register
from app.main.views import index, sign_in, register, verify

View File

@@ -1,5 +1,5 @@
from flask_wtf import Form
from wtforms import StringField, PasswordField
from wtforms import StringField, PasswordField, IntegerField
from wtforms.validators import DataRequired, Email, Length, Regexp
from app.main.validators import Blacklist
@@ -36,3 +36,8 @@ class RegisterUserForm(Form):
validators=[DataRequired(message='Please enter your password'),
Length(10, 255, message='Password must be at least 10 characters'),
Blacklist(message='That password is blacklisted, too common')])
class VerifyForm(Form):
sms_code = IntegerField(DataRequired(message='SMS code can not be empty'))
email_code = IntegerField(DataRequired(message='Email code can not be empty'))

View File

@@ -19,11 +19,6 @@ def registerfrominvite():
return render_template('register-from-invite.html')
@main.route("/verify")
def verify():
return render_template('verify.html')
@main.route("/verify-mobile")
def verifymobile():
return render_template('verify-mobile.html')

19
app/main/views/verify.py Normal file
View File

@@ -0,0 +1,19 @@
from app.main import main
from flask import render_template, redirect, jsonify
from app.main.forms import VerifyForm
@main.route('/verify', methods=['GET'])
def render_verify():
return render_template('verify.html', form=VerifyForm())
@main.route('/verify', methods=['POST'])
def process_verify():
form = VerifyForm()
if form.validate_on_submit():
return redirect('/add-service')
else:
return jsonify(form.errors), 400

View File

@@ -0,0 +1,7 @@
def test_should_return_verify_template(notifications_admin, notifications_admin_db):
response = notifications_admin.test_client().get('/verify')
assert response.status_code == 200
assert 'Activate your account' in response.get_data(as_text=True)