HiPhone_BE/market/viewsets.py

from rest_framework.decorators import action
from rest_framework.permissions import AllowAny, IsAuthenticated
from rest_framework.response import Response
from rest_framework.viewsets import ModelViewSet

from core.mixins import ActionBasedMixin
from core.permissions import IsAuthorOrReadOnly, IsAdminUserOrReadOnly
from market.models import Brand, Product, Post
from market.serializers import (
    BrandSerializer,
    ProductSerializer,
    PostSerializer,
)


class BrandViewset(ActionBasedMixin, ModelViewSet):
    queryset = Brand.objects.all()
    serializer_class = BrandSerializer
    permission_classes = [IsAdminUserOrReadOnly]
    pagination_class = None


class ProductViewset(ActionBasedMixin, ModelViewSet):
    queryset = (
        Product.objects.all()
        .select_related("brand")
        .prefetch_related("storage")
    )
    serializer_class = ProductSerializer
    serializer_class_map = {
        "posts": PostSerializer,
    }
    permission_classes = [IsAdminUserOrReadOnly]

    @action(detail=True, methods=["GET"])
    def posts(self, request, pk):
        product = self.get_object()
        queryset = product.posts.all()

        page = self.paginate_queryset(queryset)
        if page is not None:
            serializer = self.get_serializer(page, many=True)
            return self.get_paginated_response(serializer.data)

        serializer = self.get_serializer(queryset, many=True)
        return Response(serializer.data)


class PostViewset(ActionBasedMixin, ModelViewSet):
    queryset = (
        Post.objects.all()
        .select_related("storage", "product", "author", "product__brand")
        .prefetch_related("images")
    )
    serializer_class = PostSerializer
    permission_classes = [IsAuthenticated, IsAuthorOrReadOnly]
    permission_classes_map = {
        "list": [AllowAny],
        "retrieve": [AllowAny],
    }

    def perform_create(self, serializer):
        serializer.save(author=self.request.user)