Kiểm tra xem phần tử có trẻ em hay không

Question

Tôi lấy một tài liệu XML theo cách này:

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 
    # ... 

XML trông như thế này:

<item> 
    <a1>value1</a1> 
    <a2>value2</a2> 
    <a3>value3</a3> 
    <a4> 
    <a11>value222</a11> 
    <a22>value22</a22> 
    </a4> 
</item> 

Làm thế nào để kiểm tra xem a4 (trong trường hợp đặc biệt này, nhưng nó có thể là bất kỳ yếu tố nào khác) có con không?

Answer 1

Bạn có thể thử các list chức năng trên các yếu tố:

>>> 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 

Tôi đã sửa đổi mẫu của bạn vì lệnh gọi hàm findall trên gốc item không hoạt động (vì findall sẽ tìm kiếm hậu duệ trực tiếp chứ không phải phần tử hiện tại). Nếu bạn muốn truy cập văn bản của các con sau này trong chương trình làm việc của mình, bạn có thể làm:

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 

Điều này phù hợp cho một đệ quy.

Answer 2

Lớp yếu tố có phương thức nhận con. Vì vậy, bạn nên sử dụng một cái gì đó như thế này, để kiểm tra xem có trẻ em và kết quả lưu trữ trong một cuốn từ điển bằng phím name = tag:

result = {} 
for child in root.findall("item"): 
    is child.getchildren() == []: 
     result[child.tag] = child.text 

Answer 3

Cá nhân tôi khuyên bạn nên sử dụng trình phân tích cú pháp xml hỗ trợ đầy đủ các biểu thức xpath. subset supported by xml.etree không đủ cho các tác vụ như thế này.

Ví dụ, trong lxml tôi có thể làm:

"cho tôi tất cả trẻ em của những đứa trẻ của nút <item>":

doc.xpath('//item/*/child::*') #equivalent to '//item/*/*', if you're being terse 
Out[18]: [<Element a11 at 0x7f60ec1c1348>, <Element a22 at 0x7f60ec1c1888>] 

hay,

"cho tôi tất cả các <item> Con của chúng ta không có con cái mình ":

doc.xpath('/item/*[count(child::*) = 0]') 
Out[20]: 
[<Element a1 at 0x7f60ec1c1588>, 
<Element a2 at 0x7f60ec1c15c8>, 
<Element a3 at 0x7f60ec1c1608>] 

hay,

"cho tôi tất cả những yếu tố mà không có bất kỳ trẻ em":

doc.xpath('//*[count(child::*) = 0]') 
Out[29]: 
[<Element a1 at 0x7f60ec1c1588>, 
<Element a2 at 0x7f60ec1c15c8>, 
<Element a3 at 0x7f60ec1c1608>, 
<Element a11 at 0x7f60ec1c1348>, 
<Element a22 at 0x7f60ec1c1888>] 

# and if I only care about the text from those nodes... 
doc.xpath('//*[count(child::*) = 0]/text()') 
Out[30]: ['value1', 'value2', 'value3', 'value222', 'value22'] 

Answer 4

Cách đơn giản nhất tôi đã có thể tìm thấy là sử dụng các giá trị bool của nguyên tố này trực tiếp. Điều này có nghĩa bạn có thể sử dụng a4 trong một tuyên bố có điều kiện như-là:

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') 

Chạy đoạn mã này sẽ in

No kids yet 
Has kids now 

Giá trị boolean của một phần tử không nói bất cứ điều gì về text, tail hoặc các thuộc tính. Nó chỉ cho thấy sự hiện diện hay vắng mặt của trẻ em, đó là những gì câu hỏi ban đầu đã được yêu cầu.