Minor simplification for user shelf view

This commit is contained in:
Mouse Reeve 2022-01-27 12:05:26 -08:00
parent 91f085c876
commit 577bc18a47
1 changed files with 3 additions and 7 deletions

View File

@ -30,17 +30,15 @@ class User(View):
shelf_preview = [] shelf_preview = []
# only show other shelves that should be visible # only show shelves that should be visible
shelves = user.shelf_set shelves = user.shelf_set
is_self = request.user.id == user.id is_self = request.user.id == user.id
if not is_self: if not is_self:
shelves = models.Shelf.privacy_filter( shelves = models.Shelf.privacy_filter(
request.user, privacy_levels=["public", "followers"] request.user, privacy_levels=["public", "followers"]
).filter(user=user) ).filter(user=user, books__isnull=False)
for user_shelf in shelves.all(): for user_shelf in shelves[:3]:
if not user_shelf.books.count():
continue
shelf_preview.append( shelf_preview.append(
{ {
"name": user_shelf.name, "name": user_shelf.name,
@ -49,8 +47,6 @@ class User(View):
"size": user_shelf.books.count(), "size": user_shelf.books.count(),
} }
) )
if len(shelf_preview) > 2:
break
# user's posts # user's posts
activities = ( activities = (