mirror of https://github.com/jumpserver/jumpserver
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from captcha.fields import CaptchaField, CaptchaTextInput
|
|
from django import forms
|
|
from django.conf import settings
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from common.utils import get_logger, decrypt_password
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class EncryptedField(forms.CharField):
|
|
def to_python(self, value):
|
|
value = super().to_python(value)
|
|
return decrypt_password(value)
|
|
|
|
|
|
class UserLoginForm(forms.Form):
|
|
username = forms.CharField(
|
|
label=_('Username'), max_length=100,
|
|
widget=forms.TextInput(attrs={
|
|
'placeholder': _("Username"),
|
|
'autofocus': 'autofocus'
|
|
})
|
|
)
|
|
password = EncryptedField(
|
|
label=_('Password'), widget=forms.PasswordInput,
|
|
max_length=1024, strip=False
|
|
)
|
|
auto_login = forms.BooleanField(
|
|
required=False, initial=False,
|
|
widget=forms.CheckboxInput()
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
auto_login_field = self.fields['auto_login']
|
|
auto_login_field.label = _("Auto login")
|
|
if settings.SESSION_EXPIRE_AT_BROWSER_CLOSE:
|
|
auto_login_field.widget = forms.HiddenInput()
|
|
|
|
def confirm_login_allowed(self, user):
|
|
if not user.is_staff:
|
|
raise forms.ValidationError(
|
|
self.error_messages['inactive'],
|
|
code='inactive',
|
|
)
|
|
|
|
|
|
class UserCheckOtpCodeForm(forms.Form):
|
|
code = forms.CharField(label=_('MFA Code'), max_length=128, required=False)
|
|
mfa_type = forms.CharField(label=_('MFA type'), max_length=128)
|
|
|
|
|
|
class CustomCaptchaTextInput(CaptchaTextInput):
|
|
template_name = 'authentication/_captcha_field.html'
|
|
|
|
|
|
class CaptchaMixin(forms.Form):
|
|
captcha = CaptchaField(widget=CustomCaptchaTextInput, label=_('Captcha'))
|
|
|
|
|
|
class ChallengeMixin(forms.Form):
|
|
challenge = forms.CharField(
|
|
label=_('MFA code'), max_length=128, required=False,
|
|
widget=forms.TextInput(attrs={
|
|
'placeholder': _("Dynamic code"),
|
|
'style': 'width: 50%'
|
|
})
|
|
)
|
|
|
|
|
|
def get_user_login_form_cls(*, captcha=False):
|
|
bases = []
|
|
if settings.SECURITY_LOGIN_CHALLENGE_ENABLED:
|
|
bases.append(ChallengeMixin)
|
|
elif settings.SECURITY_MFA_IN_LOGIN_PAGE:
|
|
bases.append(UserCheckOtpCodeForm)
|
|
elif settings.SECURITY_LOGIN_CAPTCHA_ENABLED and captcha:
|
|
bases.append(CaptchaMixin)
|
|
bases.append(UserLoginForm)
|
|
return type('UserLoginForm', tuple(bases), {})
|