X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/85c501277e1f94ab637e1128c1e8aae7b2da2519..2628ac11201c749bbd3aea23597af8238b3a3e5a:/contrib/penfield/utils.py

diff --git a/contrib/penfield/utils.py b/contrib/penfield/utils.py
index 6f2d41d..591ac04 100644
--- a/contrib/penfield/utils.py
+++ b/contrib/penfield/utils.py
@@ -1,7 +1,9 @@
 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
 
 
 class FeedMultiViewMixin(object):
@@ -15,23 +17,24 @@ class FeedMultiViewMixin(object):
 	atom_feed = Atom1Feed
 	rss_feed = Rss201rev2Feed
 	feed_description = ''
+	list_var = 'objects'
 	
-	def page_view(self, func, page, list_var='entries'):
+	def page_view(self, func, page):
 		"""
 		Wraps an object-fetching function and renders the results as a page.
 		"""
 		def inner(request, node=None, extra_context=None, **kwargs):
-			objects, extra_context = func(request, node, extra_context, **kwargs)
+			objects, extra_context = func(request=request, node=node, extra_context=extra_context, **kwargs)
 
 			context = self.get_context()
 			context.update(extra_context or {})
 
-			if 'page' in kwargs or 'page' in request.GET:
+			if 'page' in kwargs or 'page' in request.GET or (hasattr(self, 'per_page') and self.per_page):
 				page_num = kwargs.get('page', request.GET.get('page', 1))
 				paginator, paginated_page, objects = paginate(objects, self.per_page, page_num)
-				context.update({'paginator': paginator, 'paginated_page': paginated_page, list_var: objects})
+				context.update({'paginator': paginator, 'paginated_page': paginated_page, self.list_var: objects})
 			else:
-				context.update({list_var: objects})
+				context.update({self.list_var: objects})
 
 			return page.render_to_response(node, request, extra_context=context)
 
@@ -42,7 +45,7 @@ class FeedMultiViewMixin(object):
 		Wraps an object-fetching function and renders the results as a rss or atom feed.
 		"""
 		def inner(request, node=None, extra_context=None, **kwargs):
-			objects, extra_context = func(request, node, extra_context, **kwargs)
+			objects, extra_context = func(request=request, node=node, extra_context=extra_context, **kwargs)
 	
 			if 'HTTP_ACCEPT' in request.META and 'rss' in request.META['HTTP_ACCEPT'] and 'atom' not in request.META['HTTP_ACCEPT']:
 				feed_type = 'rss'
@@ -50,9 +53,13 @@ class FeedMultiViewMixin(object):
 				feed_type = 'atom'
 			
 			feed = self.get_feed(feed_type, request, node, kwargs, reverse_name)
+			current_site = Site.objects.get_current()
 			
 			for obj in objects:
-				feed.add_item(obj.title, '/%s/%s/' % (node.get_absolute_url().strip('/'), self.get_subpath(obj).strip('/')), description=obj.excerpt)
+				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,7 +69,8 @@ class FeedMultiViewMixin(object):
 	
 	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('/'))
+		current_site = Site.objects.get_current()
+		link = 'http://%s/%s/%s/' % (current_site.domain, node.get_absolute_url().strip('/'), reverse(reverse_name, urlconf=self, kwargs=kwargs).strip('/'))
 		description = self.feed_description
 		if feed_type == 'rss':
 			return self.rss_feed(title, link, description)
@@ -75,4 +83,8 @@ class FeedMultiViewMixin(object):
 			url(r'^%s/$' % self.feed_suffix, self.feed_view(object_fetcher, feed_name), name=feed_name),
 			url(r'^$', self.page_view(object_fetcher, page), name=base_name)
 		)
-		return urlpatterns
\ No newline at end of file
+		return urlpatterns
+	
+	def add_item(self, feed, obj, kwargs=None):
+		defaults = kwargs or {}
+		feed.add_item(**defaults)
\ No newline at end of file