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