''' book list views''' from django.contrib.auth.decorators import login_required from django.db.models import Q from django.http import HttpResponseNotFound from django.shortcuts import get_object_or_404, redirect from django.template.response import TemplateResponse from django.utils.decorators import method_decorator from django.views import View from bookwyrm import forms, models from bookwyrm.activitypub import ActivitypubResponse from .helpers import is_api_request, object_visible_to_user, privacy_filter # pylint: disable=no-self-use class Lists(View): ''' book list page ''' def get(self, request): ''' display a book list ''' user = request.user if request.user.is_authenticated else None lists = models.List.objects.filter( ~Q(user=user), ).all() lists = privacy_filter(request.user, lists, ['public', 'followers']) data = { 'title': 'Lists', 'lists': lists, 'list_form': forms.ListForm() } return TemplateResponse(request, 'lists/lists.html', data) @method_decorator(login_required, name='dispatch') # pylint: disable=unused-argument def post(self, request): ''' create a book_list ''' form = forms.ListForm(request.POST) if not form.is_valid(): print(form.errors) return redirect('lists') book_list = form.save() return redirect(book_list.local_path) class List(View): ''' book list page ''' def get(self, request, list_id): ''' display a book list ''' book_list = get_object_or_404(models.List, id=list_id) if not object_visible_to_user(request.user, book_list): return HttpResponseNotFound() if is_api_request(request): return ActivitypubResponse(book_list.to_activity()) suggestions = request.user.shelfbook_set.all().filter( ~Q(book__in=book_list.books) ) data = { 'title': '%s | Lists' % book_list.name, 'list': book_list, 'suggested_books': [s.book for s in suggestions[:5]], } return TemplateResponse(request, 'lists/list.html', data) @method_decorator(login_required, name='dispatch') # pylint: disable=unused-argument def post(self, request, list_id): ''' edit a book_list ''' book_list = get_object_or_404(models.List, id=list_id) return redirect(book_list.local_path)