From ba97870276f1032817697826956bf9d5c114d997 Mon Sep 17 00:00:00 2001
From: Mouse Reeve <mousereeve@riseup.net>
Date: Fri, 6 Nov 2020 20:51:11 -0800
Subject: [PATCH] get suggested books from shelfbook

---
 bookwyrm/templates/feed.html | 2 +-
 bookwyrm/views.py            | 8 ++++++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/bookwyrm/templates/feed.html b/bookwyrm/templates/feed.html
index ea62a6772..085937d79 100644
--- a/bookwyrm/templates/feed.html
+++ b/bookwyrm/templates/feed.html
@@ -11,7 +11,7 @@
         <div class="tabs is-small">
             <ul>
             {% for shelf in suggested_books %}
-            {% if shelf.books.count %}
+            {% if shelf.books %}
             {% with shelf_counter=forloop.counter %}
                 <li>
                     <p>{{ shelf.name }}</p>
diff --git a/bookwyrm/views.py b/bookwyrm/views.py
index 58cb693a0..1f7cd5276 100644
--- a/bookwyrm/views.py
+++ b/bookwyrm/views.py
@@ -3,7 +3,7 @@ import re
 
 from django.contrib.auth.decorators import login_required, permission_required
 from django.contrib.postgres.search import TrigramSimilarity
-from django.db.models import Avg, Count, Q
+from django.db.models import Avg, Q
 from django.http import HttpResponseBadRequest, HttpResponseNotFound,\
         JsonResponse
 from django.core.exceptions import PermissionDenied
@@ -67,9 +67,13 @@ def home_tab(request, tab):
     for preset in preset_shelves:
         limit = max_books - book_count
         shelf = request.user.shelf_set.get(identifier=preset)
+
+        shelf_books = shelf.shelfbook_set.order_by(
+            '-updated_date'
+        ).all()[:limit]
         shelf_preview = {
             'name': shelf.name,
-            'books': shelf.books.all()[:limit]
+            'books': [s.book for s in shelf_books]
         }
         suggested_books.append(shelf_preview)
         book_count += len(shelf_preview['books'])