from django import forms
from django.contrib.auth.forms import UserCreationForm, UserChangeForm, PasswordChangeForm
#from django.contrib.auth.models import User
from .models import Person
from django.core.exceptions import ValidationError
import datetime
from django.forms.widgets import NumberInput

class CustomRegisterForm(UserCreationForm):
    GENDER_CHOICES = [
		('MALE','MALE'),
		('FEMALE', 'FEMALE')
	]
    ROLE_CHOICES = [
        ('Student', 'Student'),
        ('Staff', 'Staff'),
        ('Bursar', 'Bursar'),
    ]
    phone_number = forms.CharField(label='Phone Number')
    gender = forms.ChoiceField(label='Gender', choices=GENDER_CHOICES, widget=forms.RadioSelect)
    role = forms.ChoiceField(label='User Type', choices=ROLE_CHOICES)
    #subject = forms.CharField(max_length=255)
    dob = forms.DateField(widget=NumberInput(attrs={'type': 'date'}))
    #address = forms.CharField(max_length=200, required=False)
    first_name = forms.CharField(max_length=30)
    last_name = forms.CharField(max_length=30)
    email = forms.EmailField()
    username = forms.CharField(max_length=20)
    # def clean_email(self):
    #    email = self.cleaned_data.get('email')
    #    if Person.objects.filter(email=email).exists():
    #        raise ValidationError("A user with the supplied email already exists")
    #    return email

    def clean_username(self):
       username = self.cleaned_data.get('username')
       if Person.objects.filter(username=username).exists():
           raise ValidationError("A user with the supplied username already exists")
       return username

    def clean_dob(self):
       dob = self.cleaned_data.get('dob')
       if dob > datetime.date.today():
           raise ValidationError("The selected date is invalid. Please re-select another")
       return dob

    class Meta:
        model = Person
        fields = ['first_name', 'last_name', 'email', 'role', 'username', 'password1', 'password2', 'classe', 'phone_number', 'gender', 'address', 'dob',  'photograph']

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['first_name'].label = 'Surname'
        self.fields['last_name'].label = 'Other Names'
        self.fields['username'].label = 'Admission No/Staff No (Username)'
        self.fields['username'].help_text = "For staff, choose a username for logging in in subsequent times"
        self.fields['classe'].help_text = "Select only for student. Leave blank for staff"
        self.fields['email'].help_text = "This field must be a valid email address"
        self.fields['password1'].help_text = "<ul><li>Your password can’t be too similar to your other Personal information.<li>Your password must contain at least 8 characters.</li><li>Your password can’t be a commonly used password.</li><li>Your password can’t be entirely numeric.</li></ul>"
        self.fields['password2'].label = "Password Confirmation"
        self.fields['phone_number'].label = "Phone Number"
        self.fields['dob'].label = "Date of Birth"
        self.fields['role'].help_text = "Select Staff for a teacher"
###
class CustomRegisterFormAdmin(UserChangeForm):
    GENDER_CHOICES = [
		('MALE','MALE'),
		('FEMALE', 'FEMALE')
	]
    ROLE_CHOICES = [
        ('Student', 'Student'),
        ('Staff', 'Staff'),
        ('Admin', 'Admin')
    ]
    phone_number = forms.CharField(label='Phone Number')
    gender = forms.ChoiceField(label='Gender', choices=GENDER_CHOICES, widget=forms.RadioSelect)
    dob = forms.DateField(widget=NumberInput(attrs={'type': 'date'}))
    address = forms.CharField(max_length=200, required=False)
    first_name = forms.CharField(max_length=30)
    last_name = forms.CharField(max_length=30)
    email = forms.EmailField()
    username = forms.CharField(max_length=20)
    # def clean_email(self):
    #    email = self.cleaned_data.get('email')
    #    if Person.objects.filter(email=email).exists():
    #        raise ValidationError("A user with the supplied email already exists")
    #    return email
    def clean_dob(self):
       dob = self.cleaned_data.get('dob')
       if dob > datetime.date.today():
           raise ValidationError("The selected date is invalid. Please re-select another")
       return dob

    class Meta:
        model = Person
        fields = ['first_name', 'last_name', 'email', 'username', 'phone_number', 'gender', 'address', 'dob',  'photograph']

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['first_name'].label = 'Surname'
        self.fields['last_name'].label = 'Other Names'
        self.fields['username'].label = 'Staff Number'
        self.fields['email'].help_text = "This field must be a valid email address"
        self.fields['phone_number'].label = "Phone Number"
        self.fields['dob'].label = "Date of Birth"

