Python Lxml(objectify):检查标签是否存在

时间:2011-03-22 01:40:38

标签: python xml lxml objectify

我需要检查xml文件中是否存在某个标记。

例如,我想查看此代码段中是否存在该标记:

 <main>
       <elem1/>
       <elem2>Hi</elem2>
       <elem3/>
       ...
 </main>

目前,我正在使用一个带有错误检查的丑陋黑客,如下所示:

try:
   if root.elem1.tag:
      foo = elem1
except AttributeError:
   foo = "error finding elem1"

如果无法找到节点,我也想自定义字符串(即“无法找到-tagname-”)。

我必须检查一长串变量,并且我不想重复代码100次。

有什么建议吗?

编辑:

以下是实际xml文件的片段:

<main>
 <asset name="Virtual Dvaered Unpresence">
  <virtual/>
  <presence>
   <faction>Dvaered</faction>
   <value>-1000.000000</value>
   <range>0</range>
  </presence>
 </asset>
 <asset name="Virtual Empire Small">
  <virtual/>
  <presence>
   <faction>Empire</faction>
   <value>100.000000</value>
   <range>2</range>
  </presence>
 </asset>
</main>

我想检查标签是否存在,如果是,则检查内容。

修改编辑: 好的,我将结合两个答案,但我只能投一票。遗憾。

编辑3:关于XPath的相关问题:Python lxml (objectify): Xpath troubles

4 个答案:

答案 0 :(得分:27)

hasattr()适用于此:

if hasattr(root, 'elem1'):
    foo = root.elem1

答案 1 :(得分:6)

修改:更新了示例文件的答案。

我假设你想搜索每个资产的某些标签。如果是这样,以下内容对我有用:

import lxml.objectify

# Parse the file.
tree = lxml.objectify.parse('sample.xml')
root = tree.getroot()

# Which elements to find.
to_find = set(['presence/faction', 'presence/value', 'fake'])

# Go through each asset in the document.
for asset in root.findall('asset'):
    # Check for each element. 
    for name in to_find:
        node = asset.find(name)
        if node is not None:
            print 'Found %s, its value is %s' % (name, node)
        else:
            print 'Unable to find %s' % name

输出结果为:

Found presence/value, its value is -1000.0
Found presence/faction, its value is Dvaered
Unable to find fake
Found presence/value, its value is 100.0
Found presence/faction, its value is Empire
Unable to find fake

答案 2 :(得分:6)

假设您想获得elem2的值,您可以使用xpath来查找它。

tree = etree.parse(StringIO(htmlString), etree.HTMLParser()).getroot()
youWantValue = tree.xpath('/main/elem2')[0].text

答案 3 :(得分:1)

如果您的文档往往相对较短,您可以遍历<main>的所有子项,查找与您的变量名称集匹配的标记:

tree = lxml.etree.fromstring(DATA)
NAMES = set(['elem1', 'elem3'])
for node in tree.iterchildren():
    if node.tag in NAMES:
        print 'found', node.tag

或者您可以一次搜索一个变量名称:

for tag in ('elem1', 'elem3'):
    if tree.find(tag) is not None:
        print 'found', tag