如何在使用Python / ElementTree解析XML时保留注释

时间:2010-12-17 21:07:17

标签: python xml elementtree

目前使用的是Python 2.4.3,不允许升级

我想在一个或多个标签中更改给定属性的值,以及更新文件中的XML注释。

我设法创建了一个Python脚本,该脚本将XML文件作为参数,并为指定的每个标记更改属性,如下所示

def update(file, state):
    global Etree
    try:
        from elementtree import ElementTree
        print '*** using ElementTree'
    except ImportError, e:
        print '***'
        print '*** Error: Must install either ElementTree or lxml.'
        print '***'
        raise ImportError, 'must install either ElementTree or lxml'
    #end try

    doc = Etree.parse(file)
    root = doc.getroot()

    for element in root.findall('.//StateManageable'):
        element.attrib['initialState'] = state
    #end for
    doc.write(file)
#end def

这很好,属性“initialState”被更新,除了我的原始XML包含很多XML注释这一事实,但它们早已不复存在,这很糟糕。

怀疑parse只检索XML结构,但我认为XML-comments是结构的一部分。我也意识到我的原始文档的“人类可读”格式早已不复存在,但我已经意识到这是预期的行为,需要使用xmllint --format或XSL进行格式化。

1 个答案:

答案 0 :(得分:17)

我知道现在已经过时了,但我偶然发现了上面关于如何保留评论的答案。 Frederik的published instructions关于如何将注释放入树中仍然适用于当前版本的ElementTree,但至少比我需要的还要多。它将XML包装在一个元素中,这对我来说是不可取的。我也不需要保留处理指令,只需要注释。所以,我把他在网站上提供的课程缩减到了这个:

import xml.etree.ElementTree as ET

class PCParser(ET.XMLTreeBuilder):

   def __init__(self):
       ET.XMLTreeBuilder.__init__(self)
       # assumes ElementTree 1.2.X
       self._parser.CommentHandler = self.handle_comment

   def handle_comment(self, data):
       self._target.start(ET.Comment, {})
       self._target.data(data)
       self._target.end(ET.Comment)

要使用它,请将此对象的实例创建为“解析器”,然后将此参数作为参数传递给ElementTree.parse(),如下所示:

parser = PCParser()
self.tree = ET.parse(self.templateOut, parser=parser)

我对代码或ElementTree的未记录使用没有任何信任,但它对我来说只保留注释而不影响原始文档结构。请注意,对ElementTree的任何未来更改(尽管这些年后似乎不太可能)将会破坏这一点。