使用lxml.objectify测试元素的存在

时间:2014-02-14 10:03:28

标签: python xml lxml lxml.objectify

使用lxml.objectify测试某个元素是否存在的标准方式是什么?

示例XML:

<?xml version="1.0" encoding="utf-8"?>
<Test>
  <MyElement1>sdfsdfdsfd</MyElement1>
</Test>

代码

from lxml import etree, objectify
with open('config.xml') as f:
    xml = f.read()
root = objectify.fromstring(xml)

print root.MyElement1
print root.MyElement17  # AttributeError: no such child: MyElement17

然后,在特定路径上写一些东西的最简单的解决方案是什么?

root.MyElement1.Blah = 'New'  # this works because MyElement1 already exists
root.MyElement17.Blah = 'New'  # this doesn't work because MyElement17 doesn't exist
root.MyElement1.Foo.Bar = 'Hello' # this doesn't as well... How to do this shortly ?

2 个答案:

答案 0 :(得分:7)

如果元素不存在,

find方法将返回None

>>> xml = '''<?xml version="1.0" encoding="utf-8"?>
... <Test>
...   <MyElement1>sdfsdfdsfd</MyElement1>
... </Test>'''
>>>
>>> from lxml import objectify
>>> root = objectify.fromstring(xml)
>>> root.find('.//MyElement1')
'sdfsdfdsfd'
>>> root.find('.//MyElement17')
>>> root.find('.//MyElement17') is None
True

更新根据问题编辑:

>>> from lxml import objectify
>>>
>>> def add_string(parent, attr, s):
...     if len(attr) == 1:
...         setattr(parent, attr[0], s)
...     else:
...         child = getattr(parent, attr[0], None)
...         if child is None:
...             child = objectify.SubElement(parent, attr[0])
...         add_string(child, attr[1:], s)
...
>>> root = objectify.fromstring(xml)
>>> add_string(root, ['MyElement1', 'Blah'], 'New')
>>> add_string(root, ['MyElement17', 'Blah'], 'New')
>>> add_string(root, ['MyElement1', 'Foo', 'Bar'], 'Hello')
>>>
>>> root.MyElement1.Blah
'New'
>>> root.MyElement17.Blah
'New'
>>> root.MyElement1.Foo.Bar
'Hello'

答案 1 :(得分:5)

您可以使用getattr

if getattr(root, 'MyElement17', None):
     # do something