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