Как вы анализируете и вводите дополнительные узлы в расширение Jinja?

Я пытаюсь адаптировать Jinja2 WithExtension для создания общего расширения для упаковки блока (за которым следуют некоторые более сложные).

Моя цель заключается в поддержке следующих шаблонов:

{% wrap template='wrapper.html.j2' ... %}
    <img src="{{ url('image:thumbnail' ... }}">
{% endwrap %}

И для wrapper.html.j2, чтобы выглядеть как-то вроде:

<div>
    some ifs and stuff
    {{ content }}
    more ifs and stuff
</div>

Я считаю, что мой пример больше всего в этом случае, WithExtension, похоже, разбирает блок, а затем добавляет AST-представление некоторых узлов {% assign .. %} в контекст узлов, которые он разбор.

Итак, я решил, что хочу одно и то же, эти назначения, за которым следует блок include, который я ожидаю получить доступ к этим переменным при анализе АСТ и пройти через блок, который был завернут как переменная 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, а не вывода строки (то есть визуализации), - и не является ли это правильный подход. Я на правильных строках, любые идеи о том, как я должен это делать?

Ответ 1

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

<h1>dsd</h1>
{% wrap template='wrapper.html.j2' %}
    {% for i in range(3) %}
        im wrapped content {{ i }}<br>
    {% endfor %}
{% endwrap %}

wrapper.html.j2

Hello im wrapper
<br>
<hr>
{{ content|safe }}
<hr>

args/kwargs get from here 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 }}<br>
    {% endfor %}
{% endexample %}

Ответ 2

Лучший способ справиться с этим - использовать макросы. Определите это с помощью:

{% macro wrapper() -%}
<div>
    some ifs and stuff
    {{ caller() }}
    more ifs and stuff
</div>
{%- endmacro %}

И используйте позже с тегом call:

{% call wrapper() %}
    <img src="{{ url('image:thumbnail' ... }}">
{% endcall %}

Макро может иметь такие аргументы, как функция python, и может быть импортирован:

{% from macros import wrapper %}

Подробнее см. документацию для тегов macro, call и import.