帮助自定义 Jinja2 扩展

Question

我一直在努力让这个 Jinja2 自定义扩展正常工作——文档说编写扩展不适合“平民”，这并不是在开玩笑——最终设法得到了这个工作代码：

class WrapperExtension(Extension):

    tags = set(['wrap'])

    def parse(self, parser):
        lineno = parser.stream.next().lineno
        args = [parser.parse_expression()]
        args.append(nodes.Const(args[0].name))
        return nodes.CallBlock(
            self.call_method('_render', args),
            [], [], []).set_lineno(lineno)

    def _render(self, value, name, *args, **kwargs):
        if some_condition():
            return '<wrapper id="%s">%s</wrapper>' % (name, value)
        return value

正如我所说，这就是现在在职的。我不确定的是为什么我需要在 parse() 中返回 nodes.CallBlock，而不是 self.call_method() （其中返回一个 nodes.Call 对象）。如果有人有任何见解——或者可以向我指出有关编写扩展的教程——请告诉我。

Answer 1

原因是 parse() 期望返回一个语句节点，例如 CallBlock 或 Assign。 call_method() 返回一个表达式节点，您必须将其包装在 CallBlock 中才能有一个语句。