X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/b5e96c2b04a0f0a0d75aa841f543dd14549d5fe9..ee24ab2feac614a0bdc7114e8e2b4b4d0ad44395:/contrib/penfield/models.py

diff --git a/contrib/penfield/models.py b/contrib/penfield/models.py
index 1117336..a5ba85a 100644
--- a/contrib/penfield/models.py
+++ b/contrib/penfield/models.py
@@ -1,11 +1,12 @@
 from django.db import models
 from philo.models import Entity, MultiNode, Template, register_value_model
-from django.core.paginator import Paginator, InvalidPage, EmptyPage
 from django.contrib.auth.models import User
 from django.conf.urls.defaults import url, patterns
 from django.http import Http404, HttpResponse
 from django.template import RequestContext
 from datetime import datetime
+from utils import paginate
+from validators import validate_pagination_count
 
 
 class Tag(models.Model):
@@ -59,11 +60,9 @@ class BlogNode(MultiNode):
 	blog = models.ForeignKey(Blog, related_name='nodes')
 	
 	index_template = models.ForeignKey(Template, related_name='blog_index_related')
-	index_pages = models.IntegerField(help_text="Please enter a number between 0 and 9999.")
 	archive_template = models.ForeignKey(Template, related_name='blog_archive_related')
-	archive_pages = models.IntegerField(help_text="Please enter a number between 0 and 9999.")
 	tag_template = models.ForeignKey(Template, related_name='blog_tag_related')
-	tag_pages = models.IntegerField(help_text="Please enter a number between 0 and 9999.")
+	entries_per_page = models.IntegerField(blank=True, validators=[validate_pagination_count])
 	entry_template = models.ForeignKey(Template, related_name='blog_entry_related')
 	
 	entry_permalink_style = models.CharField(max_length=1, choices=PERMALINK_STYLE_CHOICES)
@@ -108,23 +107,12 @@ class BlogNode(MultiNode):
 	
 	def index_view(self, request):
 		entries = self.blog.entries.order_by('-date')
-		if self.index_pages != 0:
-			paginator = Paginator(entries, self.index_pages)
-			try:
-				page = int(request.GET.get('page', '1'))
-				entries = paginator.page(page).object_list
-				page_number = paginator.page(page)
-			except ValueError:
-				page = 1
-				entries = paginator.page(page).object_list
-				page_number = paginator.page(page)
-			try:
-				entries = paginator.page(page).object_list
-				page_number = paginator.page(page)
-			except (EmptyPage, InvalidPage):
-				entries = paginator.page(paginator.num_pages).object_list
-				page_number = paginator.page(page)
-		return HttpResponse(self.index_template.django_template.render(RequestContext(request, {'blog': self.blog, 'entries': entries, 'page_number': page_number})), mimetype=self.index_template.mimetype)
+		if self.entries_per_page:
+			page = paginate(request, entries, self.entries_per_page)
+			entries = page.object_list
+		else:
+			page = None
+		return HttpResponse(self.index_template.django_template.render(RequestContext(request, {'blog': self.blog, 'entries': entries, 'page': page})), mimetype=self.index_template.mimetype)
 	
 	def archive_view(self, request, year=None, month=None, day=None):
 		entries = self.blog.entries.all()
@@ -134,10 +122,21 @@ class BlogNode(MultiNode):
 			entries = entries.filter(date__month=month)
 		if day:
 			entries = entries.filter(date__day=day)
-		return HttpResponse(self.archive_template.django_template.render(RequestContext(request, {'blog': self.blog, 'year': year, 'month': month, 'day': day, 'entries': entries})), mimetype=self.archive_template.mimetype)
+		if self.entries_per_page:
+			page = paginate(request, entries, self.entries_per_page)
+			entries = page.object_list
+		else:
+			page = None
+		return HttpResponse(self.archive_template.django_template.render(RequestContext(request, {'blog': self.blog, 'year': year, 'month': month, 'day': day, 'entries': entries, 'page': page})), mimetype=self.archive_template.mimetype)
 	
 	def tag_view(self, request, tag=None):
-		# return HttpResponse(self.tag_template.django_template.render(RequestContext(request, {'blog': self.blog, 'tag': tag, 'entries': None})), mimetype=self.tag_template.mimetype)
+		entries = self.blog.entries.filter(tags__slug = tag)
+		if self.entries_per_page:
+			page = paginate(request, entries, self.entries_per_page)
+			entries = page.object_list
+		else:
+			page = None
+		return HttpResponse(self.tag_template.django_template.render(RequestContext(request, {'blog': self.blog, 'tag': tag, 'entries': entries, 'page': page})), mimetype=self.tag_template.mimetype)
 		raise Http404
 	
 	def entry_view(self, request, slug, year=None, month=None, day=None):