diff --git a/fedireads/templates/book.html b/fedireads/templates/book.html index c3b13050..25776181 100644 --- a/fedireads/templates/book.html +++ b/fedireads/templates/book.html @@ -21,6 +21,8 @@ + {% include 'snippets/shelve_button.html' %} + diff --git a/fedireads/templatetags/fr_display.py b/fedireads/templatetags/fr_display.py index b34fc9f6..b3c4ad80 100644 --- a/fedireads/templatetags/fr_display.py +++ b/fedireads/templatetags/fr_display.py @@ -84,7 +84,7 @@ def shelve_button_identifier(context, book): ''' check what shelf a user has a book on, if any ''' try: shelf = models.ShelfBook.objects.get( - shelf__user=context['user'], + shelf__user=context['request'].user, book=book ) except models.ShelfBook.DoesNotExist: @@ -102,7 +102,7 @@ def shelve_button_text(context, book): ''' check what shelf a user has a book on, if any ''' try: shelf = models.ShelfBook.objects.get( - shelf__user=context['user'], + shelf__user=context['request'].user, book=book ) except models.ShelfBook.DoesNotExist: