Produits Entr'ouvert » Corbo

import csv

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 django.core import validators

from django.core.exceptions import ValidationError

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)

class SubscriptionsImportForm(forms.Form):

    subscribers = forms.FileField(_('Subscribers'),

                    help_text=_('utf-8 encoded, comma separated file with email addresses on first column'))

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

        subscribers = []

        reader = csv.reader(self.cleaned_data['subscribers'])

        for idx, row in enumerate(reader, 1):

            if not row or not row[0]:

                continue

            try:

                validators.validate_email(row[0])

            except ValidationError:

                raise ValidationError(_('Invalid email address at line %d' % idx))

            subscribers.append(row[0])

        return subscribers