ElementTree: метод вставки и плохой вывод отступа

Я использую этот код для вставки нового узла в исходный .xml (файл, взятый из документации ElementTree):

import xml.etree.ElementTree as ET

tree = ET.parse('country_data.xml')
root = tree.getroot()


my_xml = """
    <country name="Togo">
        <rank updated="yes">2</rank>
        <year>2000</year>
        <gdppc>3</gdppc>
        <neighbor name="Austria" direction="E"/>
        <neighbor name="Switzerland" direction="W"/>
    </country>
"""


new_contry = ET.fromstring(my_xml)
root.insert(1,new_contry)

tree.write('essai_insert_output.xml')

root = tree.getroot()

Но результат уничтожает правильный отступ между новым узлом и следующим:

<data>
    <country name="Liechtenstein">
        <rank>1</rank>
        <year>2008</year>
        <gdppc>141100</gdppc>
        <neighbor name="Austria" direction="E" />
        <neighbor name="Switzerland" direction="W" />
    </country>
    <country name="Togo">
        <rank updated="yes">2</rank>
        <year>2000</year>
        <gdppc>3</gdppc>
        <neighbor name="Austria" direction="E" />
        <neighbor name="Switzerland" direction="W" />
    </country><country name="Singapore">
        <rank>4</rank>
        <year>2011</year>
        <gdppc>59900</gdppc>
        <neighbor name="Malaysia" direction="N" />
    </country>
    <country name="Panama">
        <rank>68</rank>
        <year>2011</year>
        <gdppc>13600</gdppc>
        <neighbor name="Costa Rica" direction="W" />
        <neighbor name="Colombia" direction="E" />
    </country>
</data>

Новый узел <country name="Togo"> хорошо расположен, но в конце у нас есть это столкновение:

</country><country name="Singapore">

Как вылечить этот эффект?


python indentation elementtree insert pretty-print
person Netchaiev    schedule 28.05.2020    source источник
comment
Это просто вопрос форматирования. Структура правильная.   -  person dabingsou    schedule 01.06.2020
comment
@dabinsou Хорошо, но как можно улучшить форматирование? Мне нужно преобразовать XML-файлы для кого-то, и дело в том, чтобы предоставить хорошо отформатированные выходные данные XML (я пробовал модуль lxml, но у меня была та же проблема).   -  person Netchaiev    schedule 01.06.2020
comment
Вот пример форматирования вывода.   -  person dabingsou    schedule 02.06.2020
comment
Вы можете сделать вывод более красивым, просто добавив немного пробела в tail элемента new_contry. Вот так: new_contry.tail = "\n ".   -  person mzjn    schedule 02.06.2020
comment
Вот как вы можете получить красивый вывод с помощью lxml: stackoverflow.com/q/42577999/407651   -  person mzjn    schedule 02.06.2020

Ответы (1)


arrow_upward
0
arrow_downward

Форматировать вывод.

from xml.dom import minidom

dom = minidom.parse('country_data.xml')
with open('dom_write.xml', 'w', encoding='UTF-8') as fh:
    dom.writexml(fh, indent='', addindent='\t', newl='', encoding='UTF-8')
person dabingsou    schedule 02.06.2020