52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
from django.contrib.auth import get_user_model
|
|
|
|
from rest_framework.viewsets import ModelViewSet
|
|
|
|
from api.event.serializers import EventSerializer, PlaceSerializer,\
|
|
ActivityTagSerializer, ActivityTemplateSerializer
|
|
from event.models import Event, Place, ActivityTag, ActivityTemplate
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class EventViewSet(ModelViewSet):
|
|
"""
|
|
This viewset automatically provides `list`, `create`, `retrieve`,
|
|
`update` and `destroy` actions.
|
|
|
|
"""
|
|
queryset = Event.objects.all()
|
|
serializer_class = EventSerializer
|
|
|
|
def perform_create(self, serializer):
|
|
serializer.save(created_by=self.request.user)
|
|
|
|
|
|
class PlaceViewSet(ModelViewSet):
|
|
"""
|
|
This viewset automatically provides `list`, `create`, `retrieve`,
|
|
`update` and `destroy` actions.
|
|
|
|
"""
|
|
queryset = Place.objects.all()
|
|
serializer_class = PlaceSerializer
|
|
|
|
|
|
class ActivityTagViewSet(ModelViewSet):
|
|
"""
|
|
This viewset automatically provides `list`, `create`, `retrieve`,
|
|
`update` and `destroy` actions.
|
|
|
|
"""
|
|
queryset = ActivityTag.objects.all()
|
|
serializer_class = ActivityTagSerializer
|
|
|
|
|
|
class ActivityTemplateViewSet(ModelViewSet):
|
|
"""
|
|
This viewset automatically provides `list`, `create`, `retrieve`,
|
|
`update` and `destroy` actions.
|
|
|
|
"""
|
|
queryset = ActivityTemplate.objects.all()
|
|
serializer_class = ActivityTemplateSerializer
|