我正在尝试调整Jinja2 WithExtension以生成包含块的通用扩展(后面是一些更复杂的扩展).
我的目标是在模板中支持以下内容:
{% wrap template='wrapper.html.j2' ... %} {% endwrap %}
并且对于wrapper.html.j2看起来像:
some ifs and stuff {{ content }} more ifs and stuff
我相信我的例子大部分都在那里,WithExtension似乎解析了块,然后将一些{% assign .. %}
节点的AST表示附加到它正在解析的节点的上下文中.
所以我想我想要相同的东西,那些赋值,然后是一个include块,我希望能够在解析AST时访问这些变量,并传递包装为变量的块content
.
到目前为止我有以下内容:
class WrapExtension(Extension): tags = set(['wrap']) def parse(self, parser): node = nodes.Scope(lineno=next(parser.stream).lineno) assignments = [] while parser.stream.current.type != 'block_end': lineno = parser.stream.current.lineno if assignments: parser.stream.expect('comma') target = parser.parse_assign_target() parser.stream.expect('assign') expr = parser.parse_expression() assignments.append(nodes.Assign(target, expr, lineno=lineno)) content = parser.parse_statements(('name:endwrap',), drop_needle=True) assignments.append(nodes.Name('content', content)) assignments.append(nodes.Include(nodes.Template('wrapper.html.j2'), True, False)) node.body = assignments return node
然而,它nodes.Include
只是在我的线上,我只是得到assert frame is None, 'no root frame allowed'
.我相信我需要传递AST nodes.Template
而不是模板名称,但我真的不知道如何解析其他节点以获得AST而不是字符串输出(即渲染) - 也不知道这是否是正确的方法.我是否在正确的路线上,对于我应该如何解决这个问题?
templatetags/wrap.py
class WrapExtension(jinja2.ext.Extension): tags = set(['wrap']) template = None def parse(self, parser): tag = parser.stream.current.value lineno = parser.stream.next().lineno args, kwargs = self.parse_args(parser) body = parser.parse_statements(['name:end{}'.format(tag)], drop_needle=True) return nodes.CallBlock(self.call_method('wrap', args, kwargs), [], [], body).set_lineno(lineno) def parse_args(self, parser): args = [] kwargs = [] require_comma = False while parser.stream.current.type != 'block_end': if require_comma: parser.stream.expect('comma') if parser.stream.current.type == 'name' and parser.stream.look().type == 'assign': key = parser.stream.current.value parser.stream.skip(2) value = parser.parse_expression() kwargs.append(nodes.Keyword(key, value, lineno=value.lineno)) else: if kwargs: parser.fail('Invalid argument syntax for WrapExtension tag', parser.stream.current.lineno) args.append(parser.parse_expression()) require_comma = True return args, kwargs @jinja2.contextfunction def wrap(self, context, caller, template=None, *args, **kwargs): return self.environment.get_template(template or self.template).render(dict(context, content=caller(), **kwargs))
base.html.j2
dsd
{% wrap template='wrapper.html.j2' %} {% for i in range(3) %} im wrapped content {{ i }}
{% endfor %} {% endwrap %}
wrapper.html.j2
Hello im wrapper
{{ content|safe }}
args/kwargs解析来自这里https://github.com/Suor/django-cacheops/blob/master/cacheops/jinja2.py
此外,上述内容可以扩展为支持使用指定为包装器的默认模板的其他标签:
templatetags/example.py
class ExampleExtension(WrapExtension): tags = set(['example']) template = 'example.html.j2'
base.html.j2
{% example otherstuff=True, somethingelse=False %} {% for i in range(3) %} im wrapped content {{ i }}
{% endfor %} {% endexample %}