Review fix.

This commit is contained in:
Nicholas Staples
2016-01-06 17:00:01 +00:00
parent 34bf03bae2
commit 52df795743
3 changed files with 4 additions and 3 deletions

View File

@@ -62,7 +62,7 @@ class RegisterUserForm(Form):
class TwoFactorForm(Form): class TwoFactorForm(Form):
sms_code = StringField('sms code', validators=[DataRequired(message='Please enter your code'), sms_code = StringField('sms code', validators=[DataRequired(message='Enter verification code'),
Regexp(regex=verify_code, message='Code must be 5 digits')]) Regexp(regex=verify_code, message='Code must be 5 digits')])
def validate_sms_code(self, a): def validate_sms_code(self, a):
@@ -114,6 +114,7 @@ class AddServiceForm(Form):
def validate_codes(field, code_type): def validate_codes(field, code_type):
codes = verify_codes_dao.get_codes(user_id=session['user_id'], code_type=code_type) codes = verify_codes_dao.get_codes(user_id=session['user_id'], code_type=code_type)
# TODO need to remove this manual logging.
print('validate_codes for user_id: {} are {}'.format(session['user_id'], codes)) print('validate_codes for user_id: {} are {}'.format(session['user_id'], codes))
if not [code for code in codes if validate_code(field, code)]: if not [code for code in codes if validate_code(field, code)]:
raise ValidationError('Code does not match') raise ValidationError('Code does not match')