from django import forms
from django_registration.forms import RegistrationFormUniqueEmail

from main.models import UserProfile


class CustomRegistrationForm(RegistrationFormUniqueEmail):
    """
    Форма для регистрации :class:`django_registration.forms.RegistrationFormUniqueEmail`
    с добавлением bootstrap-класса 'form-control'

    :param password_zen: Поле для ввода пароля от Zendesk
    :type password_zen: :class:`django.forms.CharField`
    """

    def __init__(self, *args, **kwargs):
        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):
    users = forms.ModelMultipleChoiceField(
        queryset=UserProfile.objects.filter(role='agent'),
        widget=forms.CheckboxSelectMultiple(
            attrs={
                'class': 'form-check-input'
            }
        ),
        label=''
    )