boards/views/tag_threads.py

from django.shortcuts import get_object_or_404
from boards.models import Tag, Post
from boards.views.all_threads import AllThreadsView, DEFAULT_PAGE
from boards.views.mixins import DispatcherMixin, RedirectNextMixin

__author__ = 'neko259'


class TagView(AllThreadsView, DispatcherMixin, RedirectNextMixin):

    tag_name = None

    def get_threads(self):
        tag = get_object_or_404(Tag, name=self.tag_name)

        return tag.threads.filter(archived=False).order_by('-bump_time')

    def get_context_data(self, **kwargs):
        context = super(TagView, self).get_context_data(**kwargs)

        tag = get_object_or_404(Tag, name=self.tag_name)
        context['tag'] = tag

        return context

    def get(self, request, tag_name, page=DEFAULT_PAGE):
        self.tag_name = tag_name

        dispatch_result = self.dispatch_method(request)
        if dispatch_result:
            return dispatch_result
        else:
            return super(TagView, self).get(request, page)

    def subscribe(self, request):
        user = self._get_user(request)
        tag = get_object_or_404(Tag, name=self.tag_name)

        if not tag in user.fav_tags.all():
            user.add_tag(tag)

        return self.redirect_to_next(request)

    def unsubscribe(self, request):
        user = self._get_user(request)
        tag = get_object_or_404(Tag, name=self.tag_name)

        if tag in user.fav_tags.all():
            user.remove_tag(tag)

        return self.redirect_to_next(request)