from django import forms
from django.forms import formset_factory
from .models import Poll, Choice

class PollForm(forms.ModelForm):
    class Meta:
        model = Poll
        fields = ['question', 'team', 'multiple_choice']
        widgets = {
            'question': forms.TextInput(attrs={'class': 'form-control'}),
            'team': forms.Select(attrs={'class': 'form-control'}),
            'multiple_choice': forms.CheckboxInput(attrs={'class': 'form-check-input'}),
        }

class ChoiceForm(forms.ModelForm):
    class Meta:
        model = Choice
        fields = ['choice_text']
        widgets = {
            'choice_text': forms.TextInput(attrs={'class': 'form-control', 'required': True}),
        }

ChoiceFormSet = formset_factory(ChoiceForm, extra=5, max_num=5)