add tag ids

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

diff --git a/src/social/views.py b/src/social/views.py
index 8966159..0ff0771 100644 (file)
--- a/src/social/views.py
+++ b/src/social/views.py
@@ -1,15 +1,15 @@
-# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
-# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
 #
 from django.shortcuts import render, get_object_or_404, redirect
 from django.http import HttpResponseForbidden, JsonResponse
 from django.contrib.auth.decorators import login_required
+from django.views.decorators.cache import never_cache
 from django.views.decorators.http import require_POST
-
-from ajaxable.utils import AjaxableFormView
+from django.views.generic.edit import FormView
 
 from catalogue.models import Book
-from social import forms
+from social import forms, models
 from wolnelektury.utils import is_ajax
 
 
@@ -18,13 +18,14 @@ from wolnelektury.utils import is_ajax
 # ====================
 
 
-@require_POST
+@login_required
 def like_book(request, slug):
-    if not request.user.is_authenticated:
-        return HttpResponseForbidden('Login required.')
     book = get_object_or_404(Book, slug=slug)
 
-    book.like(request.user)
+    if request.method != 'POST':
+        return redirect(book)
+
+    models.UserList.like(request.user, book)
 
     if is_ajax(request):
         return JsonResponse({"success": True, "msg": "ok", "like": True})
@@ -32,13 +33,31 @@ def like_book(request, slug):
         return redirect(book)
 
 
-@require_POST
+class AddSetView(FormView):
+    form_class = forms.AddSetForm
+    template_name = 'forms/form_detail.html'
+
+    def form_valid(self, form):
+        book, tag = form.save(self.request.user)
+
+        if is_ajax(self.request):
+            return JsonResponse(get_sets_for_book_ids([book.id], self.request.user))
+        else:
+            return redirect(book)
+
+
+class RemoveSetView(AddSetView):
+    form_class = forms.RemoveSetForm
+
+
+@login_required
 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.method != 'POST':
+        return redirect(book)
+
+    models.UserList.unlike(request.user, book)
 
     if is_ajax(request):
         return JsonResponse({"success": True, "msg": "ok", "like": False})
@@ -48,23 +67,68 @@ def unlike_book(request, slug):
 
 @login_required
 def my_shelf(request):
-    return render(request, 'social/my_shelf.html', {
-        'books': Book.tagged.with_any(request.user.tag_set.all())
+    template_name = 'social/my_shelf.html'
+    ulists = list(request.user.userlist_set.all())
+    suggest = [t for t in ulists if t.name]
+        
+    return render(request, template_name, {
+        'tags': ulists,
+        'books': Book.objects_filter(userlistitem__list__user=request.user),
+        'suggest': suggest,
     })
 
 
-class ObjectSetsFormView(AjaxableFormView):
-    form_class = forms.ObjectSetsForm
-    placeholdize = True
-    template = 'social/sets_form.html'
-    ajax_redirect = True
-    POST_login = True
-
-    def get_object(self, request, slug):
-        return get_object_or_404(Book, slug=slug)
+def get_sets_for_book_ids(book_ids, user):
+    data = {}
+    tagged = models.UserListItem.objects.filter(
+        list__user=user,
+        book_id__in=book_ids
+    ).order_by('list__name')
+    for t in tagged:
+        item = data.setdefault(t.book_id, [])
+        if t.list.name:
+            item.append({
+                "slug": t.list.slug,
+                "url": t.list.get_absolute_url(),
+                "name": t.list.name,
+            })
+    for b in book_ids:
+        if b not in data:
+            data[b] = None
+    return data
+    
+    
+
+
+@never_cache
+def my_liked(request):
+    if not request.user.is_authenticated:
+        return JsonResponse({})
+    try:
+        ids = [int(x) for x in request.GET.get('ids', '').split(',')]
+    except:
+        return JsonResponse({})
+    return JsonResponse(get_sets_for_book_ids(ids, request.user))
 
-    def context_description(self, request, obj):
-        return obj.pretty_title()
 
-    def form_args(self, request, obj):
-        return (obj, request.user), {}
+@never_cache
+@login_required
+def my_tags(request):
+    term = request.GET.get('term', '')
+    tags = models.UserList.objects.filter(user=request.user).order_by('name')
+    if term:
+        ulists = tags.filter(name__icontains=term)
+    return JsonResponse(
+        [
+            ul.name for ul in ulists
+        ], safe=False
+    )
+
+
+def confirm_user(request, key):
+    uc = get_object_or_404(models.UserConfirmation, key=key)
+    user = uc.user
+    uc.use()
+    return render(request, 'social/user_confirmation.html', {
+        'user': user,
+    })