diff --git a/server/proj/rubricks/views.py b/server/proj/rubricks/views.py index e807b86..97c9bcb 100644 --- a/server/proj/rubricks/views.py +++ b/server/proj/rubricks/views.py @@ -1,5 +1,6 @@ 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 @@ -9,5 +10,10 @@ class RubricViewSet(ViewSet): 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) + + return Response(serializer.data, status=status.HTTP_200_OK)