views.py

from rest_framework.viewsets import GenericViewSet, mixins

from main_app.serializers import (
    CategorySerializer,
    UserSerializer,
)
from main_app.models import (
    Category, User
)


class CategoryView(GenericViewSet,
                   mixins.ListModelMixin,
                   mixins.RetrieveModelMixin,
                   mixins.CreateModelMixin):
    serializer_class = CategorySerializer
    queryset = Category.objects.all()


class UserView(GenericViewSet,
               mixins.ListModelMixin,
               mixins.RetrieveModelMixin,
               mixins.CreateModelMixin):
    serializer_class = UserSerializer
    queryset = User.objects.all()