from rest_framework.viewsets import ViewSet
from .models import Rubric
from django.shortcuts import get_object_or_404
from .serializers import RubricSerializer
from rest_framework.response import Response
from rest_framework import status

class RubricViewSet(ViewSet):
    def list(self, request):
        queryset = Rubric.objects.all()
        serializer = RubricSerializer(queryset, many=True)
        return Response(serializer.data, status=status.HTTP_200_OK)
    
    def retrieve(self, request, pk=None):
        queryset = Rubric.objects.filter(id=pk)
        serializer = RubricSerializer(queryset, many=True)
        return Response(serializer.data, status=status.HTTP_200_OK)