src/social/views.py

   1 # This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
   3 #
   4 from django.shortcuts import render, get_object_or_404, redirect
   5 from django.http import HttpResponseForbidden, JsonResponse
   6 from django.contrib.auth.decorators import login_required
   7 from django.views.decorators.cache import never_cache
   8 from django.views.decorators.http import require_POST
   9 from django.views.generic.edit import FormView
  10
  11 from catalogue.models import Book
  12 from social import forms, models
  13 from wolnelektury.utils import is_ajax
  14
  15
  16 # ====================
  17 # = Shelf management =
  18 # ====================
  19
  20
  21 @login_required
  22 def like_book(request, slug):
  23     book = get_object_or_404(Book, slug=slug)
  24
  25     if request.method != 'POST':
  26         return redirect(book)
  27
  28     models.UserList.like(request.user, book)
  29
  30     if is_ajax(request):
  31         return JsonResponse({"success": True, "msg": "ok", "like": True})
  32     else:
  33         return redirect(book)
  34
  35
  36 class AddSetView(FormView):
  37     form_class = forms.AddSetForm
  38     template_name = 'forms/form_detail.html'
  39
  40     def form_valid(self, form):
  41         book, tag = form.save(self.request.user)
  42
  43         if is_ajax(self.request):
  44             return JsonResponse(get_sets_for_book_ids([book.id], self.request.user))
  45         else:
  46             return redirect(book)
  47
  48
  49 class RemoveSetView(AddSetView):
  50     form_class = forms.RemoveSetForm
  51
  52
  53 @login_required
  54 def unlike_book(request, slug):
  55     book = get_object_or_404(Book, slug=slug)
  56
  57     if request.method != 'POST':
  58         return redirect(book)
  59
  60     models.UserList.unlike(request.user, book)
  61
  62     if is_ajax(request):
  63         return JsonResponse({"success": True, "msg": "ok", "like": False})
  64     else:
  65         return redirect(book)
  66
  67
  68 @login_required
  69 def my_shelf(request):
  70     template_name = 'social/my_shelf.html'
  71     ulists = list(request.user.userlist_set.all())
  72     suggest = [t for t in ulists if t.name]
  73
  74     return render(request, template_name, {
  75         'tags': ulists,
  76         'books': Book.objects_filter(userlistitem__list__user=request.user),
  77         'suggest': suggest,
  78     })
  79
  80
  81 def get_sets_for_book_ids(book_ids, user):
  82     data = {}
  83     tagged = models.UserListItem.objects.filter(
  84         list__user=user,
  85         book_id__in=book_ids
  86     ).order_by('list__name')
  87     for t in tagged:
  88         item = data.setdefault(t.book_id, [])
  89         if t.list.name:
  90             item.append({
  91                 "slug": t.list.slug,
  92                 "url": t.list.get_absolute_url(),
  93                 "name": t.list.name,
  94             })
  95     for b in book_ids:
  96         if b not in data:
  97             data[b] = None
  98     return data
  99
 100
 101
 102
 103 @never_cache
 104 def my_liked(request):
 105     if not request.user.is_authenticated:
 106         return JsonResponse({})
 107     try:
 108         ids = [int(x) for x in request.GET.get('ids', '').split(',')]
 109     except:
 110         return JsonResponse({})
 111     return JsonResponse(get_sets_for_book_ids(ids, request.user))
 112
 113
 114 @never_cache
 115 @login_required
 116 def my_tags(request):
 117     term = request.GET.get('term', '')
 118     tags = models.UserList.objects.filter(user=request.user).order_by('name')
 119     if term:
 120         ulists = tags.filter(name__icontains=term)
 121     return JsonResponse(
 122         [
 123             ul.name for ul in ulists
 124         ], safe=False
 125     )
 126
 127
 128 def confirm_user(request, key):
 129     uc = get_object_or_404(models.UserConfirmation, key=key)
 130     user = uc.user
 131     uc.use()
 132     return render(request, 'social/user_confirmation.html', {
 133         'user': user,
 134     })