Added a proper related_name argument to the GenericRelation from Entity to Attribute...
[philo.git] / philo / templatetags / collections.py
1 """
2 The collection template tags are automatically included as builtins if :mod:`philo` is an installed app.
3
4 """
5
6 from django import template
7 from django.conf import settings
8 from django.contrib.contenttypes.models import ContentType
9
10
11 register = template.Library()
12
13
14 class MembersofNode(template.Node):
15         def __init__(self, collection, model, as_var):
16                 self.collection = template.Variable(collection)
17                 self.model = model
18                 self.as_var = as_var
19                 
20         def render(self, context):
21                 try:
22                         collection = self.collection.resolve(context)
23                         context[self.as_var] = collection.members.with_model(self.model)
24                 except:
25                         pass
26                 return ''
27
28
29 @register.tag
30 def membersof(parser, token):
31         """
32         Given a collection and a content type, sets the results of :meth:`collection.members.with_model <.CollectionMemberManager.with_model>` as a variable in the context.
33         
34         Usage::
35         
36                 {% membersof <collection> with <app_label>.<model_name> as <var> %}
37         
38         """
39         params=token.split_contents()
40         tag = params[0]
41         
42         if len(params) < 6:
43                 raise template.TemplateSyntaxError('"%s" template tag requires six parameters' % tag)
44                 
45         if params[2] != 'with':
46                 raise template.TemplateSyntaxError('"%s" template tag requires the third parameter to be "with"' % tag)
47         
48         try:
49                 app_label, model = params[3].strip('"').split('.')
50                 ct = ContentType.objects.get_by_natural_key(app_label, model)
51         except ValueError:
52                 raise template.TemplateSyntaxError('"%s" template tag option "with" requires an argument of the form app_label.model (see django.contrib.contenttypes)' % tag)
53         except ContentType.DoesNotExist:
54                 raise template.TemplateSyntaxError('"%s" template tag option "with" requires an argument of the form app_label.model which refers to an installed content type (see django.contrib.contenttypes)' % tag)
55                 
56         if params[4] != 'as':
57                 raise template.TemplateSyntaxError('"%s" template tag requires the fifth parameter to be "as"' % tag)
58         
59         return MembersofNode(collection=params[1], model=ct.model_class(), as_var=params[5])