Produits Entr'ouvert » Corbo

from django.utils.translation import ugettext_lazy as _

from .models import Announce, Category, Broadcast

from .channels import get_channel_choices

    transport_channel = forms.MultipleChoiceField(required=False,

                                                  choices=get_channel_choices(),

                                                  widget=forms.CheckboxSelectMultiple())

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

        super(AnnounceForm, self).__init__(*args, **kwargs)

        self.fields['transport_channel'].initial = [b.channel for \

                                                    b in self.instance.broadcast_set.all()]

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

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

        if instance:

            channels = self.cleaned_data['transport_channel']

            for channel in channels:

                Broadcast.objects.get_or_create(announce=instance,

                                                channel=channel)

            self.instance.broadcast_set.exclude(announce=instance,

                                                channel__in=channels).delete()

        return instance