mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.7 KiB
58 lines
1.7 KiB
from django.core.cache import cache
|
|
|
|
from authentication.mfa.base import BaseMFA
|
|
from django.shortcuts import reverse
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from authentication.mixins import MFAFaceMixin
|
|
from common.const import LicenseEditionChoices
|
|
from settings.api import settings
|
|
|
|
|
|
class MFAFace(BaseMFA, MFAFaceMixin):
|
|
name = "face"
|
|
display_name = _('Face Recognition')
|
|
placeholder = 'Face Recognition'
|
|
|
|
def check_code(self, code):
|
|
|
|
assert self.is_authenticated()
|
|
|
|
try:
|
|
code = self.get_face_code()
|
|
if not self.user.check_face(code):
|
|
return False, _('Facial comparison failed')
|
|
except Exception as e:
|
|
return False, "{}:{}".format(_('Facial comparison failed'), str(e))
|
|
return True, ''
|
|
|
|
def is_active(self):
|
|
if not self.is_authenticated():
|
|
return True
|
|
return bool(self.user.face_vector)
|
|
|
|
@staticmethod
|
|
def global_enabled():
|
|
return settings.XPACK_LICENSE_IS_VALID \
|
|
and LicenseEditionChoices.ULTIMATE == \
|
|
LicenseEditionChoices.from_key(settings.XPACK_LICENSE_EDITION) \
|
|
and settings.FACE_RECOGNITION_ENABLED
|
|
|
|
def get_enable_url(self) -> str:
|
|
return reverse('authentication:user-face-enable')
|
|
|
|
def get_disable_url(self) -> str:
|
|
return reverse('authentication:user-face-disable')
|
|
|
|
def disable(self):
|
|
assert self.is_authenticated()
|
|
self.user.face_vector = ''
|
|
self.user.save(update_fields=['face_vector'])
|
|
|
|
def can_disable(self) -> bool:
|
|
return True
|
|
|
|
@staticmethod
|
|
def help_text_of_enable():
|
|
return _("Frontal Face Recognition")
|