Вставить узел для элемента в XML с помощью Python / ElementTree - PullRequest
Новая
       17

Вставить узел для элемента в XML с помощью Python / ElementTree

9 голосов
/ 25 января 2011

Мне нужно пройти по дереву 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?

Ответы [ 2 ]

13 голосов
/ 25 января 2011

Вы, вероятно, сделали опечатку, потому что в этом примере элемент ошибки добавляется как дочерний элемент элемента со значением 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>
2 голосов
/ 25 января 2011

ElementTree iter (или getiterator для Python <2.7) рекурсивно вернет все узлы в дереве, затем просто проверит ваше состояние исоздать <a href="http://docs.python.org/library/xml.etree.elementtree.html?highlight=subelement#xml.etree.ElementTree.SubElement" rel="nofollow"> 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'))
...