Создание doctype с помощью etree из lxml

Question

Я хочу добавить типы документов в свои XML-документы, которые я создаю с помощью etree LXML.

Однако я не могу понять, как добавить doctype.Жесткое кодирование и объединение строки - это не вариант.

Я ожидал чего-то вроде того, как числа "ПИ" добавляются в etree:

pi = etree.PI(...)
doc.addprevious(pi)

Но у меня это не работает.Как добавить a в xml-документ с помощью lxml?

Answer 1

Для начала вы можете создать свой документ с помощью doctype:

# Adapted from example on http://codespeak.net/lxml/tutorial.html
import lxml.etree as et
import StringIO
s = """<?xml version="1.0" encoding="utf-8"?>
<!DOCTYPE root SYSTEM "test" [ <!ENTITY tasty "cheese"> 
<!ENTITY eacute "&#233;"> ]>
<root>
<a>&tasty; souffl&eacute;</a>
</root>
"""
tree = et.parse(StringIO.StringIO(s))
print et.tostring(tree, xml_declaration=True, encoding="utf-8")

С принтами:

<?xml version='1.0' encoding='utf-8'?>
<!DOCTYPE root SYSTEM "test" [
<!ENTITY tasty "cheese">
<!ENTITY eacute "&#233;">
]>
<root>
<a>cheese soufflé</a>
</root>

Если вы хотите добавить doctype к какому-либо XML, который не был создан с его помощью, вы можете сначала создать документ с нужным doctype (как указано выше), затем скопировать в него свой XML без doctype:

xml = et.XML("<root><test/><a>whatever</a><end_test/></root>")
root = tree.getroot()
root[:] = xml
root.text, root.tail = xml.text, xml.tail
print et.tostring(tree, xml_declaration=True, encoding="utf-8")

С принтами:

<?xml version='1.0' encoding='utf-8'?>
<!DOCTYPE root SYSTEM "test" [
<!ENTITY tasty "cheese">
<!ENTITY eacute "&#233;">
]>
<root><test/><a>whatever</a><end_test/></root>

Это то, что ты ищешь?

Answer 2

Это сработало для меня:

print etree.tostring(tree, pretty_print=True, xml_declaration=True, encoding="UTF-8", doctype="<!DOCTYPE TEST_FILE>")

Answer 3

PI фактически добавляется как предыдущий элемент из "doc".Таким образом, это не дочерний элемент "doc".Вы должны использовать "doc.getroottree()".

Вот один из примеров:

>>> root = etree.Element("root")
>>> a  = etree.SubElement(root, "a")
>>> b = etree.SubElement(root, "b")
>>> root.addprevious(etree.PI('xml-stylesheet', 'type="text/xsl" href="my.xsl"'))
>>> print etree.tostring(root, pretty_print=True, xml_declaration=True, encoding='utf-8')
<?xml version='1.0' encoding='utf-8'?>
<root>
  <a/>
  <b/>
</root>

с помощью getroottree():

>>> print etree.tostring(root.getroottree(), pretty_print=True, xml_declaration=True, encoding='utf-8')
<?xml version='1.0' encoding='utf-8'?>
<?xml-stylesheet type="text/xsl" href="my.xsl"?>
<root>
  <a/>
  <b/>
</root>