from django.forms import ModelForm
from django.forms.models import inlineformset_factory

from models import Sponsor, Sponsorship


class SponsorForm(ModelForm):

    class Meta:
        model = Sponsor


class SponsorshipForm(ModelForm):

    class Meta:
        model = Sponsorship


SponsorShipsFormSet = inlineformset_factory(Sponsor, Sponsorship,
                                            form=SponsorshipForm, extra=2)