from django import forms
from .models import Voter, Interaction, Donation, VoterLikelihood, InteractionType, DonationMethod, ElectionType, Event, EventParticipation

class VoterForm(forms.ModelForm):
    class Meta:
        model = Voter
        fields = [
            'first_name', 'last_name', 'street_address', 'city', 'state', 'zip_code',
            'latitude', 'longitude', 'phone', 'email',
            'voter_id', 'district', 'precinct', 'registration_date',
            'candidate_support', 'yard_sign'
        ]
        widgets = {
            'registration_date': forms.DateInput(attrs={'type': 'date'}),
        }

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        for field in self.fields.values():
            field.widget.attrs.update({'class': 'form-control'})
        self.fields['candidate_support'].widget.attrs.update({'class': 'form-select'})
        self.fields['yard_sign'].widget.attrs.update({'class': 'form-select'})

class InteractionForm(forms.ModelForm):
    class Meta:
        model = Interaction
        fields = ['type', 'date', 'description', 'notes']
        widgets = {
            'date': forms.DateInput(attrs={'type': 'date'}),
            'notes': forms.Textarea(attrs={'rows': 2}),
        }

    def __init__(self, *args, tenant=None, **kwargs):
        super().__init__(*args, **kwargs)
        if tenant:
            self.fields['type'].queryset = InteractionType.objects.filter(tenant=tenant)
        for field in self.fields.values():
            field.widget.attrs.update({'class': 'form-control'})
        self.fields['type'].widget.attrs.update({'class': 'form-select'})

class DonationForm(forms.ModelForm):
    class Meta:
        model = Donation
        fields = ['date', 'method', 'amount']
        widgets = {
            'date': forms.DateInput(attrs={'type': 'date'}),
        }

    def __init__(self, *args, tenant=None, **kwargs):
        super().__init__(*args, **kwargs)
        if tenant:
            self.fields['method'].queryset = DonationMethod.objects.filter(tenant=tenant)
        for field in self.fields.values():
            field.widget.attrs.update({'class': 'form-control'})
        self.fields['method'].widget.attrs.update({'class': 'form-select'})

class VoterLikelihoodForm(forms.ModelForm):
    class Meta:
        model = VoterLikelihood
        fields = ['election_type', 'likelihood']

    def __init__(self, *args, tenant=None, **kwargs):
        super().__init__(*args, **kwargs)
        if tenant:
            self.fields['election_type'].queryset = ElectionType.objects.filter(tenant=tenant)
        for field in self.fields.values():
            field.widget.attrs.update({'class': 'form-control'})
        self.fields['election_type'].widget.attrs.update({'class': 'form-select'})
        self.fields['likelihood'].widget.attrs.update({'class': 'form-select'})

class EventParticipationForm(forms.ModelForm):
    class Meta:
        model = EventParticipation
        fields = ['event']

    def __init__(self, *args, tenant=None, **kwargs):
        super().__init__(*args, **kwargs)
        if tenant:
            self.fields['event'].queryset = Event.objects.filter(tenant=tenant)
        for field in self.fields.values():
            field.widget.attrs.update({'class': 'form-control'})
        self.fields['event'].widget.attrs.update({'class': 'form-select'})