如何检查lxml元素树字符串?

时间:2017-04-26 07:12:45

标签: python xml lxml elementtree

我有一个lxml元素树列表。我想在字典中存储子树在树列表的任何一个子节点中出现的次数。例如

tree1='''<A attribute1="1"><B><C/></B></A>'''
tree2='''<A attribute1="1"><D><C attribute="2"/></D></A>'''
tree3='''<E attribute1="1"><B><C/></B></E>'''
list_trees=[tree1,tree2,tree3]
print list_trees
from collections import defaultdict
from lxml import etree as ET
mydict=defaultdict(int) 
for tree in list_trees:
    root=ET.fromstring(tree)
    for sub_root in root.iter():
        print ET.tostring(sub_root)
        mydict[ET.tostring(sub_root)]+=1
print mydict

我得到以下正确结果:

defaultdict(<type 'int'>, {'<E attribute1="1"><B><C/></B></E>': 1, '<C/>': 2, '<A attribute1="1"><D><C attribute="2"/></D></A>': 1, '<B><C/></B>': 2, '<C attribute="2"/>': 1, '<D><C attribute="2"/></D>': 1, '<A attribute1="1"><B><C/></B></A>': 1})

这仅适用于此特定示例。但是,在一般情况下,xmls可以是相同的但具有不同的属性顺序,或者额外的空格或不重要的新行。但是,这种一般情况会破坏我的系统。我知道有关于如何检查2个相同的xml树的帖子,但是,我想将xmls转换为字符串以便执行上述特定应用程序(轻松地将独特的树保持为字符串,以便于比较和更灵活在将来)并且还能够很好地将它存储在sql中。如何将xml以一致的方式制作成一个字符串,无论其顺序如何,或者额外的空格,多余的行?

编辑以提供不起作用的案例: 这3个xml树是相同的,它们只有不同的属性排序或额外的空格或新行。

tree4='''<A attribute1="1" attribute2="2"><B><C/></B></A>'''
tree5='''<A attribute1="1"     attribute2="2"   >
<B><C/></B></A>'''
tree6='''<A attribute2="2" attribute1="1"><B><C/></B></A>'''

我的输出提供以下内容:

defaultdict(<type 'int'>, {'<B><C/></B>': 3, '<A attribute1="1" attribute2="2"><B><C/></B></A>': 1, '<A attribute1="1" attribute2="2">\n<B><C/></B></A>': 1, '<C/>': 3, '<A attribute2="2" attribute1="1"><B><C/></B></A>': 1})

但是,输出应为:

defaultdict(<type 'int'>, {'<B><C/></B>': 3, '<A attribute1="1" attribute2="2"><B><C/></B></A>': 3, '<C/>': 3})

1 个答案:

答案 0 :(得分:1)

如果您坚持要比较XML树的字符串表示,我建议在lxml之上使用BeautifulSoup。特别是,在树的任何部分调用prettify()会创建一个独特的表示,忽略输入中的空格和奇怪的格式。输出字符串有点冗长但它们有效。我继续使用&#34;伪新线&#34;替换换行符。 ('\n' -> '\\n')所以输出更紧凑。

from collections import defaultdict
from bs4 import BeautifulSoup as Soup

tree4='''<A attribute1="1" attribute2="2"><B><C/></B></A>'''
tree5='''<A attribute1="1"     attribute2="2"   >
<B><C/></B></A>'''
tree6='''<A attribute2="2" attribute1="1"><B><C/></B></A>'''
list_trees = [tree4, tree5, tree6]

mydict = defaultdict(int)
for tree in list_trees:
    root = Soup(tree, 'lxml-xml') # Use the LXML XML parser.
    for sub_root in root.find_all():
        print(sub_root)
        mydict[sub_root.prettify().replace('\n', '\\n')] += 1

print('Results')
for key, value in mydict.items():
    print(u'%s: %s' % (key, value))

打印出所需的结果(带有一些额外的换行符和空格):

$ python counter.py

<A attribute1="1" attribute2="2">\n <B>\n  <C/>\n </B>\n</A>: 3
<B>\n <C/>\n</B>: 3
<C/>\n: 3