使用 lxml 添加单次出现的 xml 标签

Question

基于我在此处找到的其他几个示例，我创建了一个脚本，该脚本使用 lxml.etree 和 lxml.ebuilder 从 csv 输入创建 xml 文件。它几乎满足了我的需求——我正在努力解决的一件事是我还需要在数据顶部包含一个单次出现的标签，该标签将包含一个静态值。

这是我的示例数据：

ACTION|INV_ACCT_CLASS|EXT_INV_ID|WAREHOUSE_ID|NAME|CNTRY_CD|PHONE|ADDR_STR1|ADDR_STR2|CITY|ST|ZIP|ADD_KEY_NUM
add|2|AAA_00005|1001213|Company 1|US|9995555555|1313 Mockingbird Lane||New York|NY|10001|44433322
add|2|BBB_00008|1004312|Company 2|US|43255511110|Some other address||Stamford|CT|44112|11122233

到目前为止我的代码：

import lxml.etree
 from lxml.builder import E 
 import csv

with open("filename.csv") as csvfile:

                    
  results = E.paiInv(*(
              E.invrec(
               E.action(row['ACTION']),
                E.investor(
                E.inv_account_class(row['INV_ACCOUNT_CLASS']),
                E.ext_inv_id(row['EXT_INV_ID']),
                E.warehouse_id(row['WAREHOUSE_ID']),
                E.name(row['NAME']),
                E.cntry_cd(row['CNTRY_CD']),
                E.phone(row['PHONE']),
                E.addr_str1(row['ADDRESS_STR1']),
                E.addr_str2(row['ADDRESS_STR2']),
                E.city(row['CITY']),
                E.st(row['ST']),
                E.zip(row['ZIP']),
                E.add_key_num(row['ADD_KEY_NUM'])
                      )
                ) for row in csv.DictReader(csvfile, delimiter = '|'))
                          )


 lxml.etree.ElementTree(results).write("OutputFile.xml")

到目前为止，这是我的输出：

<paiInv>
    <invrec>
        <action>add</action>
        <investor>
            <inv_account_class>2</inv_account_class>
            <ext_inv_id>AAA_00005</ext_inv_id>
            <warehouse_id>1001213</warehouse_id>
            <name>Company 1</name>
            <cntry_cd>US</cntry_cd>
            <phone>9995555555</phone>
            <addr_str1>1313 Mockingbird Lane</addr_str1>
            <addr_str2></addr_str2>
            <city>New York</city>
            <st>NY</st>
            <zip>10001</zip>
            <add_key_num>44433322</add_key_num>
        </investor>
    </invrec>
    <invrec>
        <action>add</action>
        <investor>
            <inv_account_class>2</inv_account_class>
            <ext_inv_id>BBB_00008</ext_inv_id>
            <warehouse_id>1004312</warehouse_id>
            <name>Company 2</name>
            <cntry_cd>US</cntry_cd>
            <phone>43255511110</phone>
            <addr_str1>Some other address</addr_str1>
            <addr_str2></addr_str2>
            <city>Stamford</city>
            <st>NB</st>
            <zip>44112</zip>
            <add_key_num>11122233</add_key_num>
        </investor>
    </invrec>
</paiInv>

我需要的输出包括一个额外的（单次出现的）标记，名为 request_id，出现在数据的顶部，如下所示：

<paiInv>
    <request_id>req44</request_id>
    <invrec>
        <action>add</action>
        <investor>
            <inv_account_class>2</inv_account_class>
            <ext_inv_id>AAA_00005</ext_inv_id>
            <warehouse_id>1001213</warehouse_id>
            <name>Company 1</name>
            <cntry_cd>US</cntry_cd>
            <phone>9995555555</phone>
            <addr_str1>1313 Mockingbird Lane</addr_str1>
            <addr_str2></addr_str2>
            <city>New York</city>
            <st>NY</st>
            <zip>10001</zip>
            <add_key_num>44433322</add_key_num>
        </investor>
    </invrec>
    <invrec>
        <action>add</action>
        <investor>
            <inv_account_class>2</inv_account_class>
            <ext_inv_id>BBB_00008</ext_inv_id>
            <warehouse_id>1004312</warehouse_id>
            <name>Company 2</name>
            <cntry_cd>US</cntry_cd>
            <phone>43255511110</phone>
            <addr_str1>Some other address</addr_str1>
            <addr_str2></addr_str2>
            <city>Stamford</city>
            <st>NB</st>
            <zip>44112</zip>
            <add_key_num>11122233</add_key_num>
        </investor>
    </invrec>
</paiInv>

如有任何建议，我们将不胜感激。到目前为止，我在尝试获取额外标记时除了语法错误之外什么也没得到。

Answer 1

在保存文件之前，请尝试以下操作：

doc = lxml.etree.ElementTree(results)
ins = lxml.etree.fromstring('<request_id>req44</request_id>')
ins.tail = "\n"
dest = doc.xpath('/paiInv')[0]
dest.insert(0,ins)
print(lxml.etree.tostring(doc).decode())

输出应该就是你要找的。

使用 lxml 添加单次出现的 xml 标签

Adding a single occurrence xml tag using lxml

python

xml

csv

lxml

elementtree