96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
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.username')
|
|
|
|
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 = 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
|