使用get_template jinja2生成动态XML模板

时间:2016-07-18 09:13:46

标签: python xml jinja2

我是Python和XML的新手,我使用下面的python代码和xml文件来替换xml文件中的变量值,并使用参数{{param}}中的更新值生成另一个输出xml文件。

netconf_payload.py

#!/usr/bin/env python
import os
from jinja2 import Environment, FileSystemLoader
PATH = os.path.dirname(os.path.abspath(__file__))
TEMPLATE_ENVIRONMENT = Environment(
    autoescape=False,
    loader=FileSystemLoader(os.path.join(PATH, 'templates')),
    trim_blocks=False)
def render_template(template_filename, context):
    return TEMPLATE_ENVIRONMENT.get_template(template_filename).render(context)
def create_index_html():
    fname = "bridge.xml"
    param = [3, 'ieee', 3]
    context = {
        'param': param
    }
    #
    with open(fname, 'w') as f:
        html = render_template('bridge.xml', context)
        f.write(html)
def main():
    create_index_html()
if __name__ == "__main__":
    main()

模板/ bridge.xml

<vr>
        <vrId>0</vrId>
        <bridge>
        {% set counter = 0 -%}
        {% for param in param -%}
                <bridgeId>{{ param }}</bridgeId>
                <bridgeType>{{ param }}</bridgeType>
                <bridgeId>{{ param }}</bridgeId>
        {% set counter = counter + 1 -%}
        {% endfor -%}
        </bridge>
</vr>

现在执行命令: python netconf_payload.py 它将生成输出 bridge.xml 文件,如下所示:

<vr>
        <vrId>0</vrId>
        <bridge>
        <bridgeId>3</bridgeId>
                <bridgeType>3</bridgeType>
                <bridgeId>3</bridgeId>
        <bridgeId>ieee</bridgeId>
                <bridgeType>ieee</bridgeType>
                <bridgeId>ieee</bridgeId>
        <bridgeId>3</bridgeId>
                <bridgeType>3</bridgeType>
                <bridgeId>3</bridgeId>
        </bridge>
</vr>

我想要的预期输出是:

<vr>
        <vrId>0</vrId>
        <bridge>
                <bridgeId>3</bridgeId>
                <bridgeType>ieee</bridgeType>
                <bridgeId>3</bridgeId>
        </bridge>
</vr>

2 个答案:

答案 0 :(得分:1)

所以param是一个总是有3个值的列表?在这种情况下,我认为你不想迭代它,我认为你需要明确地引用列表中的每个项目以获得你想要的输出。 E.g。

<bridge>
     <bridgeId>{{ param[0] }}</bridgeId>
     <bridgeType>{{ param[1] }}</bridgeType>
     <bridgeId>{{ param[2] }}</bridgeId>
</bridge>

答案 1 :(得分:0)

您正在循环param,这是一个包含3个元素的列表:

param = [3, 'ieee', 3]

对于每个值,您可以创建 3个元素

{% for param in param -%}
        <bridgeId>{{ param }}</bridgeId>
        <bridgeType>{{ param }}</bridgeType>
        <bridgeId>{{ param }}</bridgeId>
{% endfor -%}

所以你的输出是完全可以预期的。

完全删除循环,只需将3个元素插入3个单独的标记中:

<vr>
        <vrId>0</vrId>
        <bridge>
                <bridgeId>{{ param[0] }}</bridgeId>
                <bridgeType>{{ param[1] }}</bridgeType>
                <bridgeId>{{ param[2] }}</bridgeId>
        </bridge>
</vr>

然而,你可能想要使用字典,因为这可以让你命名3个元素:

param = {'id1': 3, 'type': 'ieee', 'id2': 3}

<vr>
        <vrId>0</vrId>
        <bridge>
                <bridgeId>{{ param.id1 }}</bridgeId>
                <bridgeType>{{ param.type }}</bridgeType>
                <bridgeId>{{ param.id2 }}</bridgeId>
        </bridge>
</vr>

或者如果您需要制作多个bridgeId和*可选及相应的bridgeType代码,或许您需要一个词典列表

param = [{'id1': 3, 'type': 'ieee'}, {'id2': 3}]

请注意,第二个字典此处没有'type'键。渲染然后变为:

<vr>
    <vrId>0</vrId>
    <bridge>
    {% for entry in param -%}
        <bridgeId>{{ entry.id }}</bridgeId>
        {% if entry.type %}<bridgeType>{{ entry.type }}</bridgeType>{% endif %}
    {% endfor -%}
    </bridge>
</vr>