使用Jinja2.visitorNodeTransformer()实现自定义节点转换的实用技巧

Jinja2 是一个非常强大的Python模板引擎，可以用于生成各种文本文件，包括HTML、XML、配置文件等。它能够帮助我们将动态数据填充到模板中，从而生成最终的文件。

在Jinja2中，我们可以使用 NodeTransformer 类来实现自定义节点转换。节点转换是指对Jinja2模板中的语法树进行修改、添加或删除节点的操作。通过自定义节点转换，我们可以根据自己的需求，对模板进行进一步的加工和处理，从而生成符合我们要求的最终文件。

NodeTransformer 类是Jinja2中一个非常有用的工具。它提供了一系列的回调函数，用于处理不同类型的节点。我们只需要继承 NodeTransformer 类，并实现相关的回调函数，就能够对模板中的节点进行自定义的转换操作了。

下面是一个使用Jinja2 NodeTransformer 类实现自定义节点转换的例子：

from jinja2 import Environment, BaseLoader, nodes
from jinja2.ext import Extension, Markup

class MyExtension(Extension):
    tags = {'bold'}
    
    def __init__(self, environment):
        super(MyExtension, self).__init__(environment)
        self.environment.extend(
            bold_start="<b>",
            bold_end="</b>"
        )
        
    def parse(self, parser):
        token = next(parser.stream)
        
        return nodes.Output(
            [nodes.Call(
                nodes.Name('bold', 'load'),
                args=[
                    nodes.Const(token.value),
                ],
                kwargs=[],
                dyn_args=None,
                dyn_kwargs=None
            )],
            lineno=token.lineno
        )
    
    def bold(self, text):
        return Markup(f"{self.environment.bold_start}{text}{self.environment.bold_end}")

class MyNodeTransformer(nodes.NodeTransformer):
    def visit_Output(self, node):
        if isinstance(node.nodes[0], nodes.Call) and \
           isinstance(node.nodes[0].node, nodes.Name) and \
           node.nodes[0].node.name == 'bold':
            
            text_node = node.nodes[0].args[0]
            bold_node = nodes.Output([nodes.Markup("<b>"), text_node, nodes.Markup("</b>")])
            
            return bold_node
        
        return node

template_string = """
Hello, {% bold "World" %}!
"""

env = Environment(loader=BaseLoader())
env.integrate(MyExtension)

template = env.from_string(template_string)
result = template.render()

print(result)

在上述例子中，我们先定义了一个继承自 Extension 类的 MyExtension 类。在该类中，我们定义了一个自定义标签 'bold'，并重写了 parse 方法，用于解析该自定义标签。在 parse 方法中，我们将自定义标签转换为一个输出节点，节点的内容是一个 Call 节点，用于调用 bold 方法。

然后，我们定义了一个继承自 NodeTransformer 类的 MyNodeTransformer 类。在该类中，我们重写了 visit_Output 方法，对输出节点进行自定义的转换操作。在本例中，我们判断节点是否是我们自定义标签 'bold' 调用，并将该节点转换为一个新的输出节点，该节点中包含了 <b> 标签。

最后，我们创建了一个Jinja2环境对象 env，并将 MyExtension 类集成到该环境中。然后，我们使用这个环境对象加载模板字符串，并渲染出最终的结果。

在本例中，我们定义了一个模板字符串，其中包含了自定义标签 {% bold "World" %}。在模板渲染的过程中，Jinja2将会调用我们定义的 MyNodeTransformer 类的 visit_Output 方法，对包含自定义标签的节点进行转换，最终输出结果为 Hello, <b>World</b>!。