Merge branch 'distutils' into release
[philo.git] / philo / utils.py
1 from django.db import models
2 from django.contrib.contenttypes.models import ContentType
3 from django.core.paginator import Paginator, EmptyPage
4 from django.template import Context
5 from django.template.loader_tags import ExtendsNode, ConstantIncludeNode
6
7
8 class ContentTypeLimiter(object):
9         def q_object(self):
10                 return models.Q(pk__in=[])
11         
12         def add_to_query(self, query, *args, **kwargs):
13                 query.add_q(self.q_object(), *args, **kwargs)
14
15
16 class ContentTypeRegistryLimiter(ContentTypeLimiter):
17         def __init__(self):
18                 self.classes = []
19         
20         def register_class(self, cls):
21                 self.classes.append(cls)
22         
23         def unregister_class(self, cls):
24                 self.classes.remove(cls)
25         
26         def q_object(self):
27                 contenttype_pks = []
28                 for cls in self.classes:
29                         try:
30                                 if issubclass(cls, models.Model):
31                                         if not cls._meta.abstract:
32                                                 contenttype = ContentType.objects.get_for_model(cls)
33                                                 contenttype_pks.append(contenttype.pk)
34                         except:
35                                 pass
36                 return models.Q(pk__in=contenttype_pks)
37
38
39 class ContentTypeSubclassLimiter(ContentTypeLimiter):
40         def __init__(self, cls, inclusive=False):
41                 self.cls = cls
42                 self.inclusive = inclusive
43         
44         def q_object(self):
45                 contenttype_pks = []
46                 def handle_subclasses(cls):
47                         for subclass in cls.__subclasses__():
48                                 try:
49                                         if issubclass(subclass, models.Model):
50                                                 if not subclass._meta.abstract:
51                                                         if not self.inclusive and subclass is self.cls:
52                                                                 continue
53                                                         contenttype = ContentType.objects.get_for_model(subclass)
54                                                         contenttype_pks.append(contenttype.pk)
55                                         handle_subclasses(subclass)
56                                 except:
57                                         pass
58                 handle_subclasses(self.cls)
59                 return models.Q(pk__in=contenttype_pks)
60
61
62 def fattr(*args, **kwargs):
63         def wrapper(function):
64                 for key in kwargs:
65                         setattr(function, key, kwargs[key])
66                 return function
67         return wrapper
68
69
70 def paginate(objects, per_page=None, page_number=1):
71         """
72         Given a list of objects, return a (paginator, page, objects) tuple.
73         """
74         try:
75                 per_page = int(per_page)
76         except (TypeError, ValueError):
77                 # Then either it wasn't set or it was set to an invalid value
78                 paginator = page = None
79         else:
80                 # There also shouldn't be pagination if the list is too short. Try count()
81                 # first - good chance it's a queryset, where count is more efficient.
82                 try:
83                         if objects.count() <= per_page:
84                                 paginator = page = None
85                 except AttributeError:
86                         if len(objects) <= per_page:
87                                 paginator = page = None
88         
89         try:
90                 return paginator, page, objects
91         except NameError:
92                 pass
93         
94         paginator = Paginator(objects, per_page)
95         try:
96                 page_number = int(page_number)
97         except:
98                 page_number = 1
99         
100         try:
101                 page = paginator.page(page_number)
102         except EmptyPage:
103                 page = None
104         else:
105                 objects = page.object_list
106         
107         return paginator, page, objects
108
109
110 LOADED_TEMPLATE_ATTR = '_philo_loaded_template'
111 BLANK_CONTEXT = Context()
112
113
114 def get_extended(self):
115         return self.get_parent(BLANK_CONTEXT)
116
117
118 def get_included(self):
119         return self.template
120
121
122 # We ignore the IncludeNode because it will never work in a blank context.
123 setattr(ExtendsNode, LOADED_TEMPLATE_ATTR, property(get_extended))
124 setattr(ConstantIncludeNode, LOADED_TEMPLATE_ATTR, property(get_included))