from django import template
register = template.Library()
@register.tag
def duplicate(parser, token):
nodelist = parser.parse(('endduplicate',))
parser.delete_first_token()
try:
tag_name, repeat = token.split_contents()
except ValueError:
raise template.TemplateSyntaxError("%r tag requires exactly one arguments" % token.contents.split()[0])
return DuplicateNode(nodelist, repeat)
class DuplicateNode(template.Node):
def __init__(self, nodelist, repeat):
self.nodelist = nodelist
self.repeat = repeat
def render(self, context):
try:
repeat = int(self.repeat)
except ValueError:
self.repeat = template.Variable(self.repeat)
repeat = self.repeat.resolve(context)
output = ''
i = 0
while i < repeat:
output = output + self.nodelist.render(context)
i = i + 1
return output
Comments