35 lines
1.4 KiB
Python
35 lines
1.4 KiB
Python
from django.shortcuts import get_object_or_404
|
|
|
|
from rest_framework import views, permissions, generics
|
|
from rest_framework.response import Response
|
|
|
|
from core.apps.accounts.models import Like
|
|
from core.apps.products.models import Product
|
|
from core.apps.products.serializers.product import ProductListSerializer
|
|
|
|
|
|
class LikeApiView(views.APIView):
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def get(self, request, product_id):
|
|
product = get_object_or_404(Product, id=product_id)
|
|
like, created = Like.objects.get_or_create(product=product, user=request.user)
|
|
if created:
|
|
return Response({'success': True, 'message': "Liked"}, status=200)
|
|
like.delete()
|
|
return Response({'success': False, 'message': "Unliked"}, status=200)
|
|
|
|
|
|
class LikeProductListApiView(generics.GenericAPIView):
|
|
serializer_class = ProductListSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def get(self, request):
|
|
products = Product.objects.filter(likes__user=request.user).distinct()
|
|
page = self.paginate_queryset(products)
|
|
if page is not None:
|
|
serializer = self.serializer_class(page, many=True, context={'user': request.user})
|
|
return self.get_paginated_response(serializer.data)
|
|
serializer = self.serializer_class(products, many=True, context={'user': request.user})
|
|
return Response(serializer.data, status=200)
|