trustchain-oc1-orchestral
/
authentik
Archived
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

stages/authenticator_totp: fix key error 

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
Jens Langhammer 2022-05-14 19:57:00 +02:00
parent be06adcb59
commit 5644d5f3f7
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
authentik/stages/authenticator_totp/stage.py
View File

@ -40,10 +40,11 @@ class AuthenticatorTOTPChallengeResponse(ChallengeResponse):
def validate_code(self, code: int) -> int: def validate_code(self, code: int) -> int:
"""Validate totp code""" """Validate totp code"""
if self.device is not None: if not self.device:
if not self.device.verify_token(code): raise ValidationError(_("Code does not match"))
self.device.confirmed = False if not self.device.verify_token(code):
raise ValidationError(_("Code does not match")) self.device.confirmed = False
raise ValidationError(_("Code does not match"))
return code return code
@ -65,7 +66,7 @@ class AuthenticatorTOTPStageView(ChallengeStageView):
def get_response_instance(self, data: QueryDict) -> ChallengeResponse: def get_response_instance(self, data: QueryDict) -> ChallengeResponse:
response = super().get_response_instance(data) response = super().get_response_instance(data)
response.device = self.request.session[SESSION_TOTP_DEVICE] response.device = self.request.session.get(SESSION_TOTP_DEVICE)
return response return response
def get(self, request: HttpRequest, *args, **kwargs) -> HttpResponse: def get(self, request: HttpRequest, *args, **kwargs) -> HttpResponse: