mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-05-17 04:19:54 +00:00
13ae50f5d0
* feat: 自定义MFA认证 * feat: 修改翻译信息 Co-authored-by: halo <wuyihuangw@gmail.com> Co-authored-by: Jiangjie.Bai <bugatti_it@163.com>
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from django.db.models import TextChoices
|
|
|
|
from authentication.confirm import CONFIRM_BACKENDS
|
|
from .confirm import ConfirmMFA, ConfirmPassword, ConfirmReLogin
|
|
from .mfa import MFAOtp, MFASms, MFARadius, MFACustom
|
|
|
|
RSA_PRIVATE_KEY = 'rsa_private_key'
|
|
RSA_PUBLIC_KEY = 'rsa_public_key'
|
|
|
|
CONFIRM_BACKEND_MAP = {backend.name: backend for backend in CONFIRM_BACKENDS}
|
|
|
|
|
|
class ConfirmType(TextChoices):
|
|
ReLogin = ConfirmReLogin.name, ConfirmReLogin.display_name
|
|
PASSWORD = ConfirmPassword.name, ConfirmPassword.display_name
|
|
MFA = ConfirmMFA.name, ConfirmMFA.display_name
|
|
|
|
@classmethod
|
|
def get_can_confirm_types(cls, confirm_type):
|
|
start = cls.values.index(confirm_type)
|
|
types = cls.values[start:]
|
|
types.reverse()
|
|
return types
|
|
|
|
@classmethod
|
|
def get_can_confirm_backend_classes(cls, confirm_type):
|
|
types = cls.get_can_confirm_types(confirm_type)
|
|
backend_classes = [
|
|
CONFIRM_BACKEND_MAP[tp] for tp in types if tp in CONFIRM_BACKEND_MAP
|
|
]
|
|
return backend_classes
|
|
|
|
|
|
class MFAType(TextChoices):
|
|
OTP = MFAOtp.name, MFAOtp.display_name
|
|
SMS = MFASms.name, MFASms.display_name
|
|
Radius = MFARadius.name, MFARadius.display_name
|
|
Custom = MFACustom.name, MFACustom.display_name
|