haiwen/seahub
1
0
Fork 0
mirror of https://github.com/haiwen/seahub.git synced 2025-07-14 07:24:58 +00:00
Code Issues Releases Wiki Activity

2fa failed_attempt (#5288)

Browse Source
This commit is contained in:
欢乐马 2022-11-11 15:35:23 +08:00 committed by GitHub
parent 132169105a
commit 16a686d2fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
seahub/two_factor/forms.py
View File

@ -291,6 +291,9 @@ class OTPAuthenticationFormMixin(object):
class AuthenticationTokenForm(OTPAuthenticationFormMixin, Form): class AuthenticationTokenForm(OTPAuthenticationFormMixin, Form):
SESSION_KEY_TWO_FACTOR_FAILED_ATTEMPT = '2fa-failed-attempt'
otp_token = forms.IntegerField(label=_("Token"), min_value=1, otp_token = forms.IntegerField(label=_("Token"), min_value=1,
max_value=int('9' * totp_digits())) max_value=int('9' * totp_digits()))
remember_me = forms.BooleanField(required=False) remember_me = forms.BooleanField(required=False)
@ -304,7 +307,15 @@ class AuthenticationTokenForm(OTPAuthenticationFormMixin, Form):
""" """
super(AuthenticationTokenForm, self).__init__(**kwargs) super(AuthenticationTokenForm, self).__init__(**kwargs)
self.user = user self.user = user
self.request = request
def clean(self): def clean(self):
self.clean_otp(self.user) self.clean_otp(self.user)
return self.cleaned_data return self.cleaned_data
def is_valid(self):
ret = super(Form, self).is_valid()
if not ret:
failed_attempt = self.request.session.get(self.SESSION_KEY_TWO_FACTOR_FAILED_ATTEMPT, 0)
self.request.session[self.SESSION_KEY_TWO_FACTOR_FAILED_ATTEMPT] = failed_attempt + 1
return ret