Produits Entr'ouvert » Corbo

from django import forms

from django.utils.translation import ugettext_lazy as _

from django.utils.text import slugify

from django.core.exceptions import ObjectDoesNotExist

from .models import Announce, Category, Broadcast, channel_choices

class AnnounceForm(forms.ModelForm):

    class Meta:

        model = Announce

        exclude = ('identifier',)

        widgets = {

            'publication_time': forms.TextInput(attrs={'class': 'datetimepicker',

                                                       'readonly': True}),

            'expiration_time': forms.TextInput(attrs={'class': 'datetimepicker',

                                                      'readonly': True}),

            'category': forms.HiddenInput()

        }

    def save(self, *args, **kwargs):

        instance = super(AnnounceForm, self).save(*args, **kwargs)

        if instance:

            Broadcast.objects.get_or_create(announce=instance)

        return instance

class CategoryForm(forms.ModelForm):

    class Meta:

        fields = ('name', 'rss_feed_url')

        model = Category

    def save(self, commit=True):

        slug = slugify(self.instance.name)

        base_slug = slug

        if not self.instance.slug:

            i = 1

            while True:

                try:

                    c = Category.objects.get(slug=slug)

                except ObjectDoesNotExist:

                    break

                i += 1

                slug = '%s-%s' % (base_slug, i)

            self.instance.slug = slug

        return super(CategoryForm, self).save(commit=commit)