Вставьте узел для элемента в XML с Python/ElementTree

Question

мне нужно пересечь дерево XML для добавления дочернего элемента, если значение меньше 5. Например, это может быть XML-модифицирован в

<?xml version="1.0" encoding="UTF-8"?> 
<A value="45"> 
    <B value="30"> 
     <C value="10"/> 
     <C value ="20"/> 
    </B> 
    <B value="15"> 
     <C value = "5" /> 
     <C value = "10" /> 
    </B> 
</A> 

этот XML.

<?xml version="1.0" encoding="UTF-8"?> 
<A value="45"> 
    <B value="30"> 
     <C value="10"/>    
     <C value ="20"/> 
    </B> 
    <B value="15"> 
     <C value = "5"><D name="error"/></C> 
     <C value = "10" /> 
    </B> 
</A> 

Как я могу это сделать с ElementTree Python?

Answer 1

Вы, вероятно, сделал опечатку, потому что в этом примере элемент ошибки добавляется как дочерний элемент, значение которого равно 10, что не менее 5. Но я думаю, что это идея:

#!/usr/bin/env python 

from xml.etree.ElementTree import fromstring, ElementTree, Element 

def validate_node(elem): 
    for child in elem.getchildren(): 
     validate_node(child) 
     value = child.attrib.get('value', '') 
     if not value.isdigit() or int(value) < 5: 
      child.append(Element('D', {'name': 'error'})) 

if __name__ == '__main__': 
    import sys 
    xml = sys.stdin.read() # read XML from standard input 
    root = fromstring(xml) # parse into XML element tree 
    validate_node(root) 
    ElementTree(root).write(sys.stdout, encoding='utf-8') 
      # write resulting XML to standard output 

Учитывая этот вход:

<?xml version="1.0" encoding="UTF-8"?> 
<A value="45"> 
    <B value="30"> 
     <C value="1"/> 
     <C value="20"/> 
    </B> 
    <B value="15"> 
     <C value="5" /> 
     <C value="10" /> 
     <C value="foo" /> 
    </B> 
</A> 

Это является выход:

<A value="45"> 
    <B value="30"> 
     <C value="1"><D name="error" /></C> 
     <C value="20" /> 
    </B> 
    <B value="15"> 
     <C value="5" /> 
     <C value="10" /> 
     <C value="foo"><D name="error" /></C> 
    </B> 
</A> 

Answer 2

«S iter (или getiterator для Python 2.7 <)Willl рекурсивно возвращает все узлы в дереве, а затем просто проверить ваше состояние и создать SubElement:

from xml.etree import ElementTree as ET 
tree = ET.parse(input) 
for e in tree.getiterator(): 
    if int(e.get('value')) < 5: 
     ET.SubElement(e,'D',dict(name='error'))