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     favs = request.user.userlist_set.filter(favorites=True)
  72     others = request.user.userlist_set.filter(favorites=False)
  73     ulists = list(request.user.userlist_set.all())
  74     suggest = [t for t in ulists if t.name]
  75
  76     return render(request, template_name, {
  77         'favs': favs,
  78         'others': others,
  79         #'books': Book.objects.filter(userlistitem__list__user=request.user),
  80     })
  81
  82
  83 def get_sets_for_book_ids(book_ids, user):
  84     data = {}
  85     tagged = models.UserListItem.objects.filter(
  86         list__user=user,
  87         book_id__in=book_ids
  88     ).order_by('list__name')
  89     for t in tagged:
  90         item = data.setdefault(t.book_id, [])
  91         if t.list.name:
  92             item.append({
  93                 "slug": t.list.slug,
  94                 "url": t.list.get_absolute_url(),
  95                 "name": t.list.name,
  96             })
  97     for b in book_ids:
  98         if b not in data:
  99             data[b] = None
 100     return data
 101
 102
 103
 104
 105 @never_cache
 106 def my_liked(request):
 107     if not request.user.is_authenticated:
 108         return JsonResponse({})
 109     try:
 110         ids = [int(x) for x in request.GET.get('ids', '').split(',')]
 111     except:
 112         return JsonResponse({})
 113     return JsonResponse(get_sets_for_book_ids(ids, request.user))
 114
 115
 116 @never_cache
 117 @login_required
 118 def my_tags(request):
 119     term = request.GET.get('term', '')
 120     tags = models.UserList.objects.filter(user=request.user).order_by('name')
 121     if term:
 122         ulists = tags.filter(name__icontains=term)
 123     return JsonResponse(
 124         [
 125             ul.name for ul in ulists
 126         ], safe=False
 127     )
 128
 129
 130 def confirm_user(request, key):
 131     uc = get_object_or_404(models.UserConfirmation, key=key)
 132     user = uc.user
 133     uc.use()
 134     return render(request, 'social/user_confirmation.html', {
 135         'user': user,
 136     })