from rest_framework.exceptions import NotFound

from university.api.base import Endpoint
from university.models import Exam


class ExamEndpoint(Endpoint):

    def convert_args(self, request, exam_id, *args, **kwargs):
        try:
            exam = Exam.objects.get(id=exam_id)
        except Exam.DoesNotExist:
            raise NotFound

        kwargs['exam'] = exam
        return (args, kwargs)