mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-11-03 15:37:56 +00:00
75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
from django.conf import settings
|
|
from django.template.loader import render_to_string
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from common.utils import random_string
|
|
from common.utils.verify_code import SendAndVerifyCodeUtil
|
|
from settings.utils import get_login_title
|
|
from .base import BaseMFA
|
|
from ..const import MFAType
|
|
|
|
email_failed_msg = _("Email verify code invalid")
|
|
|
|
|
|
class MFAEmail(BaseMFA):
|
|
name = MFAType.Email.value
|
|
display_name = MFAType.Email.name
|
|
placeholder = _('Email verification code')
|
|
|
|
def _check_code(self, code):
|
|
assert self.is_authenticated()
|
|
sender_util = SendAndVerifyCodeUtil(self.user.email, backend=self.name)
|
|
ok = False
|
|
msg = ''
|
|
try:
|
|
ok = sender_util.verify(code)
|
|
except Exception as e:
|
|
msg = str(e)
|
|
return ok, msg
|
|
|
|
def is_active(self):
|
|
if not self.is_authenticated():
|
|
return True
|
|
return self.user.email
|
|
|
|
@staticmethod
|
|
def challenge_required():
|
|
return True
|
|
|
|
def send_challenge(self):
|
|
code = random_string(settings.SMS_CODE_LENGTH, lower=False, upper=False)
|
|
subject = '%s: %s' % (get_login_title(), _('MFA code'))
|
|
tip = _('The validity period of the verification code is {} minute').format(settings.VERIFY_CODE_TTL // 60)
|
|
context = {
|
|
'user': self.user, 'title': subject, 'code': code, 'tip': tip,
|
|
}
|
|
message = render_to_string('authentication/_msg_mfa_email_code.html', context)
|
|
content = {'subject': subject, 'message': message}
|
|
sender_util = SendAndVerifyCodeUtil(
|
|
self.user.email, code=code, backend=self.name, **content
|
|
)
|
|
sender_util.gen_and_send_async()
|
|
|
|
@staticmethod
|
|
def global_enabled():
|
|
return settings.SECURITY_MFA_BY_EMAIL
|
|
|
|
def disable(self):
|
|
return '/ui/#/profile/index'
|
|
|
|
def get_enable_url(self) -> str:
|
|
return ''
|
|
|
|
def can_disable(self) -> bool:
|
|
return False
|
|
|
|
def get_disable_url(self):
|
|
return ''
|
|
|
|
@staticmethod
|
|
def help_text_of_enable():
|
|
return ''
|
|
|
|
def help_text_of_disable(self):
|
|
return ''
|