takahe/users/views/settings/follows.py

from django.db import models
from django.views.generic import ListView

from users.models import Follow, FollowStates, IdentityStates
from users.views.base import IdentityViewMixin


class FollowsPage(IdentityViewMixin, ListView):
    """
    Shows followers/follows.
    """

    template_name = "settings/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:
            follow_dir = models.Q(target=self.identity)
        else:
            follow_dir = models.Q(source=self.identity)

        return (
            Follow.objects.filter(
                follow_dir,
                state__in=FollowStates.group_active(),
            )
            .select_related(
                "target",
                "target__domain",
                "source",
                "source__domain",
            )
            .exclude(source__state__in=IdentityStates.group_deleted())
            .exclude(target__state__in=IdentityStates.group_deleted())
            .order_by("-created")
        )

    def follows_to_identities(self, follows, attr):
        """
        Turns a list of follows into a list of identities (ith the
        follow creation date preserved on them.
        """
        result = []
        for follow in follows:
            identity = getattr(follow, attr)
            identity.follow_date = follow.state_changed
            result.append(identity)
        return result

    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 = self.follows_to_identities(
                context["page_obj"], "source"
            )
            identity_ids = [identity.id for identity in context["page_obj"]]
            context["outbound_ids"] = Follow.objects.filter(
                source=self.identity,
                target_id__in=identity_ids,
                state__in=FollowStates.group_active(),
            ).values_list("target_id", flat=True)
        else:
            context["page_obj"].object_list = self.follows_to_identities(
                context["page_obj"], "target"
            )
            identity_ids = [identity.id for identity in context["page_obj"]]
            context["inbound_ids"] = Follow.objects.filter(
                target=self.identity,
                source_id__in=identity_ids,
                state__in=FollowStates.group_active(),
            ).values_list("source_id", flat=True)
        context["inbound"] = self.inbound
        context["num_inbound"] = Follow.objects.filter(
            target=self.identity,
            state__in=FollowStates.group_active(),
        ).count()
        context["num_outbound"] = Follow.objects.filter(
            source=self.identity,
            state__in=FollowStates.group_active(),
        ).count()
        context["identity"] = self.identity
        return context