Some new api.

[wolnelektury.git] / src / social / api / views.py

diff --git a/src/social/api/views.py b/src/social/api/views.py
index a299304..2c81286 100644 (file)
--- a/src/social/api/views.py
+++ b/src/social/api/views.py
@@ -12,6 +12,7 @@ from catalogue.api.helpers import order_books, books_after
 from catalogue.api.serializers import BookSerializer
 from catalogue.models import Book
 from social.utils import likes
 from catalogue.api.serializers import BookSerializer
 from catalogue.models import Book
 from social.utils import likes

+from social.views import get_sets_for_book_ids

 
 
 @vary_on_auth
 
 
 @vary_on_auth


@@ -32,6 +33,21 @@ class LikeView(APIView):
         return Response({})
 
 
         return Response({})
 
 

+@vary_on_auth
+class LikesView(APIView):
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request):
+        slugs = request.GET.getlist('slug')
+        books = Book.objects.filter(slug__in=slugs)
+        books = {b.id: b.slug for b in books}
+        ids = books.keys()
+        res = get_sets_for_book_ids(ids, request.user)
+        res = {books[bid]: v for bid, v in res.items()}
+        return Response(res)
+        
+
+

 @vary_on_auth
 class ShelfView(ListAPIView):
     permission_classes = [IsAuthenticated]
 @vary_on_auth
 class ShelfView(ListAPIView):
     permission_classes = [IsAuthenticated]