bgneal@170: """This file contains the feed class for the forums application."""
bgneal@170: 
bgneal@176: from django.contrib.syndication.views import Feed
bgneal@170: from django.core.exceptions import ObjectDoesNotExist
bgneal@176: from django.shortcuts import get_object_or_404
bgneal@170: 
bgneal@170: from forums.models import Forum
bgneal@170: from forums.models import Post
bgneal@176: from core.functions import copyright_str
bgneal@170: 
bgneal@170: 
bgneal@170: class ForumsFeed(Feed):
bgneal@176:     """The Feed class for a specific forum"""
bgneal@170: 
bgneal@385:     ttl = '60'
bgneal@176:     author_name = 'Brian Neal'
bgneal@176:     author_email = 'admin@surfguitar101.com'
bgneal@176: 
bgneal@176:     def get_object(self, request, forum_slug):
bgneal@170:         # only return public forums
bgneal@176:         if forum_slug:
bgneal@176:             forum = get_object_or_404(Forum, slug=forum_slug)
bgneal@387:             if forum.id not in Forum.objects.public_forum_ids():
bgneal@170:                 raise ObjectDoesNotExist
bgneal@170:             return forum
bgneal@170: 
bgneal@176:         else:
bgneal@170:             # return None to indicate we want a combined feed
bgneal@170:             return None
bgneal@170: 
bgneal@170:     def title(self, obj):
bgneal@170:         if obj is None:
bgneal@170:             forum_name = 'Combined'
bgneal@170:         else:
bgneal@170:             forum_name = obj.name
bgneal@170: 
bgneal@170:         return 'SurfGuitar101.com %s Forum Feed' % forum_name
bgneal@170: 
bgneal@170:     def link(self, obj):
bgneal@170:         if obj is None:
bgneal@170:             bits = ''
bgneal@170:         else:
bgneal@170:             bits = obj.slug + '/'
bgneal@170: 
bgneal@170:         return '/feeds/forums/' + bits
bgneal@170: 
bgneal@170:     def description(self, obj):
bgneal@170:         if obj is None:
bgneal@170:             return "User posts to SurfGuitar101.com forums."
bgneal@170:         return obj.description
bgneal@170: 
bgneal@176:     def feed_copyright(self):
bgneal@176:         return copyright_str()
bgneal@170: 
bgneal@170:     def items(self, obj):
bgneal@170:         if obj is None:
bgneal@170:             # return a combined feed of public forum threads
bgneal@387:             #
bgneal@387:             # This didn't work real well on InnoDb:
bgneal@387:             # items = Post.objects.filter(
bgneal@387:             #         topic__forum__in=Forum.objects.public_forums())
bgneal@387:             #
bgneal@388:             # For some reason, MySQL wasn't using an index when "in" was used.
bgneal@388:             # So let's do this the hard way as a work-around:
bgneal@387: 
bgneal@387:             public_forum_ids = Forum.objects.public_forum_ids()
bgneal@388:             posts = []
bgneal@388:             for forum_id in public_forum_ids:
bgneal@388:                 posts.extend(list(Post.objects.filter(
bgneal@388:                     topic__forum__id=forum_id).order_by('-creation_date').select_related(depth=2)[:30]))
bgneal@387: 
bgneal@388:             posts.sort(key=lambda x: x.creation_date, reverse=True)
bgneal@388:             return posts[:30]
bgneal@387: 
bgneal@170:         else:
bgneal@388:             return Post.objects.filter(topic__forum__id=obj.id).order_by(
bgneal@388:                     '-creation_date').select_related(depth=2)[:30]
bgneal@170: 
bgneal@176:     def item_title(self, item):
bgneal@176:         return item.topic.name
bgneal@176: 
bgneal@176:     def item_description(self, item):
bgneal@176:         return item.html
bgneal@176: 
bgneal@176:     def item_author_name(self, item):
bgneal@176:         return item.user.username
bgneal@176: 
bgneal@170:     def item_pubdate(self, item):
bgneal@170:        return item.creation_date
bgneal@170: 
bgneal@170:     def item_categories(self, item):
bgneal@170:        return (item.topic.forum.name, )