moviewyrm/bookwyrm/views/list.py

221 lines
7.6 KiB
Python
Raw Normal View History

2021-03-08 16:49:10 +00:00
""" book list views"""
2021-01-31 05:33:41 +00:00
from django.contrib.auth.decorators import login_required
2021-02-01 01:34:06 +00:00
from django.core.paginator import Paginator
from django.db import IntegrityError
2021-02-09 19:40:35 +00:00
from django.db.models import Count, Q
2021-01-31 18:34:25 +00:00
from django.http import HttpResponseNotFound, HttpResponseBadRequest
2021-01-31 05:33:41 +00:00
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
2021-01-31 18:34:25 +00:00
from django.views.decorators.http import require_POST
2021-01-31 05:33:41 +00:00
2021-01-31 16:08:52 +00:00
from bookwyrm import forms, models
2021-01-31 05:33:41 +00:00
from bookwyrm.activitypub import ActivitypubResponse
2021-01-31 19:11:26 +00:00
from bookwyrm.connectors import connector_manager
2021-01-31 16:41:11 +00:00
from .helpers import is_api_request, object_visible_to_user, privacy_filter
2021-02-01 19:34:08 +00:00
from .helpers import get_user_from_username
2021-01-31 05:33:41 +00:00
# pylint: disable=no-self-use
class Lists(View):
2021-03-08 16:49:10 +00:00
""" book list page """
2021-01-31 05:33:41 +00:00
def get(self, request):
2021-03-08 16:49:10 +00:00
""" display a book list """
2021-02-01 01:34:06 +00:00
try:
2021-03-08 16:49:10 +00:00
page = int(request.GET.get("page", 1))
2021-02-01 01:34:06 +00:00
except ValueError:
page = 1
2021-01-31 05:33:41 +00:00
user = request.user if request.user.is_authenticated else None
2021-02-09 19:40:35 +00:00
# hide lists with no approved books
2021-03-08 16:49:10 +00:00
lists = (
models.List.objects.filter(
~Q(user=user),
)
.annotate(item_count=Count("listitem", filter=Q(listitem__approved=True)))
.filter(item_count__gt=0)
.distinct()
.all()
)
2021-02-24 19:35:19 +00:00
lists = privacy_filter(
2021-03-08 16:49:10 +00:00
request.user, lists, privacy_levels=["public", "followers"]
)
2021-02-01 01:34:06 +00:00
paginated = Paginator(lists, 12)
2021-01-31 16:08:52 +00:00
data = {
2021-03-08 16:49:10 +00:00
"lists": paginated.page(page),
"list_form": forms.ListForm(),
"path": "/list",
2021-01-31 16:08:52 +00:00
}
2021-03-08 16:49:10 +00:00
return TemplateResponse(request, "lists/lists.html", data)
2021-01-31 05:33:41 +00:00
2021-03-08 16:49:10 +00:00
@method_decorator(login_required, name="dispatch")
2021-01-31 05:33:41 +00:00
# pylint: disable=unused-argument
def post(self, request):
2021-03-08 16:49:10 +00:00
""" create a book_list """
2021-01-31 16:41:11 +00:00
form = forms.ListForm(request.POST)
if not form.is_valid():
2021-03-08 16:49:10 +00:00
return redirect("lists")
2021-01-31 16:41:11 +00:00
book_list = form.save()
2021-02-02 17:37:46 +00:00
2021-01-31 05:33:41 +00:00
return redirect(book_list.local_path)
2021-03-08 16:49:10 +00:00
2021-02-01 19:34:08 +00:00
class UserLists(View):
2021-03-08 16:49:10 +00:00
""" a user's book list page """
2021-02-01 19:34:08 +00:00
def get(self, request, username):
2021-03-08 16:49:10 +00:00
""" display a book list """
2021-02-01 19:34:08 +00:00
try:
2021-03-08 16:49:10 +00:00
page = int(request.GET.get("page", 1))
2021-02-01 19:34:08 +00:00
except ValueError:
page = 1
user = get_user_from_username(request.user, username)
2021-02-01 19:34:08 +00:00
lists = models.List.objects.filter(user=user).all()
2021-02-24 19:35:19 +00:00
lists = privacy_filter(request.user, lists)
2021-02-01 19:34:08 +00:00
paginated = Paginator(lists, 12)
data = {
2021-03-08 16:49:10 +00:00
"user": user,
"is_self": request.user.id == user.id,
"lists": paginated.page(page),
"list_form": forms.ListForm(),
"path": user.local_path + "/lists",
2021-02-01 19:34:08 +00:00
}
2021-03-08 16:49:10 +00:00
return TemplateResponse(request, "user/lists.html", data)
2021-02-01 19:34:08 +00:00
2021-01-31 05:33:41 +00:00
class List(View):
2021-03-08 16:49:10 +00:00
""" book list page """
2021-01-31 05:33:41 +00:00
def get(self, request, list_id):
2021-03-08 16:49:10 +00:00
""" display a book list """
2021-01-31 05:33:41 +00:00
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):
2021-02-01 20:03:11 +00:00
return ActivitypubResponse(book_list.to_activity(**request.GET))
2021-01-31 17:08:06 +00:00
2021-03-08 16:49:10 +00:00
query = request.GET.get("q")
2021-01-31 21:38:26 +00:00
suggestions = None
if query and request.user.is_authenticated:
2021-01-31 19:11:26 +00:00
# search for books
suggestions = connector_manager.local_search(query, raw=True)
2021-01-31 21:38:26 +00:00
elif request.user.is_authenticated:
2021-01-31 19:11:26 +00:00
# just suggest whatever books are nearby
suggestions = request.user.shelfbook_set.filter(
~Q(book__in=book_list.books.all())
)
suggestions = [s.book for s in suggestions[:5]]
if len(suggestions) < 5:
2021-01-31 19:21:50 +00:00
suggestions += [
2021-03-08 16:49:10 +00:00
s.default_edition
for s in models.Work.objects.filter(
~Q(editions__in=book_list.books.all()),
).order_by("-updated_date")
][: 5 - len(suggestions)]
2021-01-31 17:08:06 +00:00
2021-01-31 05:33:41 +00:00
data = {
2021-03-08 16:49:10 +00:00
"list": book_list,
"items": book_list.listitem_set.filter(approved=True),
"pending_count": book_list.listitem_set.filter(approved=False).count(),
"suggested_books": suggestions,
"list_form": forms.ListForm(instance=book_list),
"query": query or "",
2021-01-31 05:33:41 +00:00
}
2021-03-08 16:49:10 +00:00
return TemplateResponse(request, "lists/list.html", data)
2021-01-31 05:33:41 +00:00
2021-03-08 16:49:10 +00:00
@method_decorator(login_required, name="dispatch")
2021-01-31 05:33:41 +00:00
# pylint: disable=unused-argument
def post(self, request, list_id):
2021-03-08 16:49:10 +00:00
""" edit a list """
2021-01-31 05:33:41 +00:00
book_list = get_object_or_404(models.List, id=list_id)
2021-01-31 18:34:25 +00:00
form = forms.ListForm(request.POST, instance=book_list)
if not form.is_valid():
2021-03-08 16:49:10 +00:00
return redirect("list", book_list.id)
2021-01-31 18:34:25 +00:00
book_list = form.save()
2021-01-31 05:33:41 +00:00
return redirect(book_list.local_path)
2021-01-31 18:34:25 +00:00
2021-01-31 20:07:54 +00:00
class Curate(View):
2021-03-08 16:49:10 +00:00
""" approve or discard list suggestsions """
@method_decorator(login_required, name="dispatch")
2021-01-31 20:07:54 +00:00
def get(self, request, list_id):
2021-03-08 16:49:10 +00:00
""" display a pending list """
2021-01-31 20:07:54 +00:00
book_list = get_object_or_404(models.List, id=list_id)
if not book_list.user == request.user:
# only the creater can curate the list
return HttpResponseNotFound()
data = {
2021-03-08 16:49:10 +00:00
"list": book_list,
"pending": book_list.listitem_set.filter(approved=False),
"list_form": forms.ListForm(instance=book_list),
2021-01-31 20:07:54 +00:00
}
2021-03-08 16:49:10 +00:00
return TemplateResponse(request, "lists/curate.html", data)
2021-01-31 20:07:54 +00:00
2021-03-08 16:49:10 +00:00
@method_decorator(login_required, name="dispatch")
2021-01-31 20:07:54 +00:00
# pylint: disable=unused-argument
def post(self, request, list_id):
2021-03-08 16:49:10 +00:00
""" edit a book_list """
2021-01-31 20:07:54 +00:00
book_list = get_object_or_404(models.List, id=list_id)
2021-03-08 16:49:10 +00:00
suggestion = get_object_or_404(models.ListItem, id=request.POST.get("item"))
approved = request.POST.get("approved") == "true"
2021-01-31 20:07:54 +00:00
if approved:
suggestion.approved = True
suggestion.save()
else:
suggestion.delete()
2021-03-08 16:49:10 +00:00
return redirect("list-curate", book_list.id)
2021-01-31 20:07:54 +00:00
2021-01-31 18:34:25 +00:00
@require_POST
def add_book(request, list_id):
2021-03-08 16:49:10 +00:00
""" put a book on a list """
2021-01-31 18:34:25 +00:00
book_list = get_object_or_404(models.List, id=list_id)
if not object_visible_to_user(request.user, book_list):
return HttpResponseNotFound()
2021-03-08 16:49:10 +00:00
book = get_object_or_404(models.Edition, id=request.POST.get("book"))
2021-01-31 18:34:25 +00:00
# do you have permission to add to the list?
try:
2021-03-08 16:49:10 +00:00
if request.user == book_list.user or book_list.curation == "open":
# go ahead and add it
models.ListItem.objects.create(
book=book,
book_list=book_list,
user=request.user,
)
2021-03-08 16:49:10 +00:00
elif book_list.curation == "curated":
# make a pending entry
models.ListItem.objects.create(
approved=False,
book=book,
book_list=book_list,
user=request.user,
)
else:
# you can't add to this list, what were you THINKING
return HttpResponseBadRequest()
except IntegrityError:
# if the book is already on the list, don't flip out
pass
2021-01-31 18:34:25 +00:00
2021-03-08 16:49:10 +00:00
return redirect("list", list_id)
2021-01-31 20:15:38 +00:00
@require_POST
def remove_book(request, list_id):
2021-03-08 16:49:10 +00:00
""" put a book on a list """
2021-01-31 20:15:38 +00:00
book_list = get_object_or_404(models.List, id=list_id)
2021-03-08 16:49:10 +00:00
item = get_object_or_404(models.ListItem, id=request.POST.get("item"))
2021-01-31 20:15:38 +00:00
if not book_list.user == request.user and not item.user == request.user:
2021-01-31 20:15:38 +00:00
return HttpResponseNotFound()
item.delete()
2021-03-08 16:49:10 +00:00
return redirect("list", list_id)