summaryrefslogtreecommitdiffstats
path: root/activities/views/follows.py
blob: f5f590942f0e85f5e3a108b8e46c33c6741a7927 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
from django.utils.decorators import method_decorator
from django.views.generic import ListView

from users.decorators import identity_required
from users.models import Follow, FollowStates


@method_decorator(identity_required, name="dispatch")
class Follows(ListView):
    """
    Shows followers/follows.
    """

    template_name = "activities/follows.html"
    extra_context = {
        "section": "follows",
    }
    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):
        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()
        # 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