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