我正在尝试在jinja2中编写一个非常简单的树行走模板,使用一些带有重载特殊方法的自定义对象(getattr,getitem等)这看起来很简单,并且树的等效python行程工作正常,但有一些关于Jinja的递归工作的方式,我不明白。代码如下所示:
from jinja2 import Template
class Category(object):
def __init__(self, name):
self.name = name
self.items = {}
self.children = True
def __iter__(self):
return iter(self.items)
def add(self, key, item):
self.items[key] = item
return item
def __getitem__(self, item):
return self.items[item]
def __getattr__(self, attr):
try:
return self.items[attr]
except KeyError:
raise AttributeError(attr)
def __str__(self):
return "<Category '%s'>" % self.name
template = '''
<saved_data>
{% for key in category recursive %}
{% set item = category[key] %}
{% if item.children %}
<category name="{{key}}">
{{ loop(item) }}
</category>
{% else %}
<item name="{{ key }}" value="{{ item }}" />
{% endif %}
{% endfor %}
</saved_data>
'''
b = Category('root')
c = b.add("numbers", Category('numbers'))
c.add("one", 1)
c.add("two", 2)
c.add("three", 3)
d = b.add("letters", Category('letters'))
d.add('ay','a')
d.add('bee','b')
d.add('cee','c')
e = d.add("bools", Category('bools'))
e.add('tru', True)
e.add('fals', False)
def walk(c, depth=0):
for key in c:
item = c[key]
print (' '*depth) + str(item)
if hasattr(item, 'children'):
walk(item, depth+3)
print "Python walking the tree:"
walk(b)
print ""
print "Jinja2 Walking the tree:"
t = Template(template)
print t.render(category = b)
该模板正在引发异常,就好像递归实际上没有发生一样。内部调用已经完成,但不知何故,对“类别”的引用仍然是指父类。什么给这里?对于这些递归模板应该如何工作,必须有一些非常基本的东西。 (或者我正在做的那些根本就是愚蠢的东西,我根本看不到。
答案 0 :(得分:7)
正如我从您的代码中看到的那样,您理解递归是正确的,除了一件事:它确实替换了for语句中的iterable,但不更新最初在其中使用的变量(category
)。因此,您嵌套循环遍历子项,但set
标记查找原始category
,而不是一个传递给loop()
。
我建议将__iter__()
方法更改为self.items.iteritems()
,将模板更改为:
<saved_data>
{% for key, item in category recursive %}
{% if item.children %}
<category name="{{key}}">
{{ loop(item) }}
</category>
{% else %}
<item name="{{ key }}" value="{{ item }}" />
{% endif %}
{% endfor %}
</saved_data>