# -*- 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), {})