8

我正在尝试在 jinja2 中编写一个非常简单的树遍历模板，使用一些带有重载特殊方法（getattr、getitem 等）的自定义对象，看起来很简单，树的等效 python walk 工作正常，但是有一些东西关于 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)

模板引发异常，就好像递归实际上没有发生一样。进行了内部调用，但是以某种方式对“类别”的引用仍然是指父级。这里给出了什么？关于这些递归模板应该如何工作，我必须缺少一些非常基本的东西。（或者我正在做的一些非常愚蠢的事情，我只是看不到。

4

1 回答 1

8

正如我从您的代码中看到的那样，您正确理解了递归，除了一件事：它确实替换了 for 语句中的 iterable，但不会更新category最初使用的变量（在您的代码中）。因此，您的嵌套循环遍历子项，但set在 original 中进行标记查找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>
于 2009-10-14T08:56:56.620 回答