duck-booking-tool/api/views.py

# -*- coding: utf-8 -*-
"""
Views for the Duck Booking Tool API
"""

import re

from django.conf import settings
from django.shortcuts import get_object_or_404
from rest_framework import viewsets
from rest_framework.decorators import detail_route, list_route
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response

from .serializers import DuckSerializer, CompetenceSerializer
from booking.models import Duck, Competence, Booking, Species, Location

class DuckViewSet(viewsets.ModelViewSet):
    """
    View set for duck handling
    """

    serializer_class = DuckSerializer
    queryset = Duck.objects.all()

    @detail_route(methods=['post'], permission_classes=[IsAuthenticated])
    def book(self, request, pk=None):
        """
        API call to book a duck
        """

        duck = self.get_object()
        competence = get_object_or_404(Competence, pk=request.data['competence'])
        force = request.data.get('force', False)

        # If the duck is already booked, return 'already-booked' as the
        # result
        if duck.booked_by() != None:
            return Response({'status': 'already-booked'})

        # Result 'fail' means a problem
        result = 'fail'
        comp_level = 0

        # Check if the duck has the requested competence
        dcomp_list = duck.competences.filter(comp=competence)

        if len(dcomp_list) < 1:
            comp_level = 0
        else:
            comp_level = dcomp_list[0].level()

        # If the competence level is too low, set result to 'bad-comp'
        if comp_level <= settings.COMP_WARN_LEVEL:
            result = 'bad-comp'

        # If the duck has high enough competence or the booking is
        # forced, return status 'success'
        if result != 'bad-comp' or force:
            result = 'ok'

            booking = Booking(duck=duck, user=request.user, comp_req=competence)
            booking.save()

        return Response({'status': result})

    @list_route(methods=['post'], permission_classes=[IsAuthenticated])
    def donate(self, request):
        """
        API call to donate a new duck
        """

        color = request.data.get('color')
        species_id = request.data.get('species')
        location_id = request.data.get('location')

        if None in (color, species_id, location_id):
            return Response({'status': 'incomplete-request'}, status=400)

        try:
            species = Species.objects.get(pk=species_id)
        except Species.DoesNotExist:
            return Response({'status': 'bad-species'}, status=404)

        try:
            location = Location.objects.get(pk=location_id)
        except Location.DoesNotExist:
            return Response({'status': 'bad-location'}, status=404)

        if not re.match(r'^[0-9a-f]{6}$', color, flags=re.IGNORECASE):
            return Response({'status': 'bad-color'}, status=400)

        color = color.lower()

        duck = Duck.objects.create(donated_by=request.user,
                                   species=species,
                                   location=location,
                                   color=color)

        return Response({'status': 'ok', 'id': duck.pk})

class CompetenceViewSet(viewsets.ModelViewSet):
    """
    View set for competence handling
    """

    serializer_class = CompetenceSerializer
    queryset = Competence.objects.all()