Как создатьс Питоном cElementTree - PullRequest
Новая
       21

Как создатьс Питоном cElementTree

10 голосов
/ 15 января 2012

Я пытался использовать ответ в этом вопросе, но не могу заставить его работать: Как создать "виртуальный корень" с ElementTree в Python?

Вот мой код: 

import xml.etree.cElementTree as ElementTree
from StringIO import StringIO
s = '<?xml version=\"1.0\" encoding=\"UTF-8\" ?><!DOCTYPE tmx SYSTEM \"tmx14a.dtd\" ><tmx version=\"1.4a\" />'
tree = ElementTree.parse(StringIO(s)).getroot()
header = ElementTree.SubElement(tree,'header',{'adminlang': 'EN',})
body = ElementTree.SubElement(tree,'body')
ElementTree.ElementTree(tree).write('myfile.tmx','UTF-8')

Когда я открываю результирующий файл 'myfile.tmx', он содержит: 

<?xml version='1.0' encoding='UTF-8'?>
<tmx version="1.4a"><header adminlang="EN" /><body /></tmx>

Чего мне не хватает? или есть лучший инструмент?

Ответы [ 4 ]

13 голосов
/ 15 января 2012

Вы можете установить xml_declaration аргумент для функции write на False, так что вывод не будет иметь объявления xml с кодировкой, а затем просто добавьте нужный заголовок вручную. На самом деле, если вы установите кодировку как 'utf-8' (в нижнем регистре), объявление xml также не будет добавлено. 

import xml.etree.cElementTree as ElementTree

tree = ElementTree.Element('tmx', {'version': '1.4a'})
ElementTree.SubElement(tree, 'header', {'adminlang': 'EN'})
ElementTree.SubElement(tree, 'body')

with open('myfile.tmx', 'wb') as f:
    f.write('<?xml version="1.0" encoding="UTF-8" ?><!DOCTYPE tmx SYSTEM "tmx14a.dtd">'.encode('utf8'))
    ElementTree.ElementTree(tree).write(f, 'utf-8')

Результирующий файл (новые строки добавлены вручную для удобства чтения): 

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE tmx SYSTEM "tmx14a.dtd">
<tmx version="1.4a">
    <header adminlang="EN" />
    <body />
</tmx>
8 голосов
/ 15 января 2012

Вы можете использовать lxml и его функцию tostring: 

from lxml import etree

s = """<?xml version="1.0" encoding="UTF-8"?>
<tmx version="1.4a"/>""" 

tree = etree.fromstring(s)
header = etree.SubElement(tree,'header',{'adminlang': 'EN'})
body = etree.SubElement(tree,'body')

print etree.tostring(tree, encoding="UTF-8",
                     xml_declaration=True,
                     pretty_print=True,
                     doctype='<!DOCTYPE tmx SYSTEM "tmx14a.dtd">')

=> 

<?xml version='1.0' encoding='UTF-8'?>
<!DOCTYPE tmx SYSTEM "tmx14a.dtd">
<tmx version="1.4a">
  <header adminlang="EN"/>
  <body/>
</tmx>
1 голос
/ 25 марта 2019

Я использовал другое решение для добавления DOCTYPE, очень просто, очень глупо. 

import xml.etree.ElementTree as ET

with open(path_file, "w", encoding='UTF-8') as xf:
    doc_type = '<?xml version="1.0" encoding="UTF-8"?><!DOCTYPE dlg:window ' \
               'PUBLIC "-//OpenOffice.org//DTD OfficeDocument 1.0//EN" "dialog.dtd">'
    tostring = ET.tostring(root).decode('utf-8')
    file = f"{doc_type}{tostring}"
    xf.write(file)
0 голосов
/ 11 мая 2017

Я не смог найти решение этой проблемы ни с помощью vanilla ElementTree, а решение, предложенное demalexx, создало недействительный XML, который был отклонен моим приложением (DITA).То, что я предлагаю, это обходной путь с участием других модулей, и он отлично работает для меня. 

import re
# found no way for cleanly specify a <!DOCTYPE ...> stanza in ElementTree so
# so we substitute the current <?xml ... ?> stanza with a full <?xml... + <!DOCTYPE...
new_header = '<?xml version="1.0" encoding="UTF-8" ?>\n' \
                 '<!DOCTYPE topic PUBLIC "-//OASIS//DTD DITA Topic//EN" "topic.dtd">\n'

target_xml = re.sub(u"\<\?xml .+?>", new_header, source_xml)
with open(filename, 'w') as catalog_file:
    catalog_file.write(target_xml.encode('utf8'))
...