diff options
Diffstat (limited to 'activities')
| -rw-r--r-- | activities/views/follows.py | 47 |
1 files changed, 32 insertions, 15 deletions
diff --git a/activities/views/follows.py b/activities/views/follows.py index 6b0881a..f5f5909 100644 --- a/activities/views/follows.py +++ b/activities/views/follows.py @@ -1,4 +1,3 @@ -from django.db.models import Q from django.utils.decorators import method_decorator from django.views.generic import ListView @@ -18,22 +17,40 @@ class Follows(ListView): } paginate_by = 50 + def get(self, request, *args, **kwargs): + self.inbound = self.request.GET.get("inbound") + return super().get(request, *args, **kwargs) + def get_queryset(self): - return Follow.objects.filter( - Q(source=self.request.identity) | Q(target=self.request.identity), - state__in=FollowStates.group_active(), - ).order_by("-created") + if self.inbound: + return Follow.objects.filter( + target=self.request.identity, + state__in=FollowStates.group_active(), + ).order_by("-created") + else: + return Follow.objects.filter( + source=self.request.identity, + state__in=FollowStates.group_active(), + ).order_by("-created") def get_context_data(self): context = super().get_context_data() - identities = [] - for follow in context["page_obj"].object_list: - if follow.source == self.request.identity: - identity = follow.target - follow_type = "outbound" - else: - identity = follow.source - follow_type = "inbound" - identities.append((identity, follow_type)) - context["page_obj"].object_list = identities + # Go work out if any of these people also follow us/are followed + if self.inbound: + context["page_obj"].object_list = [ + follow.source for follow in context["page_obj"] + ] + identity_ids = [identity.id for identity in context["page_obj"]] + context["outbound_ids"] = Follow.objects.filter( + source=self.request.identity, target_id__in=identity_ids + ).values_list("target_id", flat=True) + else: + context["page_obj"].object_list = [ + follow.target for follow in context["page_obj"] + ] + identity_ids = [identity.id for identity in context["page_obj"]] + context["inbound_ids"] = Follow.objects.filter( + target=self.request.identity, source_id__in=identity_ids + ).values_list("source_id", flat=True) + context["inbound"] = self.inbound return context |