class CustomPasswordChangeForm(PasswordChangeForm):
    class Meta:
        model = Person
        fields = ['old_password', 'new_password1', 'new_password2']
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.fields['new_password1'].help_text = "<ul><li>Your password can’t be too similar to your other Personal information.<li>Your password must contain at least 8 characters.</li><li>Your password can’t be a commonly used password.</li><li>Your password can’t be entirely numeric.</li></ul>"

class CustomRegisterFormStaff(UserChangeForm):
    GENDER_CHOICES = [
		('MALE','MALE'),
		('FEMALE', 'FEMALE')
	]
    ROLE_CHOICES = [
        ('Student', 'Student'),
        ('Staff', 'Staff'),
        ('Admin', 'Admin')
    ]
    phone_number = forms.CharField(label='Phone Number')
    gender = forms.ChoiceField(label='Gender', choices=GENDER_CHOICES, widget=forms.RadioSelect)
    dob = forms.DateField(widget=NumberInput(attrs={'type': 'date'}))
    address = forms.CharField(max_length=200, required=False)
    first_name = forms.CharField(max_length=30)
    last_name = forms.CharField(max_length=30)
    email = forms.EmailField()
    username = forms.CharField(max_length=20)
    photograph = forms.ImageField(widget=forms.FileInput, required=False)
    # def clean_email(self):
    #    email = self.cleaned_data.get('email')
    #    if Person.objects.filter(email=email).exists():
    #        raise ValidationError("A user with the supplied email already exists")
    #    return email
    def clean_dob(self):
       dob = self.cleaned_data.get('dob')
       if dob > datetime.date.today():
           raise ValidationError("The selected date is invalid. Please re-select another")
       return dob

    class Meta:
        model = Person
        fields = ['first_name', 'last_name', 'email', 'username', 'phone_number', 'gender', 'address', 'dob', 'photograph']

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['first_name'].label = 'Surname'
        self.fields['last_name'].label = 'Other Names'
        self.fields['username'].label = 'Staff Number'
        self.fields['email'].help_text = "This field must be a valid email address"
        self.fields['phone_number'].label = "Phone Number"
        self.fields['dob'].label = "Date of Birth"

