Add a bus

apps/wei/api/serializers.py

@@ -3,7 +3,7 @@
 
 from rest_framework import serializers
 
-from ..models import WEIClub
+from ..models import WEIClub, Bus, BusTeam
 
 
 class WEIClubSerializer(serializers.ModelSerializer):
@@ -15,3 +15,25 @@ class WEIClubSerializer(serializers.ModelSerializer):
     class Meta:
         model = WEIClub
         fields = '__all__'
+
+
+class BusSerializer(serializers.ModelSerializer):
+    """
+    REST API Serializer for Bus.
+    The djangorestframework plugin will analyse the model `Bus` and parse all fields in the API.
+    """
+
+    class Meta:
+        model = Bus
+        fields = '__all__'
+
+
+class BusTeamSerializer(serializers.ModelSerializer):
+    """
+    REST API Serializer for Bus teams.
+    The djangorestframework plugin will analyse the model `BusTeam` and parse all fields in the API.
+    """
+
+    class Meta:
+        model = BusTeam
+        fields = '__all__'

apps/wei/api/urls.py

@@ -1,11 +1,13 @@
 # Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
 # SPDX-License-Identifier: GPL-3.0-or-later
 
-from .views import WEIClubViewSet
+from .views import WEIClubViewSet, BusViewSet, BusTeamViewSet
 
 
 def register_wei_urls(router, path):
     """
     Configure router for Member REST API.
     """
-    router.register(path + '/club/', WEIClubViewSet)
+    router.register(path + '/club', WEIClubViewSet)
+    router.register(path + '/bus', BusViewSet)
+    router.register(path + '/team', BusTeamViewSet)

apps/wei/api/views.py

@@ -4,8 +4,8 @@
 from rest_framework.filters import SearchFilter
 from api.viewsets import ReadProtectedModelViewSet
 
-from .serializers import WEIClubSerializer
-from ..models import WEIClub
+from .serializers import WEIClubSerializer, BusSerializer, BusTeamSerializer
+from ..models import WEIClub, Bus, BusTeam
 
 
 class WEIClubViewSet(ReadProtectedModelViewSet):
@@ -18,3 +18,27 @@ class WEIClubViewSet(ReadProtectedModelViewSet):
     serializer_class = WEIClubSerializer
     filter_backends = [SearchFilter]
     search_fields = ['$name', ]
+
+
+class BusViewSet(ReadProtectedModelViewSet):
+    """
+    REST API View set.
+    The djangorestframework plugin will get all `Bus` objects, serialize it to JSON with the given serializer,
+    then render it on /api/wei/bus/
+    """
+    queryset = Bus.objects.all()
+    serializer_class = BusSerializer
+    filter_backends = [SearchFilter]
+    search_fields = ['$name', ]
+
+
+class BusTeamViewSet(ReadProtectedModelViewSet):
+    """
+    REST API View set.
+    The djangorestframework plugin will get all `BusTeam` objects, serialize it to JSON with the given serializer,
+    then render it on /api/wei/team/
+    """
+    queryset = BusTeam.objects.all()
+    serializer_class = BusTeamSerializer
+    filter_backends = [SearchFilter]
+    search_fields = ['$name', ]