This commit is contained in:
Qwann 2017-07-21 17:03:33 +02:00
parent 5d26f35f72
commit 92e8e374fd
6 changed files with 106 additions and 10 deletions

5
api/event/mixins.py Normal file
View file

@ -0,0 +1,5 @@
class EventNestedMixin(object):
pass
class ventSpecificSerializerMixin(object):
pass

View file

@ -13,10 +13,33 @@ class EventSerializer(serializers.HyperlinkedModelSerializer):
# TODO rajouter des permissions
class PlaceSerializer(serializers.HyperlinkedModelSerializer):
event = EventSerializer(allow_null=True)
class Meta:
model = Place
fields = ('url', 'id', 'name', 'description',)
fields = ('url', 'id', 'name', 'description', 'event')
def update(self, instance, validated_data):
try:
data = validated_data.pop('event')
event = Event.objects.get_object_or_404(**data)
except KeyError:
event = None
[setattr(instance, key, value)
for key, value in validated_data.items()]
setattr(instance, 'event', event)
instance.save()
return instance
def create(self, validated_data):
ModelClass = self.Meta.model
try:
data = validated_data.pop('event')
event = Event.objects.get_object_or_404(**data)
except KeyError:
event = None
instance = ModelClass.objects.create(event=event, **validated_data)
return instance
# TODO rajouter des permissions
@ -39,19 +62,18 @@ class ActivityTemplateSerializer(serializers.HyperlinkedModelSerializer):
def update(self, instance, validated_data):
tags_data = validated_data.pop('tags')
[setattr(instance, key, value) for key, value in validated_data.items()]
[setattr(instance, key, value)
for key, value in validated_data.items()]
instance.save()
tags = [ActivityTag.objects.get_or_create(**tag_data)[0]
for tag_data in tags_data]
instance.tags = tags
return instance
return instance
def create(self, validated_data):
tags_data = validated_data.pop('tags')
activity_template = ActivityTemplate.objects.create(**validated_data)
tags = [ ActivityTag.objects.get_or_create(**tag_data)[0]
tags = [ActivityTag.objects.get_or_create(**tag_data)[0]
for tag_data in tags_data]
activity_template.tags = tags
return activity_template

View file

@ -1,9 +1,10 @@
from django.contrib.auth import get_user_model
from rest_framework.viewsets import ModelViewSet
from rest_framework.viewsets import ModelViewSet, ViewSet
from api.event.serializers import EventSerializer, PlaceSerializer,\
ActivityTagSerializer, ActivityTemplateSerializer
from api.event.mixins import EventNestedMixin
from event.models import Event, Place, ActivityTag, ActivityTemplate
User = get_user_model()
@ -32,6 +33,21 @@ class PlaceViewSet(ModelViewSet):
serializer_class = PlaceSerializer
class PlaceEventViewSet(ModelViewSet):
queryset = Place.objects.all()
serializer_class = PlaceSerializer
def list(self, request, event_pk=None):
queryset = self.queryset.filter(event=event_pk)
page = self.paginate_queryset(queryset)
if page is not None:
serializer = self.get_serializer(page, many=True)
return self.get_paginated_response(serializer.data)
serializer = self.get_serializer(queryset, many=True)
return Response(serializer.data)
class ActivityTagViewSet(ModelViewSet):
"""
This viewset automatically provides `list`, `create`, `retrieve`,