github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

fix: Fix the uncaught exception when face capture fails

pull/14531/head
Aaron3S 2024-11-22 18:08:57 +08:00 committed by feng626
parent 77598a0f23
commit 01b8c1f7a8
2 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/authentication/templates/authentication/face_capture.html
View File

@ -85,7 +85,7 @@
} }
$('#retry_button').on('click', function () { $('#retry_button').on('click', function () {
window.location.href = "{% url 'authentication:login-face-capture' %}"; window.location.href = "{{ request.get_full_path }}";
}); });
}); });
</script> </script>

25
apps/users/views/profile/face.py
View File

@ -27,24 +27,31 @@ class UserFaceCaptureView(AuthMixin, FormView):
return super().form_valid(form) return super().form_valid(form)
def get_context_data(self, **kwargs): def get_context_data(self, **kwargs):
context = super().get_context_data() context = super().get_context_data(**kwargs)
if not self.get_form().is_bound: if 'form' not in kwargs:
form = self.get_form()
context['form'] = form
if not context['form'].is_bound:
context.update({ context.update({
"active": True, "active": True,
}) })
kwargs.update(context) return context
return kwargs
class UserFaceEnableView(UserFaceCaptureView, MFAFaceMixin): class UserFaceEnableView(MFAFaceMixin, UserFaceCaptureView):
def form_valid(self, form): def form_valid(self, form):
code = self.get_face_code()
user = self.get_user_from_session() try:
user.face_vector = code code = self.get_face_code()
user.save(update_fields=['face_vector']) user = self.get_user_from_session()
user.face_vector = code
user.save(update_fields=['face_vector'])
except Exception as e:
form.add_error("code", str(e))
return super().form_invalid(form)
auth_logout(self.request) auth_logout(self.request)
return super().form_valid(form) return super().form_valid(form)