from rest_framework.exceptions import NotFound

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


class MyOnlineExamEndpoint(Endpoint):

    def convert_args(self, request, exam_slug, *args, **kwargs):
        try:
            my_onlineexam = UserOnlineExam.objects.get(
                user=request.user,
                online_exam__slug=exam_slug
            )
        except UserOnlineExam.DoesNotExist:
            raise NotFound

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