'''
usage
{% ++ <var_name> %}
For example
{% ++ a %}
'''
def increment_var(parser, token):
parts = token.split_contents()
if len(parts) < 2:
raise template.TemplateSyntaxError("'increment' tag must be of the form: {% increment <var_name> %}")
return IncrementVarNode(parts[1])
register.tag('++', increment_var)
class IncrementVarNode(template.Node):
def __init__(self, var_name):
self.var_name = var_name
def render(self,context):
try:
value = context[self.var_name]
context[self.var_name] = value + 1
return u""
except:
raise template.TemplateSyntaxError("The variable does not exist.")
Comments