diff --git a/apps/product/views/common.py b/apps/product/views/common.py index e8f3a966..daa46fd7 100644 --- a/apps/product/views/common.py +++ b/apps/product/views/common.py @@ -10,14 +10,15 @@ from comment.serializers import CommentRUDSerializer class ProductBaseView(generics.GenericAPIView): """Product base view""" - permission_classes = (permissions.AllowAny, ) + permission_classes = (permissions.AllowAny,) def get_queryset(self): """Override get_queryset method.""" return Product.objects.published() \ - .with_base_related() \ - .by_country_code(self.request.country_code) \ - .order_by('-created') + .with_base_related() \ + .annotate_in_favorites(self.request.user) \ + .by_country_code(self.request.country_code) \ + .order_by('-created') class ProductListView(ProductBaseView, generics.ListAPIView): @@ -67,9 +68,9 @@ class ProductCommentListView(generics.ListAPIView): """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') + model='product') \ + .by_object_id(object_id=product.pk) \ + .order_by('-created') class ProductCommentRUDView(generics.RetrieveUpdateDestroyAPIView):