X-Git-Url: https://git.mdrn.pl/django-pagination.git/blobdiff_plain/bad1d3d94971e05e5879b0f588b4b1a84d6ddaaf..69ed0512ee955e1d7497b83d93c218280bb78e89:/pagination/templatetags/pagination_tags.py

diff --git a/pagination/templatetags/pagination_tags.py b/pagination/templatetags/pagination_tags.py
index 7c6c94d..bba21b8 100644
--- a/pagination/templatetags/pagination_tags.py
+++ b/pagination/templatetags/pagination_tags.py
@@ -6,7 +6,7 @@ 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 QuerySetPaginator, InvalidPage
+from django.core.paginator import Paginator, QuerySetPaginator, InvalidPage
 #from django.template.loader import render_to_string
 
 register = template.Library()
@@ -32,11 +32,20 @@ class AutoPaginateNode(template.Node):
             try:
                 key = self.queryset_var.var
                 value = self.queryset_var.resolve(context)
-                model = value.model
+                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 = QuerySetPaginator(value, pagination)
+                paginator = paginator_class(value, pagination)
                 try:
                     page_obj = paginator.page(context['request'].page)
                 except:
@@ -65,36 +74,43 @@ class AutoPaginateNode(template.Node):
         return u''
 
 def paginate(context, window=4):
-    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:
-        pages.extend(sorted(list(first)))
-        pages.append(None)
-        pages.extend(sorted(list(current)))
-    else:
-        pages.extend(sorted(list(first.union(current))))
-    if len(current.intersection(last)) == 0:
-        pages.append(None)
-        pages.extend(sorted(list(last)))
-    else:
-        pages.extend(sorted(list(last.difference(current))))
-    return { 
-        'pages': pages,
-        'page_obj': page_obj,
-        'paginator': paginator,
-        'is_paginated': paginator.count > paginator.per_page,
-    }
-    
+    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)