Adds shelf views

bookwyrm/views/__init__.py

@@ -13,3 +13,6 @@ from .books import Book, EditBook, Editions
 from .books import upload_cover, add_description, switch_edition, resolve_book
 from .author import Author, EditAuthor
 from .tag import Tag, AddTag, RemoveTag
+from .shelf import Shelf
+from .shelf import user_shelves_page, create_shelf, delete_shelf
+from .shelf import shelve, unshelve

bookwyrm/views/helpers.py

@@ -4,7 +4,9 @@ from requests import HTTPError
 from django.db.models import Q
 
 from bookwyrm import activitypub, models
+from bookwyrm.broadcast import broadcast
 from bookwyrm.connectors import ConnectorException, get_data
+from bookwyrm.status import create_generated_note
 from bookwyrm.utils import regex
 
 
@@ -147,3 +149,25 @@ def get_edition(book_id):
     return book
 
 
+def handle_reading_status(user, shelf, book, privacy):
+    ''' post about a user reading a book '''
+    # tell the world about this cool thing that happened
+    try:
+        message = {
+            'to-read': 'wants to read',
+            'reading': 'started reading',
+            'read': 'finished reading'
+        }[shelf.identifier]
+    except KeyError:
+        # it's a non-standard shelf, don't worry about it
+        return
+
+    status = create_generated_note(
+        user,
+        message,
+        mention_books=[book],
+        privacy=privacy
+    )
+    status.save()
+
+    broadcast(user, status.to_create_activity(user))

bookwyrm/views/shelf.py

@@ -0,0 +1,172 @@
+''' shelf views'''
+from django.contrib.auth.decorators import login_required
+from django.http import HttpResponseBadRequest, 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 django.views.decorators.http import require_POST
+
+from bookwyrm import forms, models
+from bookwyrm.activitypub import ActivitypubResponse
+from bookwyrm.broadcast import broadcast
+from .helpers import is_api_request, get_edition, get_user_from_username
+from .helpers import handle_reading_status
+
+
+# pylint: disable= no-self-use
+class Shelf(View):
+    ''' shelf page '''
+    def get(self, request, username, shelf_identifier):
+        ''' display a shelf '''
+        try:
+            user = get_user_from_username(username)
+        except models.User.DoesNotExist:
+            return HttpResponseNotFound()
+
+        if shelf_identifier:
+            shelf = user.shelf_set.get(identifier=shelf_identifier)
+        else:
+            shelf = user.shelf_set.first()
+
+        is_self = request.user == user
+
+        shelves = user.shelf_set
+        if not is_self:
+            follower = user.followers.filter(id=request.user.id).exists()
+            # make sure the user has permission to view the shelf
+            if shelf.privacy == 'direct' or \
+                    (shelf.privacy == 'followers' and not follower):
+                return HttpResponseNotFound()
+
+            # only show other shelves that should be visible
+            if follower:
+                shelves = shelves.filter(privacy__in=['public', 'followers'])
+            else:
+                shelves = shelves.filter(privacy='public')
+
+
+        if is_api_request(request):
+            return ActivitypubResponse(shelf.to_activity(**request.GET))
+
+        books = models.ShelfBook.objects.filter(
+            added_by=user, shelf=shelf
+        ).order_by('-updated_date').all()
+
+        data = {
+            'title': '%s\'s %s shelf' % (user.display_name, shelf.name),
+            'user': user,
+            'is_self': is_self,
+            'shelves': shelves.all(),
+            'shelf': shelf,
+            'books': [b.book for b in books],
+        }
+
+        return TemplateResponse(request, 'shelf.html', data)
+
+    @method_decorator(login_required, name='dispatch')
+    def post(self, request, username, shelf_id):
+        ''' user generated shelves '''
+        if not request.user.username == username:
+            return HttpResponseBadRequest()
+
+        shelf = get_object_or_404(models.Shelf, id=shelf_id)
+        if request.user != shelf.user:
+            return HttpResponseBadRequest()
+        if not shelf.editable and request.POST.get('name') != shelf.name:
+            return HttpResponseBadRequest()
+
+        form = forms.ShelfForm(request.POST, instance=shelf)
+        if not form.is_valid():
+            return redirect(shelf.local_path)
+        shelf = form.save()
+        return redirect(shelf.local_path)
+
+
+def user_shelves_page(request, username):
+    ''' default shelf '''
+    return Shelf.as_view()(request, username, None)
+
+
+@login_required
+@require_POST
+def create_shelf(request):
+    ''' user generated shelves '''
+    form = forms.ShelfForm(request.POST)
+    if not form.is_valid():
+        return redirect(request.headers.get('Referer', '/'))
+
+    shelf = form.save()
+    return redirect('/user/%s/shelf/%s' % \
+            (request.user.localname, shelf.identifier))
+
+
+@login_required
+@require_POST
+def delete_shelf(request, shelf_id):
+    ''' user generated shelves '''
+    shelf = get_object_or_404(models.Shelf, id=shelf_id)
+    if request.user != shelf.user or not shelf.editable:
+        return HttpResponseBadRequest()
+
+    shelf.delete()
+    return redirect('/user/%s/shelves' % request.user.localname)
+
+
+@login_required
+@require_POST
+def shelve(request):
+    ''' put a  on a user's shelf '''
+    book = get_edition(request.POST.get('book'))
+
+    desired_shelf = models.Shelf.objects.filter(
+        identifier=request.POST.get('shelf'),
+        user=request.user
+    ).first()
+
+    if request.POST.get('reshelve', True):
+        try:
+            current_shelf = models.Shelf.objects.get(
+                user=request.user,
+                edition=book
+            )
+            handle_unshelve(request.user, book, current_shelf)
+        except models.Shelf.DoesNotExist:
+            # this just means it isn't currently on the user's shelves
+            pass
+    shelfbook = models.ShelfBook(
+        book=book, shelf=desired_shelf, added_by=request.user)
+    shelfbook.save()
+
+    broadcast(request.user, shelfbook.to_add_activity(request.user))
+
+    # post about "want to read" shelves
+    if desired_shelf.identifier == 'to-read':
+        handle_reading_status(
+            request.user,
+            desired_shelf,
+            book,
+            privacy=desired_shelf.privacy
+        )
+
+    return redirect('/')
+
+
+@login_required
+@require_POST
+def unshelve(request):
+    ''' put a  on a user's shelf '''
+    book = models.Edition.objects.get(id=request.POST['book'])
+    current_shelf = models.Shelf.objects.get(id=request.POST['shelf'])
+
+    handle_unshelve(request.user, book, current_shelf)
+    return redirect(request.headers.get('Referer', '/'))
+
+
+def handle_unshelve(user, book, shelf):
+    ''' unshelve a book '''
+    row = models.ShelfBook.objects.get(book=book, shelf=shelf)
+    activity = row.to_remove_activity(user)
+    row.delete()
+
+    broadcast(user, activity)