from rest_framework import viewsets, permissions
from .models import QCResult
from .serializers import QCResultSerializer, QCResultCreateSerializer


class QCResultViewSet(viewsets.ModelViewSet):
    queryset = QCResult.objects.all()
    permission_classes = [permissions.IsAuthenticated]

    def get_serializer_class(self):
        if self.action == "create":
            return QCResultCreateSerializer
        return QCResultSerializer