class CustomRegisterFormStudent(UserChangeForm):
    GENDER_CHOICES = [
		('MALE','MALE'),
		('FEMALE', 'FEMALE')
	]
    ROLE_CHOICES = [
        ('Student', 'Student'),
        ('Staff', 'Staff'),
        ('Admin', 'Admin')
    ]
    phone_number = forms.CharField(label='Phone Number')
    gender = forms.ChoiceField(label='Gender', choices=GENDER_CHOICES, widget=forms.RadioSelect)
    dob = forms.DateField(widget=NumberInput(attrs={'type': 'date'}))
    address = forms.CharField(max_length=200, required=False)
    first_name = forms.CharField(max_length=30)
    last_name = forms.CharField(max_length=30)
    email = forms.EmailField()
    username = forms.CharField(max_length=20)
    # def clean_email(self):
    #    email = self.cleaned_data.get('email')
    #    if Person.objects.filter(email=email).exists():
    #        raise ValidationError("A user with the supplied email already exists")
    #    return email

    def clean_dob(self):
       dob = self.cleaned_data.get('dob')
       if dob > datetime.date.today():
           raise ValidationError("The selected date is invalid. Please re-select another")
       return dob

    def __init__(self, *args, **kwargs):
        super(CustomRegisterFormStudent, self).__init__(*args, **kwargs)
        instance = getattr(self, 'instance', None)
        if instance and instance.id:
            self.fields['username'].required = False
            self.fields['username'].widget.attrs['disabled'] = 'disabled'
            self.fields['first_name'].required = False
            self.fields['first_name'].widget.attrs['disabled'] = 'disabled'
            self.fields['last_name'].required = False
            self.fields['last_name'].widget.attrs['disabled'] = 'disabled'
            self.fields['classe'].required = False
            self.fields['classe'].widget.attrs['disabled'] = 'disabled'

            self.fields['dob'].required = False
            self.fields['dob'].widget.attrs['disabled'] = 'disabled'
            self.fields['photograph'].required = False
            self.fields['photograph'].widget.attrs['disabled'] = 'disabled'
            self.fields['gender'].required = False
            self.fields['gender'].widget.attrs['disabled'] = 'disabled'
            self.fields['phone_number'].required = False
            self.fields['phone_number'].widget.attrs['disabled'] = 'disabled'

            self.fields['first_name'].label = 'First Name'
            self.fields['last_name'].label = 'Last Name'
            self.fields['username'].label = 'Admission No'
            self.fields['email'].help_text = "This field must be a valid email address"
            self.fields['phone_number'].label = "Phone Number"
            self.fields['dob'].label = "Date of Birth"

    def clean_username(self):
        instance = getattr(self, 'instance', None)
        if instance:
            return instance.username
        else:
            return self.cleaned_data.get('username', None)

    def clean_first_name(self):
        instance = getattr(self, 'instance', None)
        if instance:
            return instance.first_name
        else:
            return self.cleaned_data.get('first_name', None)

    def clean_last_name(self):
        instance = getattr(self, 'instance', None)
        if instance:
            return instance.last_name
        else:
            return self.cleaned_data.get('last_name', None)

    def clean_classe(self):
        instance = getattr(self, 'instance', None)
        if instance:
            return instance.classe
        else:
            return self.cleaned_data.get('classe', None)

    def clean_dob(self):
        instance = getattr(self, 'instance', None)
        if instance:
            return instance.dob
        else:
            return self.cleaned_data.get('dob', None)

    def clean_photograph(self):
        instance = getattr(self, 'instance', None)
        if instance:
            return instance.photograph
        else:
            return self.cleaned_data.get('photograph', None)

    def clean_gender(self):
        instance = getattr(self, 'instance', None)
        if instance:
            return instance.gender
        else:
            return self.cleaned_data.get('gender', None)

    def clean_phone_number(self):
        instance = getattr(self, 'instance', None)
        if instance:
            return instance.phone_number
        else:
            return self.cleaned_data.get('phone_number', None)

    class Meta:
        model = Person
        fields = ['first_name', 'last_name', 'email', 'username', 'classe', 'phone_number', 'gender', 'address', 'dob',  'photograph']

class CustomRegisterFormStud(UserChangeForm):
    GENDER_CHOICES = [
		('MALE','MALE'),
		('FEMALE', 'FEMALE')
	]
    ROLE_CHOICES = [
        ('Student', 'Student'),
        ('Staff', 'Staff'),
        ('A---n', 'A---n')
    ]
    phone_number = forms.CharField(label='Phone Number')
    gender = forms.ChoiceField(label='Gender', choices=GENDER_CHOICES, widget=forms.RadioSelect)
    dob = forms.DateField(widget=NumberInput(attrs={'type': 'date'}))
    address = forms.CharField(max_length=200, required=False)
    first_name = forms.CharField(max_length=30)
    last_name = forms.CharField(max_length=30)
    email = forms.EmailField()
    username = forms.CharField(max_length=20)
    # def clean_email(self):
    #    email = self.cleaned_data.get('email')
    #    if Person.objects.filter(email=email).exists():
    #        raise ValidationError("A user with the supplied email already exists")
    #    return email

    def clean_dob(self):
       dob = self.cleaned_data.get('dob')
       if dob > datetime.date.today():
           raise ValidationError("The selected date is invalid. Please re-select another")
       return dob


    class Meta:
        model = Person
        fields = ['first_name', 'last_name', 'email', 'username', 'classe', 'phone_number', 'gender', 'address', 'dob',  'photograph']