Source code for news.forms

__copyright__ = "Copyright © Stichting SciPost (SciPost Foundation)"
__license__ = "AGPL v3"


from django import forms

from .models import NewsLetter, NewsItem, NewsLetterNewsItemsTable


class NewsLetterForm(forms.ModelForm):

    class Meta:
        model = NewsLetter
        fields = ['date', 'intro', 'closing', 'published']


class NewsItemForm(forms.ModelForm):

    class Meta:
        model = NewsItem
        fields = ['date', 'headline', 'blurb_short', 'blurb',
                  'image', 'css_class',
                  'followup_link', 'followup_link_text',
                  'published', 'on_homepage']


class NewsLetterNewsItemsTableForm(forms.ModelForm):

    class Meta:
        model = NewsLetterNewsItemsTable
        fields = ['newsitem']


class NewsLetterNewsItemsTableFormSet(forms.BaseModelFormSet):
    def save(self, *args, **kwargs):
        objects = super().save(*args, **kwargs)
        for form in self.ordered_forms:
            form.instance.order = form.cleaned_data['ORDER']
            form.instance.save()
        return objects


NewsLetterNewsItemsOrderingFormSet = forms.modelformset_factory(
    NewsLetterNewsItemsTable, fields=(), can_order=True, extra=0,
    formset=NewsLetterNewsItemsTableFormSet)