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': "Аккаунт не активен.",
    }