X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/ae60b2a3949e96357477cc04f90fd0873cee8a92..114d7edc5dab67cc91565960a6e8bbc040563901:/src/social/views.py diff --git a/src/social/views.py b/src/social/views.py index 49c9b708a..757b5a26d 100644 --- a/src/social/views.py +++ b/src/social/views.py @@ -13,7 +13,6 @@ from catalogue.models import Book from ssify import ssi_included from social import forms from .models import Cite -from social.utils import get_set, likes, set_sets # ==================== @@ -26,9 +25,8 @@ def like_book(request, slug): if not request.user.is_authenticated(): return HttpResponseForbidden('Login required.') book = get_object_or_404(Book, slug=slug) - if not likes(request.user, book): - tag = get_set(request.user, '') - set_sets(request.user, book, [tag]) + + book.like(request.user) if request.is_ajax(): return JsonResponse({"success": True, "msg": "ok", "like": True}) @@ -36,10 +34,25 @@ def like_book(request, slug): return redirect(book) +@require_POST +def unlike_book(request, slug): + if not request.user.is_authenticated(): + return HttpResponseForbidden('Login required.') + book = get_object_or_404(Book, slug=slug) + + book.unlike(request.user) + + if request.is_ajax(): + return JsonResponse({"success": True, "msg": "ok", "like": False}) + else: + return redirect(book) + + @login_required def my_shelf(request): - books = Book.tagged.with_any(request.user.tag_set.all()) - return render(request, 'social/my_shelf.html', locals()) + return render(request, 'social/my_shelf.html', { + 'books': Book.tagged.with_any(request.user.tag_set.all()) + }) class ObjectSetsFormView(AjaxableFormView): @@ -59,20 +72,6 @@ class ObjectSetsFormView(AjaxableFormView): return (obj, request.user), {} -@require_POST -def unlike_book(request, slug): - if not request.user.is_authenticated(): - return HttpResponseForbidden('Login required.') - book = get_object_or_404(Book, slug=slug) - if likes(request.user, book): - set_sets(request.user, book, []) - - if request.is_ajax(): - return JsonResponse({"success": True, "msg": "ok", "like": False}) - else: - return redirect(book) - - @ssi_included def cite(request, pk, main=False): cite = get_object_or_404(Cite, pk=pk)