I retrieve an XML documents this way:
import xml.etree.ElementTree as ET
root = ET.parse(urllib2.urlopen(url))
for child in root.findall("item"):
a1 = child[0].text # ok
a2 = child[1].text # ok
a3 = child[2].text # ok
a4 = child[3].text # BOOM
# ...
The XML looks like this:
<item>
<a1>value1</a1>
<a2>value2</a2>
<a3>value3</a3>
<a4>
<a11>value222</a11>
<a22>value22</a22>
</a4>
</item>
How do I check if a4
(in this particular case, but it might've been any other element) has children?
etree only returns real Elements, i.e. tree nodes that have a string tag name. Without a filter, both libraries iterate over all nodes. Note that currently only lxml. etree supports passing the Element factory function as filter to select only Elements.
You could try the list
function on the element:
>>> xml = """<item>
<a1>value1</a1>
<a2>value2</a2>
<a3>value3</a3>
<a4>
<a11>value222</a11>
<a22>value22</a22>
</a4>
</item>"""
>>> root = ET.fromstring(xml)
>>> list(root[0])
[]
>>> list(root[3])
[<Element 'a11' at 0x2321e10>, <Element 'a22' at 0x2321e48>]
>>> len(list(root[3]))
2
>>> print "has children" if len(list(root[3])) else "no child"
has children
>>> print "has children" if len(list(root[2])) else "no child"
no child
>>> # Or simpler, without a call to list within len, it also works:
>>> print "has children" if len(root[3]) else "no child"
has children
I modified your sample because the findall
function call on the item
root did not work (as findall
will search for direct descendants, and not the current element). If you want to access text of the subchildren afterward in your working program, you could do:
for child in root.findall("item"):
# if there are children, get their text content as well.
if len(child):
for subchild in child:
subchild.text
# else just get the current child text.
else:
child.text
This would be a good fit for a recursive though.
The simplest way I have been able to find is to use the bool
value of the element directly. This means you can use a4
in a conditional statement as-is:
a4 = Element('a4')
if a4:
print('Has kids')
else:
print('No kids yet')
a4.append(Element('x'))
if a4:
print('Has kids now')
else:
print('Still no kids')
Running this code will print
No kids yet
Has kids now
The boolean value of an element does not say anything about text
, tail
or attributes. It only indicates the presence or absence of children, which is what the original question was asking.
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With