create readthroughs

bookwyrm/view_actions.py

@@ -272,51 +272,6 @@ def upload_cover(request, book_id):
     return redirect('/book/%s' % book.id)
 
 
-@login_required
-def edit_readthrough(request):
-    ''' can't use the form because the dates are too finnicky '''
-    try:
-        readthrough = models.ReadThrough.objects.get(id=request.POST.get('id'))
-    except models.ReadThrough.DoesNotExist:
-        return HttpResponseNotFound()
-
-    # don't let people edit other people's data
-    if request.user != readthrough.user:
-        return HttpResponseBadRequest()
-
-    # convert dates into a legible format
-    start_date = request.POST.get('start_date')
-    try:
-        start_date = dateutil.parser.parse(start_date)
-    except ParserError:
-        start_date = None
-    readthrough.start_date = start_date
-    finish_date = request.POST.get('finish_date')
-    try:
-        finish_date = dateutil.parser.parse(finish_date)
-    except ParserError:
-        finish_date = None
-    readthrough.finish_date = finish_date
-    readthrough.save()
-    return redirect(request.headers.get('Referer', '/'))
-
-
-@login_required
-def delete_readthrough(request):
-    ''' remove a readthrough '''
-    try:
-        readthrough = models.ReadThrough.objects.get(id=request.POST.get('id'))
-    except models.ReadThrough.DoesNotExist:
-        return HttpResponseNotFound()
-
-    # don't let people edit other people's data
-    if request.user != readthrough.user:
-        return HttpResponseBadRequest()
-
-    readthrough.delete()
-    return redirect(request.headers.get('Referer', '/'))
-
-
 @login_required
 def shelve(request):
     ''' put a  on a user's shelf '''
@@ -351,6 +306,107 @@ def unshelve(request):
     return redirect(request.headers.get('Referer', '/'))
 
 
+@login_required
+def start_reading(request):
+    ''' begin reading a book '''
+    book = books_manager.get_edition(request.POST['book'])
+    shelf = models.Shelf.objects.filter(
+        identifier='reading',
+        user=request.user
+    ).first()
+
+    # create a readthrough
+    readthrough = update_readthrough(request, book=book)
+    if readthrough.start_date:
+        readthrough.save()
+
+    # shelve the book
+    if request.POST.get('reshelve', True):
+        try:
+            current_shelf = models.Shelf.objects.get(
+                    user=request.user,
+                edition=book
+            )
+            outgoing.handle_unshelve(request.user, book, current_shelf)
+        except models.Shelf.DoesNotExist:
+            # this just means it isn't currently on the user's shelves
+            pass
+    outgoing.handle_shelve(request.user, book, shelf)
+
+    # post about it (if you want)
+    if request.POST.get('post-status'):
+        privacy = request.POST.get('privacy')
+        outgoing.handle_reading_status(request.user, shelf, book, privacy)
+
+    return redirect(request.headers.get('Referer', '/'))
+
+
+@login_required
+def finish_reading(request):
+    ''' a user completed a book, yay '''
+    book = books_manager.get_edition(request.POST['book'])
+    shelf = models.Shelf.objects.filter(
+        identifier='read',
+        user=request.user
+    ).first()
+
+    # update or create a readthrough
+    readthrough = update_readthrough(request, book=book)
+    if readthrough.start_date or readthrough.finish_date:
+        readthrough.save()
+
+    # shelve the book
+    if request.POST.get('reshelve', True):
+        try:
+            current_shelf = models.Shelf.objects.get(
+                user=request.user,
+                edition=book
+            )
+            outgoing.handle_unshelve(request.user, book, current_shelf)
+        except models.Shelf.DoesNotExist:
+            # this just means it isn't currently on the user's shelves
+            pass
+    outgoing.handle_shelve(request.user, book, shelf)
+
+    # post about it (if you want)
+    if request.POST.get('post-status'):
+        privacy = request.POST.get('privacy')
+        outgoing.handle_reading_status(request.user, shelf, book, privacy)
+
+    return redirect(request.headers.get('Referer', '/'))
+
+
+@login_required
+def edit_readthrough(request):
+    ''' can't use the form because the dates are too finnicky '''
+    readthrough = update_readthrough(request, create=False)
+    if not readthrough:
+        return HttpResponseNotFound()
+
+    # don't let people edit other people's data
+    if request.user != readthrough.user:
+        return HttpResponseBadRequest()
+    readthrough.save()
+
+    return redirect(request.headers.get('Referer', '/'))
+
+
+@login_required
+def delete_readthrough(request):
+    ''' remove a readthrough '''
+    try:
+        readthrough = models.ReadThrough.objects.get(id=request.POST.get('id'))
+    except models.ReadThrough.DoesNotExist:
+        return HttpResponseNotFound()
+
+    # don't let people edit other people's data
+    if request.user != readthrough.user:
+        return HttpResponseBadRequest()
+
+    readthrough.delete()
+    return redirect(request.headers.get('Referer', '/'))
+
+
 @login_required
 def rate(request):
     ''' just a star rating for a book '''
@@ -578,3 +634,37 @@ def create_invite(request):
     invite.save()
 
     return redirect('/invite')
+
+
+def update_readthrough(request, book=None, create=True):
+    ''' updates but does not save dates on a readthrough '''
+    try:
+        read_id = request.POST.get('id')
+        if not read_id:
+            raise models.ReadThrough.DoesNotExist
+        readthrough = models.ReadThrough.objects.get(id=read_id)
+    except models.ReadThrough.DoesNotExist:
+        if not create or not book:
+            return None
+        readthrough = models.ReadThrough(
+            user=request.user,
+            book=book,
+        )
+
+    start_date = request.POST.get('start_date')
+    if start_date:
+        try:
+            start_date = dateutil.parser.parse(start_date)
+            readthrough.start_date = start_date
+        except ParserError:
+            pass
+
+    finish_date = request.POST.get('finish_date')
+    if finish_date:
+        try:
+            finish_date = dateutil.parser.parse(finish_date)
+            readthrough.finish_date = finish_date
+        except ParserError:
+            pass
+
+    return readthrough