from rest_framework import viewsets
from .models import ProgrammedReport
from .serializers import ProgrammedReportSerializer
from .permissions import IsOmixomUser


class ProgrammedReportViewSet(viewsets.ModelViewSet):
    # El ViewSet solo es valido si el Token enviado es de un usuario de Omixom.
    permission_classes = [IsOmixomUser]
    serializer_class = ProgrammedReportSerializer

    # Solo se devuelven los modelos para el user_id asociado al token. (Ver IsOmixomUser)
    def get_queryset(self):
        user_id = self.request.data["user_id"]
        qs = ProgrammedReport.objects.filter(user_id=user_id)
        return qs