fix

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

diff --git a/src/social/views.py b/src/social/views.py
index 4aaf54e..8f27b87 100644 (file)
--- a/src/social/views.py
+++ b/src/social/views.py
@@ -24,7 +24,7 @@ def like_book(request, slug):
     book = get_object_or_404(Book, slug=slug)
 
     if request.method != 'POST':
     book = get_object_or_404(Book, slug=slug)
 
     if request.method != 'POST':

-        return redirect(book.get_absolute_url())
+        return redirect(book)

 
     book.like(request.user)
 
 
     book.like(request.user)
 


@@ -51,12 +51,13 @@ class RemoveSetView(AddSetView):
     form_class = forms.RemoveSetForm
 
 
     form_class = forms.RemoveSetForm
 
 

-@require_POST
+@login_required

 def unlike_book(request, slug):
 def unlike_book(request, slug):

-    if not request.user.is_authenticated:
-        return HttpResponseForbidden('Login required.')

     book = get_object_or_404(Book, slug=slug)
 
     book = get_object_or_404(Book, slug=slug)
 

+    if request.method != 'POST':
+        return redirect(book)
+

     book.unlike(request.user)
 
     if is_ajax(request):
     book.unlike(request.user)
 
     if is_ajax(request):