Come faccio a ottenere in Python ElementTree una bella stampa su un file XML?

Question

Sfondo

Sto utilizzando SQLite per accedere a un database e recuperare le informazioni desiderate. Sto usando ElementTree in Python versione 2.6 per creare un file XML con queste informazioni.

Codice

import sqlite3 
import xml.etree.ElementTree as ET 

# NOTE: Omitted code where I acccess the database, 
# pull data, and add elements to the tree 

tree = ET.ElementTree(root) 

# Pretty printing to Python shell for testing purposes 
from xml.dom import minidom 
print minidom.parseString(ET.tostring(root)).toprettyxml(indent = " ") 

####### Here lies my problem ####### 
tree.write("New_Database.xml") 

tentativi

Ho provato con tree.write("New_Database.xml", "utf-8") al posto del l'ultima riga di codice di cui sopra, ma non ha di modifica del layout del XML a tutti - è ancora un pasticcio confuso.

ho anche deciso di smanettare e provato a fare:
tree = minidom.parseString(ET.tostring(root)).toprettyxml(indent = " ")
invece di stampare questo alla shell Python, che dà l'errore di AttributeError : oggetto 'unicode' non ha alcun attributo 'scrittura'.

Domande

Quando scrivo il mio albero in un file XML sull'ultima riga, c'è un modo per stampare abbastanza al file XML, come fa per la shell Python?

Posso usare toprettyxml() qui o c'è un altro modo per farlo?

Answer 1

Qualunque sia la stringa XML, è possibile scriverla nel file desiderato aprendo un file per scrivere e scrivere la stringa sul file.

from xml.dom import minidom 

xmlstr = minidom.parseString(ET.tostring(root)).toprettyxml(indent=" ") 
with open("New_Database.xml", "w") as f: 
    f.write(xmlstr) 

C'è una possibile complicanza, soprattutto in Python 2, che è sia meno rigida e meno sofisticata sui caratteri Unicode in stringhe. Se il metodo toprettyxml restituisce una stringa Unicode (u"something"), è possibile eseguire il cast in una codifica file adatta, come UTF-8. Per esempio. sostituire la linea uno di scrittura con:

f.write(xmlstr.encode('utf-8')) 

Answer 2

Installare bs4

pip install bs4 

Usa questo codice per Pretty Print:

from bs4 import BeautifulSoup 

x = your xml 

print(BeautifulSoup(x, "xml").prettify()) 

Answer 3

Date un'occhiata al modulo vkbeautify.

L'input e l'output possono essere stringhe/file in qualsiasi combinazione. È molto compatto e non ha alcuna dipendenza.

import vkbeautify as vkb 

a) pretty_text = vkb.xml(your_xml_text) #return String 

b) vkb.xml(your_xml_text, 'path/to/dest/file') #save in file