Better tables

apps/wei/api/serializers.py

@@ -3,7 +3,7 @@
 
 from rest_framework import serializers
 
-from ..models import WEIClub, Bus, BusTeam
+from ..models import WEIClub, Bus, BusTeam, WEIRole, WEIRegistration, WEIMembership
 
 
 class WEIClubSerializer(serializers.ModelSerializer):
@@ -37,3 +37,36 @@ class BusTeamSerializer(serializers.ModelSerializer):
     class Meta:
         model = BusTeam
         fields = '__all__'
+
+
+class WEIRoleSerializer(serializers.ModelSerializer):
+    """
+    REST API Serializer for WEI roles.
+    The djangorestframework plugin will analyse the model `WEIRole` and parse all fields in the API.
+    """
+
+    class Meta:
+        model = WEIRole
+        fields = '__all__'
+
+
+class WEIRegistrationSerializer(serializers.ModelSerializer):
+    """
+    REST API Serializer for WEI registrations.
+    The djangorestframework plugin will analyse the model `WEIRegistration` and parse all fields in the API.
+    """
+
+    class Meta:
+        model = WEIRegistration
+        fields = '__all__'
+
+
+class WEIMembershipSerializer(serializers.ModelSerializer):
+    """
+    REST API Serializer for WEI memberships.
+    The djangorestframework plugin will analyse the model `WEIMembership` and parse all fields in the API.
+    """
+
+    class Meta:
+        model = WEIMembership
+        fields = '__all__'

apps/wei/api/urls.py

@@ -1,7 +1,8 @@
 # Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
 # SPDX-License-Identifier: GPL-3.0-or-later
 
-from .views import WEIClubViewSet, BusViewSet, BusTeamViewSet
+from .views import WEIClubViewSet, BusViewSet, BusTeamViewSet, WEIRoleViewSet, WEIRegistrationViewSet, \
+    WEIMembershipViewSet
 
 
 def register_wei_urls(router, path):
@@ -11,3 +12,6 @@ def register_wei_urls(router, path):
     router.register(path + '/club', WEIClubViewSet)
     router.register(path + '/bus', BusViewSet)
     router.register(path + '/team', BusTeamViewSet)
+    router.register(path + '/role', WEIRoleViewSet)
+    router.register(path + '/registration', WEIRegistrationViewSet)
+    router.register(path + '/membership', WEIMembershipViewSet)

apps/wei/api/views.py

@@ -4,8 +4,9 @@ from django_filters.rest_framework import DjangoFilterBackend
 from rest_framework.filters import SearchFilter
 from api.viewsets import ReadProtectedModelViewSet
 
-from .serializers import WEIClubSerializer, BusSerializer, BusTeamSerializer
-from ..models import WEIClub, Bus, BusTeam
+from .serializers import WEIClubSerializer, BusSerializer, BusTeamSerializer, WEIRoleSerializer, \
+    WEIRegistrationSerializer, WEIMembershipSerializer
+from ..models import WEIClub, Bus, BusTeam, WEIRole, WEIRegistration, WEIMembership
 
 
 class WEIClubViewSet(ReadProtectedModelViewSet):
@@ -27,7 +28,7 @@ class BusViewSet(ReadProtectedModelViewSet):
     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()
+    queryset = Bus.objects
     serializer_class = BusSerializer
     filter_backends = [SearchFilter, DjangoFilterBackend]
     search_fields = ['$name', ]
@@ -40,8 +41,46 @@ class BusTeamViewSet(ReadProtectedModelViewSet):
     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()
+    queryset = BusTeam.objects
     serializer_class = BusTeamSerializer
     filter_backends = [SearchFilter, DjangoFilterBackend]
     search_fields = ['$name', ]
     filterset_fields = ['name', 'bus', 'bus__wei', ]
+
+
+class WEIRoleViewSet(ReadProtectedModelViewSet):
+    """
+    REST API View set.
+    The djangorestframework plugin will get all `WEIRole` objects, serialize it to JSON with the given serializer,
+    then render it on /api/wei/role/
+    """
+    queryset = WEIRole.objects
+    serializer_class = WEIRoleSerializer
+    filter_backends = [SearchFilter]
+    search_fields = ['$name', ]
+
+
+class WEIRegistrationViewSet(ReadProtectedModelViewSet):
+    """
+    REST API View set.
+    The djangorestframework plugin will get all WEIRegistration objects, serialize it to JSON with the given serializer,
+    then render it on /api/wei/registration/
+    """
+    queryset = WEIRegistration.objects
+    serializer_class = WEIRegistrationSerializer
+    filter_backends = [SearchFilter, DjangoFilterBackend]
+    search_fields = ['$user__username', ]
+    filterset_fields = ['user', 'wei', ]
+
+
+class WEIMembershipViewSet(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/membership/
+    """
+    queryset = WEIMembership.objects
+    serializer_class = WEIMembershipSerializer
+    filter_backends = [SearchFilter, DjangoFilterBackend]
+    search_fields = ['$user__username', '$bus__name', '$team__name', ]
+    filterset_fields = ['user', 'club', 'bus', 'team', ]