23 lines
810 B
Python
23 lines
810 B
Python
from rest_framework import viewsets
|
|
|
|
from .models import Leerplein, SubgroepGroep
|
|
from .serializers import LeerpleinSerializer, SubgroepGroepSerializer
|
|
|
|
|
|
class LeerpleinViewSet(viewsets.ModelViewSet):
|
|
queryset = Leerplein.objects.prefetch_related("groepen", "subgroepen", "leerlingen").all()
|
|
serializer_class = LeerpleinSerializer
|
|
|
|
|
|
class SubgroepGroepViewSet(viewsets.ModelViewSet):
|
|
serializer_class = SubgroepGroepSerializer
|
|
|
|
def get_queryset(self):
|
|
qs = SubgroepGroep.objects.select_related("groep", "subgroep").all()
|
|
groep = self.request.query_params.get("groep")
|
|
if groep:
|
|
qs = qs.filter(groep_id=groep)
|
|
subgroep = self.request.query_params.get("subgroep")
|
|
if subgroep:
|
|
qs = qs.filter(subgroep_id=subgroep)
|
|
return qs
|