from django import forms
from .models import InvitationCode, CustomUser

class InvitationCodeForm(forms.Form):
    code = forms.CharField(max_length=255, label="Einladungscode")

    def clean_code(self):
        code = self.cleaned_data.get('code')
        try:
            invitation_code = InvitationCode.objects.get(code=code)
            if not invitation_code.is_valid():
                raise forms.ValidationError("Dieser Einladungscode ist nicht mehr gültig.")
        except InvitationCode.DoesNotExist:
            raise forms.ValidationError("Ungültiger Einladungscode.")
        return code

class CustomUserCreationForm(forms.ModelForm):
    password = forms.CharField(widget=forms.PasswordInput)
    birth_date = forms.DateField(input_formats=['%d.%m.%Y'], widget=forms.DateInput(format='%d.%m.%Y'))
    class Meta:
        model = CustomUser
        fields = ('username', 'first_name', 'last_name', 'email', 'birth_date', 'player_number', 'password')

class CustomUserChangeForm(forms.ModelForm):
    birth_date = forms.DateField(input_formats=['%d.%m.%Y'], widget=forms.DateInput(format='%d.%m.%Y'))
    class Meta:
        model = CustomUser
        fields = ('username', 'first_name', 'last_name', 'email', 'birth_date', 'player_number', 'team')