X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/4d60dbcc208a2fdc82b44d475d4ebe764c203072..0763589c29458e7c4e7b7bfb5424b724a112931d:/contrib/penfield/utils.py diff --git a/contrib/penfield/utils.py b/contrib/penfield/utils.py index 1a53cfe..8103d9d 100644 --- a/contrib/penfield/utils.py +++ b/contrib/penfield/utils.py @@ -1,5 +1,6 @@ from django.utils.feedgenerator import Atom1Feed, Rss201rev2Feed from django.conf.urls.defaults import url, patterns +from django.contrib.sites.models import Site from django.core.urlresolvers import reverse from django.http import HttpResponse from philo.utils import paginate @@ -15,7 +16,8 @@ class FeedMultiViewMixin(object): feeds_enabled = True atom_feed = Atom1Feed rss_feed = Rss201rev2Feed - feed_description = '' + feed_title = None + feed_description = None list_var = 'objects' def page_view(self, func, page): @@ -51,10 +53,18 @@ class FeedMultiViewMixin(object): else: feed_type = 'atom' - feed = self.get_feed(feed_type, request, node, kwargs, reverse_name) + current_site = Site.objects.get_current() + + feed_kwargs = { + 'link': 'http://%s/%s/%s/' % (current_site.domain, node.get_absolute_url().strip('/'), reverse(reverse_name, urlconf=self, kwargs=kwargs).strip('/')) + } + feed = self.get_feed(feed_type, extra_context, feed_kwargs) for obj in objects: - feed.add_item(obj.title, '/%s/%s/' % (node.get_absolute_url().strip('/'), self.get_subpath(obj).strip('/')), description=self.get_obj_description(obj)) + kwargs = { + 'link': 'http://%s/%s/%s/' % (current_site.domain, node.get_absolute_url().strip('/'), self.get_subpath(obj).strip('/')) + } + self.add_item(feed, obj, kwargs=kwargs) response = HttpResponse(mimetype=feed.mime_type) feed.write(response, 'utf-8') @@ -62,14 +72,19 @@ class FeedMultiViewMixin(object): return inner - def get_feed(self, feed_type, request, node, kwargs, reverse_name): - title = self.feed_title - link = '/%s/%s/' % (node.get_absolute_url().strip('/'), reverse(reverse_name, urlconf=self, kwargs=kwargs).strip('/')) - description = self.feed_description + def get_feed(self, feed_type, extra_context, kwargs=None): + defaults = { + 'description': '' + } + defaults.update(kwargs or {}) + if feed_type == 'rss': - return self.rss_feed(title, link, description) + return self.rss_feed(**defaults) + + if 'description' in defaults and defaults['description'] and 'subtitle' not in defaults: + defaults['subtitle'] = defaults['description'] - return self.atom_feed(title, link, description, subtitle=description) + return self.atom_feed(**defaults) def feed_patterns(self, object_fetcher, page, base_name): feed_name = '%s_feed' % base_name @@ -79,5 +94,6 @@ class FeedMultiViewMixin(object): ) return urlpatterns - def get_obj_description(self, obj): - raise NotImplementedError \ No newline at end of file + def add_item(self, feed, obj, kwargs=None): + defaults = kwargs or {} + feed.add_item(**defaults) \ No newline at end of file