Adds shared books as a metric for recommending follows

bookwyrm/views/directory.py

@@ -1,13 +1,12 @@
 """ who all's here? """
 from django.contrib.auth.decorators import login_required
 from django.core.paginator import Paginator
-from django.db.models import Count, Q
 from django.shortcuts import redirect
 from django.template.response import TemplateResponse
 from django.views import View
 from django.utils.decorators import method_decorator
 
-from bookwyrm import models
+from .helpers import get_suggested_users
 
 # pylint: disable=no-self-use
 @method_decorator(login_required, name="dispatch")
@@ -30,21 +29,12 @@ class Directory(View):
         if scope == "local":
             filters["local"] = True
 
-        users = models.User.objects.filter(
-            discoverable=True, is_active=True, **filters
-        ).annotate(
-            mutuals=Count(
-                "following",
-                filter=Q(
-                    ~Q(id=request.user.id), following__in=request.user.following.all()
-                ),
-            )
-        )
+        users = get_suggested_users(request.user, **filters)
         sort = request.GET.get("sort")
         if sort == "recent":
             users = users.order_by("-last_active_date")
         else:
-            users = users.order_by("-mutuals", "-last_active_date")
+            users = users.order_by("-mutuals", "-shared_books", "-last_active_date")
 
         paginated = Paginator(users, 12)
 

bookwyrm/views/feed.py

@@ -11,7 +11,7 @@ from django.views import View
 from bookwyrm import activitystreams, forms, models
 from bookwyrm.activitypub import ActivitypubResponse
 from bookwyrm.settings import PAGE_LENGTH, STREAMS
-from .helpers import get_user_from_username, privacy_filter
+from .helpers import get_user_from_username, privacy_filter, get_suggested_users
 from .helpers import is_api_request, is_bookwyrm_request, object_visible_to_user
 
 
@@ -34,23 +34,9 @@ class Feed(View):
 
         paginated = Paginator(activities, PAGE_LENGTH)
 
-        suggested_users = (
-            models.User.objects.filter(
-                ~Q(id__in=request.user.following.all()),
-                ~Q(id=request.user.id),
-                discoverable=True,
-                is_active=True,
-                bookwyrm_user=True,
-            )
-            .annotate(
-                mutuals=Count(
-                    "following",
-                    filter=Q(following__in=request.user.following.all()),
-                )
-            )
-            .order_by("-mutuals", "-last_active_date")
-            .all()[:5]
-        )
+        suggested_users = get_suggested_users(
+            request.user, ~Q(id=request.user.id), bookwyrm_user=True
+        ).order_by("-mutuals", "-last_active_date").all()[:5]
 
         data = {
             **feed_page_data(request.user),

bookwyrm/views/helpers.py

@@ -2,7 +2,7 @@
 import re
 from requests import HTTPError
 from django.core.exceptions import FieldError
-from django.db.models import Max, Q
+from django.db.models import Count, Max, Q
 
 from bookwyrm import activitypub, models
 from bookwyrm.connectors import ConnectorException, get_data
@@ -190,3 +190,29 @@ def get_discover_books():
             .order_by("-review__published_date__max")[:6]
         )
     )
+
+
+def get_suggested_users(user, *args, **kwargs):
+    """ Users, annotated with things they have in common """
+    return models.User.objects.filter(
+        discoverable=True, is_active=True, *args, **kwargs
+    ).exclude(
+        Q(id__in=user.blocks.all()) | Q(blocks=user)
+    ).annotate(
+        mutuals=Count(
+            "following",
+            filter=Q(
+                ~Q(id=user.id),
+                ~Q(id__in=user.following.all()),
+                following__in=user.following.all()
+            ),
+        ),
+        shared_books=Count(
+            "shelfbook",
+            filter=Q(
+                ~Q(id=user.id),
+                shelfbook__book__parent_work__in=[
+                    s.book.parent_work for s in user.shelfbook_set.all()]
+            )
+        )
+    )