boards/rss.py

from django.contrib.syndication.views import Feed
from django.core.urlresolvers import reverse
from django.shortcuts import get_object_or_404
from boards.models import Post, Tag, Thread
from boards import settings
from boards.models.thread import STATUS_ARCHIVE

__author__ = 'nekorin'


MAX_ITEMS = settings.get_int('RSS', 'MaxItems')


# TODO Make tests for all of these
class AllThreadsFeed(Feed):

    title = settings.get('Version', 'SiteName') + ' - All threads'
    link = '/'
    description_template = 'boards/rss/post.html'

    def items(self):
        return Thread.objects.exclude(status=STATUS_ARCHIVE).order_by('-id')[:MAX_ITEMS]

    def item_title(self, item):
        return item.get_opening_post().title

    def item_link(self, item):
        return reverse('thread', args={item.get_opening_post_id()})

    def item_pubdate(self, item):
        return item.get_pub_time()


class TagThreadsFeed(Feed):

    link = '/'
    description_template = 'boards/rss/post.html'

    def items(self, obj):
        return obj.get_threads().exclude(status=STATUS_ARCHIVE).order_by('-id')[:MAX_ITEMS]

    def get_object(self, request, tag_name):
        return get_object_or_404(Tag, name=tag_name)

    def item_title(self, item):
        return item.get_opening_post().title

    def item_link(self, item):
        return reverse('thread', args={item.get_opening_post_id()})

    def item_pubdate(self, item):
        return item.get_pub_time()

    def title(self, obj):
        return obj.name


class ThreadPostsFeed(Feed):

    link = '/'
    description_template = 'boards/rss/post.html'

    def items(self, obj):
        return obj.get_thread().get_replies().order_by('-pub_time')[:MAX_ITEMS]

    def get_object(self, request, post_id):
        return get_object_or_404(Post, id=post_id)

    def item_title(self, item):
        return item.title

    def item_link(self, item):
        if not item.is_opening():
            return reverse('thread', args={
                item.get_thread().get_opening_post_id()
            }) + "#" + str(item.id)
        else:
            return reverse('thread', args={item.id})

    def item_pubdate(self, item):
        return item.pub_time

    def title(self, obj):
        return obj.title