from rest_framework import viewsets

from .serializers import CourseSerializer
from . import ExtendedModelViewSet

from apps.course.models import Course


class CourseViewSet(ExtendedModelViewSet):
    queryset = Course.objects.select_related(
        'author', 'category'
    ).prefetch_related(
        'likes', 'materials'
    ).all()
    serializer_class = CourseSerializer
    filter_fields = ('category',)