109526520: Use Regex validator to test the code is 5 digits.

This commit is contained in:
Rebecca Law
2015-12-08 15:40:45 +00:00
parent 9923c14e73
commit ae19161b32
2 changed files with 24 additions and 3 deletions

View File

@@ -20,6 +20,7 @@ class LoginForm(Form):
gov_uk_email = "(^[^@^\\s]+@[^@^\\.^\\s]+(\\.[^@^\\.^\\s]*)*.gov.uk)"
mobile_number = "^\\+44[\\d]{10}$"
verify_code = "[\\d]{5}$"
class RegisterUserForm(Form):
@@ -43,10 +44,10 @@ class RegisterUserForm(Form):
class VerifyForm(Form):
sms_code = StringField("Text message confirmation code",
validators=[DataRequired(message='SMS code can not be empty'),
Length(min=5, max=5, message='Code must be 5 digits')])
Regexp(regex=verify_code, message='Code must be 5 digits')])
email_code = StringField("Email confirmation code",
validators=[DataRequired(message='Email code can not be empty'),
Length(min=5, max=5, message='Code must be 5 digits')])
Regexp(regex=verify_code, message='Code must be 5 digits')])
def validate_email_code(self, a):
if self.email_code.data is not None: