1 from django.utils.feedgenerator import Atom1Feed, Rss201rev2Feed
2 from django.conf.urls.defaults import url, patterns
3 from django.contrib.sites.models import Site
4 from django.core.urlresolvers import reverse
5 from django.http import HttpResponse
6 from philo.utils import paginate
9 class FeedMultiViewMixin(object):
11 This mixin provides common methods for adding feeds to multiviews. In order to use this mixin,
12 the multiview must define feed_title (probably as properties that return values
13 on related objects.) feed_description may also be defined; it defaults to an empty string.
18 rss_feed = Rss201rev2Feed
22 def page_view(self, func, page):
24 Wraps an object-fetching function and renders the results as a page.
26 def inner(request, node=None, extra_context=None, **kwargs):
27 objects, extra_context = func(request=request, node=node, extra_context=extra_context, **kwargs)
29 context = self.get_context()
30 context.update(extra_context or {})
32 if 'page' in kwargs or 'page' in request.GET or (hasattr(self, 'per_page') and self.per_page):
33 page_num = kwargs.get('page', request.GET.get('page', 1))
34 paginator, paginated_page, objects = paginate(objects, self.per_page, page_num)
35 context.update({'paginator': paginator, 'paginated_page': paginated_page, self.list_var: objects})
37 context.update({self.list_var: objects})
39 return page.render_to_response(node, request, extra_context=context)
43 def feed_view(self, func, reverse_name):
45 Wraps an object-fetching function and renders the results as a rss or atom feed.
47 def inner(request, node=None, extra_context=None, **kwargs):
48 objects, extra_context = func(request=request, node=node, extra_context=extra_context, **kwargs)
50 if 'HTTP_ACCEPT' in request.META and 'rss' in request.META['HTTP_ACCEPT'] and 'atom' not in request.META['HTTP_ACCEPT']:
55 feed = self.get_feed(feed_type, request, node, kwargs, reverse_name)
56 current_site = Site.objects.get_current()
60 'link': 'http://%s/%s/%s/' % (current_site.domain, node.get_absolute_url().strip('/'), self.get_subpath(obj).strip('/'))
62 self.add_item(feed, obj, kwargs=kwargs)
64 response = HttpResponse(mimetype=feed.mime_type)
65 feed.write(response, 'utf-8')
70 def get_feed(self, feed_type, request, node, kwargs, reverse_name):
71 title = self.feed_title
72 current_site = Site.objects.get_current()
73 link = 'http://%s/%s/%s/' % (current_site.domain, node.get_absolute_url().strip('/'), reverse(reverse_name, urlconf=self, kwargs=kwargs).strip('/'))
74 description = self.feed_description
75 if feed_type == 'rss':
76 return self.rss_feed(title, link, description)
78 return self.atom_feed(title, link, description, subtitle=description)
80 def feed_patterns(self, object_fetcher, page, base_name):
81 feed_name = '%s_feed' % base_name
82 urlpatterns = patterns('',
83 url(r'^%s/$' % self.feed_suffix, self.feed_view(object_fetcher, feed_name), name=feed_name),
84 url(r'^$', self.page_view(object_fetcher, page), name=base_name)
88 def add_item(self, feed, obj, kwargs=None):
89 defaults = kwargs or {}
90 feed.add_item(**defaults)