from rest_framework.exceptions import NotFound

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


class AdminCourseEndpoint(Endpoint):

    def convert_args(self, request, course_id, *args, **kwargs):
        try:
            course = Course.objects.get(id=course_id)
        except Exception:
            raise NotFound

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