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

feat: 设置人脸上下文存活时间

pull/14483/head
Aaron3S 2024-11-19 17:31:05 +08:00 committed by feng626
parent 5b4de02fff
commit 5eeff0aabf
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/authentication/api/mfa.py
View File

@ -18,7 +18,7 @@ from common.utils import get_logger
from users.models.user import User from users.models.user import User
from .. import errors from .. import errors
from .. import serializers from .. import serializers
from ..const import MFA_FACE_CONTEXT_CACHE_KEY_PREFIX, MFA_FACE_SESSION_KEY from ..const import MFA_FACE_CONTEXT_CACHE_KEY_PREFIX, MFA_FACE_SESSION_KEY, MFA_FACE_CONTEXT_CACHE_TTL
from ..errors import SessionEmptyError from ..errors import SessionEmptyError
from ..mixins import AuthMixin from ..mixins import AuthMixin
@ -74,7 +74,7 @@ class MFAFaceCallbackApi(AuthMixin, CreateAPIView):
def _update_cache(self, context): def _update_cache(self, context):
cache_key = self.get_face_cache_key(context['token']) cache_key = self.get_face_cache_key(context['token'])
cache.set(cache_key, context, 3600) cache.set(cache_key, context, MFA_FACE_CONTEXT_CACHE_TTL)
def _handle_success(self, context, face_code): def _handle_success(self, context, face_code):
context.update({ context.update({
@ -100,7 +100,7 @@ class MFAFaceContextApi(AuthMixin, RetrieveAPIView, CreateAPIView):
"token": token, "token": token,
"is_finished": False "is_finished": False
} }
cache.set(cache_key, face_context) cache.set(cache_key, face_context, MFA_FACE_CONTEXT_CACHE_TTL)
self.request.session[self.face_token_session_key] = token self.request.session[self.face_token_session_key] = token
return token return token

1
apps/authentication/const.py
View File

@ -41,4 +41,5 @@ class MFAType(TextChoices):
MFA_FACE_CONTEXT_CACHE_KEY_PREFIX = "MFA_FACE_RECOGNITION_CONTEXT" MFA_FACE_CONTEXT_CACHE_KEY_PREFIX = "MFA_FACE_RECOGNITION_CONTEXT"
MFA_FACE_CONTEXT_CACHE_TTL = 60
MFA_FACE_SESSION_KEY = "mfa_face_token" MFA_FACE_SESSION_KEY = "mfa_face_token"