X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/82b08f79564159d7acbcaf255ed1ac1fb4882e64..d19e216035b14d8f60b24dda0c0670e6997f16ce:/philo/templatetags/collections.py

diff --git a/philo/templatetags/collections.py b/philo/templatetags/collections.py
new file mode 100644
index 0000000..38b3f91
--- /dev/null
+++ b/philo/templatetags/collections.py
@@ -0,0 +1,51 @@
+from django import template
+from django.conf import settings
+from django.contrib.contenttypes.models import ContentType
+
+
+register = template.Library()
+
+
+class MembersofNode(template.Node):
+	def __init__(self, collection, model, as_var):
+		self.collection = template.Variable(collection)
+		self.model = model
+		self.as_var = as_var
+		
+	def render(self, context):
+		try:
+			collection = self.collection.resolve(context)
+			context[self.as_var] = collection.members.with_model(self.model)
+		except:
+			pass
+		return ''
+
+
+def do_membersof(parser, token):
+	"""
+	{% membersof <collection> with <app_label>.<model_name> as <var> %}
+	"""
+	params=token.split_contents()
+	tag = params[0]
+	
+	if len(params) < 6:
+		raise template.TemplateSyntaxError('"%s" template tag requires six parameters' % tag)
+		
+	if params[2] != 'with':
+		raise template.TemplateSyntaxError('"%s" template tag requires the third parameter to be "with"' % tag)
+	
+	try:
+		app_label, model = params[3].strip('"').split('.')
+		ct = ContentType.objects.get(app_label=app_label, model=model)
+	except ValueError:
+		raise template.TemplateSyntaxError('"%s" template tag option "with" requires an argument of the form app_label.model (see django.contrib.contenttypes)' % tag)
+	except ContentType.DoesNotExist:
+		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)
+		
+	if params[4] != 'as':
+		raise template.TemplateSyntaxError('"%s" template tag requires the fifth parameter to be "as"' % tag)
+	
+	return MembersofNode(collection=params[1], model=ct.model_class(), as_var=params[5])
+
+
+register.tag('membersof', do_membersof)
\ No newline at end of file