68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
from django import forms
|
||
from django.contrib.auth.forms import AuthenticationForm
|
||
from django_registration.forms import RegistrationFormUniqueEmail
|
||
|
||
from main.models import UserProfile
|
||
|
||
|
||
class CustomRegistrationForm(RegistrationFormUniqueEmail):
|
||
"""
|
||
Форма для регистрации :class:`django_registration.forms.RegistrationFormUniqueEmail`
|
||
|
||
с добавлением bootstrap-класса "form-control"
|
||
|
||
:param visible_fields.email: Поле для ввода email, зарегистирированного на Zendesk
|
||
:type visible_fields.email: :class:`django_registration.forms.RegistrationFormUniqueEmail`
|
||
"""
|
||
|
||
def __init__(self, *args, **kwargs) -> RegistrationFormUniqueEmail:
|
||
super().__init__(*args, **kwargs)
|
||
for visible in self.visible_fields():
|
||
if visible.field.widget.attrs.get('class', False):
|
||
if visible.field.widget.attrs['class'].find('form-control') < 0:
|
||
visible.field.widget.attrs['class'] += 'form-control'
|
||
else:
|
||
visible.field.widget.attrs['class'] = 'form-control'
|
||
if visible.html_name !='email':
|
||
visible.field.required = False
|
||
|
||
class Meta(RegistrationFormUniqueEmail.Meta):
|
||
fields = RegistrationFormUniqueEmail.Meta.fields
|
||
|
||
|
||
class AdminPageUsers(forms.Form):
|
||
"""
|
||
Форма для установки статусов engineer или light_agent пользователям
|
||
|
||
:param users: Поле для установки статуса
|
||
:type users: :class:`ModelMultipleChoiceField`
|
||
"""
|
||
|
||
users = forms.ModelMultipleChoiceField(
|
||
queryset=UserProfile.objects.filter(role='agent'),
|
||
widget=forms.CheckboxSelectMultiple(
|
||
attrs={
|
||
'class': 'form-check-input'
|
||
}
|
||
),
|
||
label=''
|
||
)
|
||
|
||
|
||
class CustomAuthenticationForm(AuthenticationForm):
|
||
"""
|
||
Форма для авторизации :class:`django.contrib.auth.forms.AuthenticationForm`
|
||
с изменением поля username на email
|
||
"""
|
||
username = forms.CharField(
|
||
label="Электронная почта",
|
||
widget=forms.EmailInput(),
|
||
)
|
||
error_messages = {
|
||
'invalid_login':
|
||
"Пожалуйста, введите правильные электронную почту и пароль. Оба поля "
|
||
"могут быть чувствительны к регистру."
|
||
,
|
||
'inactive': "Аккаунт не активен.",
|
||
}
|