Probably fixed a potential bug with extraneous ellipses.

[django-pagination.git] / pagination / templatetags / pagination_tags.py

diff --git a/pagination/templatetags/pagination_tags.py b/pagination/templatetags/pagination_tags.py
index 0bd9805..bba21b8 100644 (file)
--- a/pagination/templatetags/pagination_tags.py
+++ b/pagination/templatetags/pagination_tags.py
@@ -2,34 +2,115 @@ try:
     set
 except NameError:
     from sets import Set as set
+from django import template
+from pagination.registration import get_registry, default_pagination
+registry = get_registry()
+from django.db.models.query import QuerySet
+from django.core.paginator import Paginator, QuerySetPaginator, InvalidPage
+#from django.template.loader import render_to_string
 
-def paginate(context, window=4):
-    paginator = context['paginator']
-    page_obj = context['page_obj']
-    page_range = paginator.page_range
-    first = set(page_range[:window])
-    last_start = len(page_range)-window
-    if last_start < 0:
-        last_start = 0
-    last = set(page_range[last_start:])
-    current_start = page_obj.number-1-window
-    if current_start < 0:
-        current_start = 0
-    current = set(page_range[current_start:page_obj.number-1+window])
-    pages = []
-    if len(first.intersection(current)) == 0:
-        pages.extend(list(first))
-        pages.append(None)
-        pages.extend(list(current))
-    else:
-        pages.extend(first.union(current))
-    if len(current.intersection(last)) == 0:
-        pages.append(None)
-        pages.extend(list(last))
+register = template.Library()
+
+def do_autopaginate(parser, token):
+    split = token.split_contents()
+    if len(split) == 1:
+        return AutoPaginateNode()
+    elif len(split) == 2:
+        return AutoPaginateNode(queryset_var=split[1])
     else:
-        pages.extend(list(last.difference(current)))
-    return { 
-        'pages': pages,
-        'page_obj': page_obj,
-        'paginator': paginator,
-    }
\ No newline at end of file
+        raise template.TemplateSyntaxError('%r tag takes only one optional argument.' % split[0])
+
+class AutoPaginateNode(template.Node):
+    def __init__(self, queryset_var=None):
+        if queryset_var:
+            self.queryset_var = template.Variable(queryset_var)
+        else:
+            self.queryset_var = None
+
+    def render(self, context):
+        if self.queryset_var is not None:
+            try:
+                key = self.queryset_var.var
+                value = self.queryset_var.resolve(context)
+                if issubclass(value.__class__, QuerySet):
+                    model = value.model
+                    paginator_class = QuerySetPaginator
+                else:
+                    value = list(value)
+                    try:
+                        model = value[0].__class__
+                    except IndexError:
+                        return u''
+                    paginator_class = Paginator
+                pagination = registry.get_for_model(model)
+                if pagination is None:
+                    pagination = default_pagination
+                paginator = paginator_class(value, pagination)
+                try:
+                    page_obj = paginator.page(context['request'].page)
+                except:
+                    return u''
+                context[key] = page_obj.object_list
+                context['paginator'] = paginator
+                context['page_obj'] = page_obj
+                return u''
+            except template.VariableDoesNotExist:
+                pass
+        for d in context:
+            for key, value in d.iteritems():
+                if issubclass(value.__class__, QuerySet):
+                    model = value.model
+                    pagination = registry.get_for_model(model)
+                    if pagination is not None:
+                        paginator = QuerySetPaginator(value, pagination)
+                        try:
+                            page_obj = paginator.page(context['request'].page)
+                        except:
+                            return u''
+                        context[key] = page_obj.object_list
+                        context['paginator'] = paginator
+                        context['page_obj'] = page_obj
+                        return u''
+        return u''
+
+def paginate(context, window=4):
+    try:
+        paginator = context['paginator']
+        page_obj = context['page_obj']
+        page_range = paginator.page_range
+        first = set(page_range[:window])
+        last = set(page_range[-window:])
+        current_start = page_obj.number-1-window
+        if current_start < 0:
+            current_start = 0
+        current_end = page_obj.number-1+window
+        if current_end < 0:
+            current_end = 0
+        current = set(page_range[current_start:current_end])
+        pages = []
+        if len(first.intersection(current)) == 0:
+            first_list = sorted(list(first))
+            second_list = sorted(list(current))
+            pages.extend(first_list)
+            if first_list[-1] + 1 != second_list[0]:
+                pages.append(None)
+            pages.extend(second_list)
+        else:
+            pages.extend(sorted(list(first.union(current))))
+        if len(current.intersection(last)) == 0:
+            second_list = sorted(list(last))
+            if pages[-1] + 1 != second_list[0]:
+                pages.append(None)
+            pages.extend(second_list)
+        else:
+            pages.extend(sorted(list(last.difference(current))))
+        return { 
+            'pages': pages,
+            'page_obj': page_obj,
+            'paginator': paginator,
+            'is_paginated': paginator.count > paginator.per_page,
+        }
+    except KeyError:
+        return u''
+register.inclusion_tag('pagination/pagination.html', takes_context=True)(paginate)
+register.tag('autopaginate', do_autopaginate)