Code ¶from django import template
from django.utils.translation import gettext_lazy as _
register = template.Library()
class PyIfNode(template.Node):
def __init__(self, nodeslist):
self.nodeslist = nodeslist
def __repr__(self):
return "<PyIf node>"
def render(self, context):
for e, nodes in self.nodeslist:
clist = list(context)
clist.reverse()
d = {}
d['_'] = _
for c in clist:
d.update(c)
v = eval(e, d)
if v:
return nodes.render(context)
return ''
def do_pyif(parser, token):
nodeslist = []
while 1:
v = token.contents.split(None, 1)
if v[0] == 'endif':
break
if v[0] in ('pyif', 'elif'):
if len(v) < 2:
raise template.TemplateSyntaxError, "'pyif' statement requires at least one argument"
if len(v) == 2:
tagname, arg = v
else:
tagname, arg = v[0], 'True'
nodes = parser.parse(('else', 'endif', 'elif'))
nodeslist.append((arg, nodes))
token = parser.next_token()
# parser.delete_first_token()
return PyIfNode(nodeslist)
do_pyif = register.tag("pyif", do_pyif)
How to use it ¶ {% pyif i == 1 %}
<p>i=1</p>
{% elif i == 3 %}
<p>i=3</p>
{% else %}
<p>other</p>
{% endif %}
Warning: For now, django don't support elif, so you can use it. And I'v submit a patch about to fix it ( #3090 ). If the patch is accepted, you can use elif tag. |