Misago/misago/threads/views/thread.py

from django.shortcuts import render
from django.views import View

from misago.threads.viewmodels import ForumThread, PrivateThread, ThreadPosts


class ThreadBase(View):
    thread = None
    posts = ThreadPosts

    template_name = None

    def get(self, request, pk, slug, page=0):
        thread = self.get_thread(request, pk, slug)
        posts = self.get_posts(request, thread, page)

        frontend_context = self.get_frontend_context(request, thread, posts)
        request.frontend_context['store'].update(frontend_context)

        template_context = self.get_template_context(request, thread, posts)
        return render(request, self.template_name, template_context)

    def get_thread(self, request, pk, slug):
        return self.thread(
            request,
            pk,
            slug,
            path_aware=True,
            read_aware=True,
            subscription_aware=True,
            poll_votes_aware=True,
        )

    def get_posts(self, request, thread, page):
        return self.posts(request, thread, page)

    def get_default_frontend_context(self):
        return {}

    def get_frontend_context(self, request, thread, posts):
        context = self.get_default_frontend_context()

        context.update({
            'thread': thread.get_frontend_context(),
            'posts': posts.get_frontend_context(),
        })

        return context

    def get_template_context(self, request, thread, posts):
        context = {
            'url_name': ':'.join(request.resolver_match.namespaces + [
                request.resolver_match.url_name,
            ])
        }

        context.update(thread.get_template_context())
        context.update(posts.get_template_context())

        return context


class ThreadView(ThreadBase):
    thread = ForumThread
    template_name = 'misago/thread/thread.html'


class PrivateThreadView(ThreadBase):
    thread = PrivateThread
    template_name = 'misago/thread/private_thread.html'