from rest_framework import viewsets

from .serializers import CategorySerializer, CourseSerializer
from . import ExtendedModelViewSet

from apps.course.models import Category, Course


class CategoryViewSet(ExtendedModelViewSet):
    queryset = Category.objects.all()
    serializer_class = CategorySerializer


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