Python：使用 lxml 库将 XML 内容写入文件

Question

我有一个将 xml 字符串写入文件的 Python 函数。代码如下：

我的 file_contents 是一个字符串，如下：

<A>Some Text</A>
<B>Some other Text</B>

我想把这个字符串附在里面，并将 xml 标题 : 添加到文件中。我似乎无法弄清楚这一点。

from lxml import etree

def create_or_overwrite_file(file_name, file_contents):
    try:        
        print('file contents to be written: %s' % file_contents)

        '''
        #root = etree.fromstring(file_contents) #this line definitely fails because file_contents is NOT a valid XML

        #print('root: %s' % str(root))
        #newroot = etree.Element('RootElement')
        #newroot.insert(0, root)
        #print('newroot: %s' % newroot)
        #new_file_contents = etree.tostring(newroot)
        #print('NewFileContents: %s' % str(new_file_contents))
         '''

        root = etree.fromstring(file_contents).getroot()
        et = etree.ElementTree(root)
        with open(str(file_name), "wb") as f:
            et.write(f, encoding="utf-8", xml_declaration=True, pretty_print=True)
            print('wrote file_contents to %s' % str(file_name))
    except Exception as f:
        print(str(f))

我似乎无法让这段代码工作。任何帮助表示赞赏。

Answer 1

如果您从字符串中解析（而不是从文件中读取）。然后fromstring 已经返回根。你不需要.getroot() 电话：

    root = etree.fromstring(file_contents)

通过这一更改，您的代码对我有效。

现在，尚不清楚您为什么要这样做。如果file_contents 已经是XML，因为它必须这样才能工作，那么为什么不只是open(file_name).write(file_contents)？为什么要解码 XML 然后重新编码？

---

跟进

如果你想要的只是一个包装器，那么这会起作用：

def create_or_overwrite_file(file_name, file_contents):
    f = open(file_name,'w')
    print( '<?xml version="1.0"?>\n<root>', file=f )
    f.write(file_contents)
    print( '</root>', file=f )

不涉及lxml 可能更有效。更有意义的地方取决于您。