from django.template import Library, Node
register = Library()
class SplitListNode(Node):
def __init__(self, list, cols, new_list):
self.list, self.cols, self.new_list = list, cols, new_list
def split_seq(self, list, cols=2):
start = 0
for i in xrange(cols):
stop = start + len(list[i::cols])
yield list[start:stop]
start = stop
def render(self, context):
context[self.new_list] = self.split_seq(context[self.list], int(self.cols))
return ''
def list_to_columns(parser, token):
"""Parse template tag: {% list_to_columns list as new_list 2 %}"""
bits = token.contents.split()
if len(bits) != 5:
raise TemplateSyntaxError, "list_to_columns list as new_list 2"
if bits[2] != 'as':
raise TemplateSyntaxError, "second argument to the list_to_columns tag must be 'as'"
return SplitListNode(bits[1], bits[4], bits[3])
list_to_columns = register.tag(list_to_columns)
Comments
Example usage:
#
This snippet is really useful. Thanks for sharing.
One word of warning. If you pass a queryset to
list_to_columns, a large number of SQL queries will be executed. To reduce the number of queries to one you could force the evaluation of the queryset prior to splitting the list:Not sure if this is the best approach, but it reduces the number of queries.
#
Also, try using
in order to utilise django's variable parsing ability. (Your solution fails if you're trying to split a sub-variable, for eg
#