from rest_framework import viewsets

from .serializers import CourseSerializer
from . import ExtendedModelViewSet

from apps.course.models import Course


class CourseViewSet(ExtendedModelViewSet):
    queryset = Course.objects.all()
    serializer_class = CourseSerializer
    filter_fields = ('category',)