views.py

from rest_framework import generics, status
from rest_framework.response import Response
from rest_framework.viewsets import ModelViewSet
from project.organisations.models import Organisation, Membership
from project.organisations.serializers import (
    CreateOrganisationSerializer, OrganisationMembersSerializer,
    RetrieveOrganisationSerializer, OrganisationDetailSerializer
)


class RetrieveOrganisationView(generics.RetrieveAPIView):

    serializer_class = RetrieveOrganisationSerializer


class CreateOrganisationView(generics.CreateAPIView):

    serializer_class = CreateOrganisationSerializer


class OrganisationMembersView(generics.ListAPIView):

    serializer_class = OrganisationMembersSerializer

    def get_queryset(self):
        organisation = Organisation.objects.order_by('?').first()
        return Membership.objects.filter(organisation=organisation)


class LeaveOrganisationView(generics.DestroyAPIView):

    def get_object(self):
        return Membership.objects.order_by('?').first()

    def delete(self, request, *args, **kwargs):
        instance = self.get_object()
        self.perform_destroy(instance)
        return Response(status=status.HTTP_204_NO_CONTENT)


class OrganisationViewSet(ModelViewSet):
    queryset = Organisation.objects.all()
    serializer_class = OrganisationDetailSerializer