98 lines
3.4 KiB
Python
98 lines
3.4 KiB
Python
"""Product app views."""
|
|
from rest_framework import generics, permissions
|
|
from django.shortcuts import get_object_or_404
|
|
from product.models import Product
|
|
from comment.models import Comment
|
|
from product import serializers
|
|
from product import filters
|
|
from comment.serializers import CommentRUDSerializer
|
|
|
|
|
|
class ProductBaseView(generics.GenericAPIView):
|
|
"""Product base view"""
|
|
permission_classes = (permissions.AllowAny,)
|
|
|
|
def get_queryset(self):
|
|
"""Override get_queryset method."""
|
|
return Product.objects.published() \
|
|
.with_base_related() \
|
|
.annotate_in_favorites(self.request.user) \
|
|
.by_country_code(self.request.country_code) \
|
|
.order_by('-created')
|
|
|
|
|
|
class ProductListView(ProductBaseView, generics.ListAPIView):
|
|
"""List view for model Product."""
|
|
serializer_class = serializers.ProductBaseSerializer
|
|
filter_class = filters.ProductFilterSet
|
|
|
|
def get_queryset(self):
|
|
qs = super().get_queryset().with_extended_related()
|
|
return qs
|
|
|
|
|
|
class ProductDetailView(ProductBaseView, generics.RetrieveAPIView):
|
|
"""Detail view fro model Product."""
|
|
lookup_field = 'slug'
|
|
serializer_class = serializers.ProductDetailSerializer
|
|
|
|
|
|
class CreateFavoriteProductView(generics.CreateAPIView,
|
|
generics.DestroyAPIView):
|
|
"""View for create/destroy product in favorites."""
|
|
serializer_class = serializers.ProductFavoritesCreateSerializer
|
|
lookup_field = 'slug'
|
|
|
|
def get_object(self):
|
|
"""
|
|
Returns the object the view is displaying.
|
|
"""
|
|
product = get_object_or_404(Product, slug=self.kwargs['slug'])
|
|
favorites = get_object_or_404(product.favorites.filter(user=self.request.user))
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, favorites)
|
|
return favorites
|
|
|
|
|
|
class ProductCommentCreateView(generics.CreateAPIView):
|
|
"""View for create new comment."""
|
|
lookup_field = 'slug'
|
|
serializer_class = serializers.ProductCommentCreateSerializer
|
|
queryset = Comment.objects.all()
|
|
|
|
|
|
class ProductCommentListView(generics.ListAPIView):
|
|
"""View for return list of product comments."""
|
|
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = serializers.ProductCommentCreateSerializer
|
|
|
|
def get_queryset(self):
|
|
"""Override get_queryset method"""
|
|
product = get_object_or_404(Product, slug=self.kwargs['slug'])
|
|
return Comment.objects.by_content_type(app_label='product',
|
|
model='product') \
|
|
.by_object_id(object_id=product.pk) \
|
|
.order_by('-created')
|
|
|
|
|
|
class ProductCommentRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""View for retrieve/update/destroy product comment."""
|
|
serializer_class = CommentRUDSerializer
|
|
queryset = Product.objects.all()
|
|
|
|
def get_object(self):
|
|
"""Returns the object the view is displaying."""
|
|
queryset = self.filter_queryset(self.get_queryset())
|
|
|
|
product_obj = get_object_or_404(queryset,
|
|
slug=self.kwargs['slug'])
|
|
comment_obj = get_object_or_404(product_obj.comments.by_user(self.request.user),
|
|
pk=self.kwargs['comment_id'])
|
|
|
|
# May raise a permission denied
|
|
self.check_object_permissions(self.request, comment_obj)
|
|
|
|
return comment_obj
|