Misago/misago/users/viewmodels/threads.py

from django.conf import settings

from misago.acl import add_acl
from misago.core.shortcuts import paginate, pagination_dict
from misago.readtracker import threadstracker
from misago.threads.permissions import exclude_invisible_threads
from misago.threads.threadtypes import trees_map
from misago.threads.subscriptions import make_subscription_aware
from misago.threads.utils import add_categories_to_items, add_likes_to_posts
from misago.threads.viewmodels import ThreadsRootCategory
from misago.users.online.utils import make_users_status_aware

from ..serializers import UserFeedSerializer


class UserThreads(object):
    def __init__(self, request, profile, page=0):
        root_category = ThreadsRootCategory(request)
        threads_categories = [root_category.unwrap()] + root_category.subcategories

        threads_queryset = self.get_threads_queryset(
            request, threads_categories, profile)
        posts_queryset = self.get_posts_queryset(
            request.user, profile, threads_queryset
            ).filter(
                is_hidden=False,
                is_unapproved=False
            ).order_by('-pk')

        list_page = paginate(
            posts_queryset, page, settings.MISAGO_POSTS_PER_PAGE, settings.MISAGO_POSTS_TAIL)
        paginator = pagination_dict(list_page, include_page_range=False)

        posts = list(list_page.object_list)

        posters = []
        threads = []

        for post in posts:
            threads.append(post.thread)

            if post.poster:
                posters.append(post.poster)

        add_categories_to_items(
            root_category.unwrap(), threads_categories, posts + threads)

        add_acl(request.user, threads)
        add_acl(request.user, posts)

        threadstracker.make_threads_read_aware(request.user, threads)
        for post in posts:
            threadstracker.make_posts_read_aware(request.user, post.thread, [post])

        add_likes_to_posts(request.user, posts)

        make_users_status_aware(request.user, posters)

        self._user = request.user

        self.posts = posts
        self.paginator = paginator

    def get_threads_queryset(self, request, threads_categories, profile):
        return exclude_invisible_threads(
            request.user, threads_categories, profile.thread_set)

    def get_posts_queryset(self, user, profile, threads_queryset):
        return profile.post_set.select_related('thread').filter(
            id__in=threads_queryset.values('first_post_id')
        )

    def get_frontend_context(self):
        context = {
            'results': UserFeedSerializer(self.posts, many=True, context={'user': self._user}).data
        }

        context.update(self.paginator)

        return context

    def get_template_context(self):
        return {
            'feed': self.posts,
            'paginator': self.paginator
        }