Refactor sms_code functionality into the class

So it’s all in one place, not two.
This commit is contained in:
Chris Hill-Scott
2018-05-07 22:57:18 +01:00
parent 60c56be048
commit 02907afce1
2 changed files with 9 additions and 10 deletions

View File

@@ -161,19 +161,15 @@ def password(label='Password'):
class SMSCode(StringField):
def __call__(self, **kwargs):
return super().__call__(
type='tel', pattern='[0-9]*', **kwargs
)
def sms_code():
return SMSCode('Text message code', validators=[
validators = [
DataRequired(message='Cant be empty'),
Regexp(regex='^\d+$', message='Numbers only'),
Length(min=5, message='Not enough numbers'),
Length(max=5, message='Too many numbers'),
])
]
def __call__(self, **kwargs):
return super().__call__(type='tel', pattern='[0-9]*', **kwargs)
def organisation_type():
@@ -323,7 +319,7 @@ class TwoFactorForm(StripWhitespaceForm):
self.validate_code_func = validate_code_func
super(TwoFactorForm, self).__init__(*args, **kwargs)
sms_code = sms_code()
sms_code = SMSCode('Text message code')
def validate(self):