from django.shortcuts import get_object_or_404 from rest_framework import serializers from event.models import Event, ActivityTag, Place, ActivityTemplate # Event Serializer class EventSerializer(serializers.HyperlinkedModelSerializer): created_by = serializers.ReadOnlyField(source='created_by.get_full_name') creation_date = serializers.ReadOnlyField() class Meta: model = Event fields = ('url', 'id', 'title', 'slug', 'created_by', 'creation_date', 'description', 'beginning_date', 'ending_date') # Classes utilitaires class EventSpecificSerializer(serializers.ModelSerializer): """ Provide `update` and `create` methods for nested view with an Event For example for Models which extends EventSpecificMixin the event id has to be provided in the `save` method Works fine with view.EventSpecificViewSet Also provides : event = eventserializer(allow_null=true, read_only=true) """ event = EventSerializer(allow_null=True, read_only=True) def update(self, instance, validated_data): """ Note : does NOT change the event value of the instance """ validated_data.pop('event_pk') [setattr(instance, key, value) for key, value in validated_data.items()] instance.save() return instance def create(self, validated_data): ModelClass = self.Meta.model event_pk = validated_data.pop('event_pk', None) event = event_pk and get_object_or_404(Event, id=event_pk) or None instance = ModelClass.objects.create(event=event, **validated_data) return instance # Serializers # TODO rajouter des permissions class PlaceSerializer(EventSpecificSerializer): class Meta: model = Place fields = ('url', 'id', 'name', 'description', 'event') # TODO rajouter des permissions class ActivityTagSerializer(EventSpecificSerializer): class Meta: model = ActivityTag fields = ('url', 'id', 'name', 'is_public', 'color', 'event') # TODO rajouter des permissions class ActivityTemplateSerializer(serializers.ModelSerializer): event = EventSerializer(read_only=True) tags = ActivityTagSerializer(many=True) class Meta: model = ActivityTemplate fields = ('id', 'title', 'event', 'is_public', 'has_perm', 'min_perm', 'max_perm', 'description', 'remarks', 'tags',) def update(self, instance, validated_data): tags_data = validated_data.pop('tags') validated_data.pop('event_pk') event = instance.event [setattr(instance, key, value) for key, value in validated_data.items()] instance.save() tags = [ActivityTag.objects.get_or_create(event=event, **tag_data)[0] for tag_data in tags_data] instance.tags.set(tags) return instance def create(self, validated_data): tags_data = validated_data.pop('tags') event_pk = validated_data.pop('event_pk') event = event_pk and get_object_or_404(Event, id=event_pk) or None activity_template = ActivityTemplate.objects.create(event=event, **validated_data) tags = [ActivityTag.objects.get_or_create(event=event, **tag_data)[0] for tag_data in tags_data] activity_template.tags = tags return activity_template class ActivitySerializer(serializers.ModelSerializer): pass