使用 lxml 添加单个出现的 xml 标记

如何解决使用 lxml 添加单个出现的 xml 标记

基于我在这里找到的几个其他示例，我创建了一个脚本，该脚本使用 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>

任何建议将不胜感激。到目前为止，我尝试获取额外标签时，除了语法错误之外，没有发现任何其他问题。

解决方法

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

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())

输出应该是您要查找的内容。