ПИТОН: Как добавить корневой узел в XML

Question

У меня есть файл XML выглядит как этот

<A> 
    <B> 
    <C> 
     .... 
    </C> 
    </B> 
</A> 

Я хочу, чтобы добавить корень поверх элемента «A». Я узнал, как добавить элементы в root. Но как изменить существующий корень и добавить поверх него с помощью python.

После добавления корня к XML он должен выглядеть следующим образом

<ROOT> 
    <A> 
    <B> 
     <C> 
     .... 
     </C> 
    </B> 
    </A> 
</ROOT> 

Answer 1

import lxml.etree as ET 
tree = ET.parse('data') 
root = tree.getroot() 
newroot = ET.Element("root") 
newroot.insert(0, root) 
print(ET.tostring(newroot, pretty_print=True)) 

дает

<root> 
    <A> 
    <B> 
    <C> 
     .... 
    </C> 
    </B> 
</A> 
</root> 

---

Но на самом деле, если вам не нужно, чтобы добавить что-то более сложное, простая строка может быть достаточно форматирования:

with open('data', 'rb') as f, open('newdata', 'wb') as g: 
    g.write('<ROOT>{}</ROOT>'.format(f.read()))