# -*- coding:utf-8 -* # # Copyright 2016,2017 # - Skia # # Ce fichier fait partie du site de l'Association des Étudiants de l'UTBM, # http://ae.utbm.fr. # # This program is free software; you can redistribute it and/or modify it under # the terms of the GNU General Public License a published by the Free Software # Foundation; either version 3 of the License, or (at your option) any later # version. # # This program is distributed in the hope that it will be useful, but WITHOUT # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS # FOR A PARTICULAR PURPOSE. See the GNU General Public License for more # details. # # You should have received a copy of the GNU General Public License along with # this program; if not, write to the Free Sofware Foundation, Inc., 59 Temple # Place - Suite 330, Boston, MA 02111-1307, USA. # # from rest_framework import serializers from rest_framework.response import Response from rest_framework.decorators import action from launderette.models import Launderette, Machine, Token from api.views import RightModelViewSet class LaunderettePlaceSerializer(serializers.ModelSerializer): machine_list = serializers.ListField( child=serializers.IntegerField(), read_only=True ) token_list = serializers.ListField(child=serializers.IntegerField(), read_only=True) class Meta: model = Launderette fields = ( "id", "name", "counter", "machine_list", "token_list", "get_absolute_url", ) class LaunderetteMachineSerializer(serializers.ModelSerializer): class Meta: model = Machine fields = ("id", "name", "type", "is_working", "launderette") class LaunderetteTokenSerializer(serializers.ModelSerializer): class Meta: model = Token fields = ( "id", "name", "type", "launderette", "borrow_date", "user", "is_avaliable", ) class LaunderettePlaceViewSet(RightModelViewSet): """ Manage Launderette (api/v1/launderette/place/) """ serializer_class = LaunderettePlaceSerializer queryset = Launderette.objects.all() class LaunderetteMachineViewSet(RightModelViewSet): """ Manage Washing Machines (api/v1/launderette/machine/) """ serializer_class = LaunderetteMachineSerializer queryset = Machine.objects.all() class LaunderetteTokenViewSet(RightModelViewSet): """ Manage Launderette's tokens (api/v1/launderette/token/) """ serializer_class = LaunderetteTokenSerializer queryset = Token.objects.all() @action(detail=False) def washing(self, request): """ Return all washing tokens (api/v1/launderette/token/washing) """ self.queryset = self.queryset.filter(type="WASHING") serializer = self.get_serializer(self.queryset, many=True) return Response(serializer.data) @action(detail=False) def drying(self, request): """ Return all drying tokens (api/v1/launderette/token/drying) """ self.queryset = self.queryset.filter(type="DRYING") serializer = self.get_serializer(self.queryset, many=True) return Response(serializer.data) @action(detail=False) def avaliable(self, request): """ Return all avaliable tokens (api/v1/launderette/token/avaliable) """ self.queryset = self.queryset.filter( borrow_date__isnull=True, user__isnull=True ) serializer = self.get_serializer(self.queryset, many=True) return Response(serializer.data) @action(detail=False) def unavaliable(self, request): """ Return all unavaliable tokens (api/v1/launderette/token/unavaliable) """ self.queryset = self.queryset.filter( borrow_date__isnull=False, user__isnull=False ) serializer = self.get_serializer(self.queryset, many=True) return Response(serializer.data)