from django import template
register = template.Library()
@register.tag(name='counter')
def do_counter(parser, token):
try:
tag_name, args = token.contents.split(None, 1)
except ValueError:
raise template.TemplateSyntaxError("'counter' node requires a variable name.")
return CounterNode(args)
class CounterNode(template.Node):
def __init__(self, varname):
self.varname = varname
def render(self, context):
try:
var = template.resolve_variable(self.varname, context)
except:
var = 0
deep = len(context.dicts)-1
context.dicts[deep][self.varname] = var+1
return ''
Comments