Сохранение структуры XML после добавления элементов с использованием lxml в python - PullRequest
Новая
       63

Сохранение структуры XML после добавления элементов с использованием lxml в python

0 голосов
/ 23 сентября 2019

Во-первых, я новичок в переполнении стека, поэтому, пожалуйста, будьте добры и предложите конструктивную критику относительно того, как улучшить этот вопрос, если есть место для этого.

Проблема: Iхотите сохранить структуру XML-файла, созданного с помощью кода ниже.Я хочу, чтобы это выглядело так: 

<?xml version="1.0" encoding="UTF-8" ?>
<save>
    <header version="2" />
    <version major="3" minor="6" revision="2" build="0" />
    <region id="ModuleSettings">
        <node id="root">
            <children>
                <node id="ModOrder">
                    <children>
                        <node id="Module">
                            <attribute id="UUID" value="627f624f-e2b8-4b37-977e-03044e500fec" type="22" />
                        </node>
                        <node id="Module">
                            <attribute id="UUID" value="7e737d2f-31d2-4751-963f-be6ccc59cd0c" type="22" />
                        </node>
                    </children>
                </node>
                <node id="Mods">
                    <children>
                        <node id="ModuleShortDesc">
                            <attribute id="Folder" value="somestuff1" type="30" />
                            <attribute id="MD5" value="somestuff1" type="23" />
                            <attribute id="Name" value="somestuff1" type="22" />
                            <attribute id="UUID" value="627f624f-e2b8-4b37-977e-03044e500fec" type="22" />
                            <attribute id="Version" value="1" type="4" />
                        </node>
                        <node id="ModuleShortDesc">
                            <attribute id="Folder" value="somestuff2" type="30" />
                            <attribute id="MD5" value="" type="23" />
                            <attribute id="Name" value="somestuff2" type="22" />
                            <attribute id="UUID" value="7e737d2f-31d2-4751-963f-be6ccc59cd0c" type="22" />
                            <attribute id="Version" value="2" type="4" />
                        </node>
                    </children>
                </node>
            </children>
        </node>
    </region>
</save>

, но вместо этого получите это: 

<?xml version="1.0" encoding="UTF-8" ?>
<save>
    <header version="2" />
    <version major="3" minor="6" revision="2" build="0" />
    <region id="ModuleSettings">
        <node id="root">
            <children>
                <node id="ModOrder">
                    <children>
                    <node id="Module"><attribute id="UUID" value="627f624f-e2b8-4b37-977e-03044e500fec" type="22" /></node><node id="Module"><attribute id="UUID" value="7e737d2f-31d2-4751-963f-be6ccc59cd0c" type="22" /></node></children>
                </node>
                <node id="Mods">
                    <children>
                    <node id="ModuleShortDesc"><attribute id="Folder" value="somestuff1" type="30" /><attribute id="MD5" value="somestuff1" type="23" /><attribute id="Name" value="somestuff1" type="22" /><attribute id="UUID" value="627f624f-e2b8-4b37-977e-03044e500fec" type="22" /><attribute id="Version" value="1" type="4" /></node><node id="ModuleShortDesc"><attribute id="Folder" value="somestuff2" type="30" /><attribute id="MD5" value="" type="23" /><attribute id="Name" value="somestuff2" type="22" /><attribute id="UUID" value="7e737d2f-31d2-4751-963f-be6ccc59cd0c" type="22" /><attribute id="Version" value="2" type="4" /></node></children></node>
            </children>
        </node>
    </region>
</save>

Сосредоточив внимание только на узле ModOrder, вот мой текущий код: 

# Create a Module element as object:
def new_module(uuid, ModOrder):

    ''' Example Module:
        <node id="Module">
            <attribute id="UUID" value="627f624f-e2b8-4b37-977e-03044e500fec" type="22" />
        </node>
    '''

    uuid = str(uuid)

    module = et.SubElement(ModOrder, "node")
    module.set("id", "Module")

    attribute_uuid = et.SubElement(module, "attribute")
    attribute_uuid.set("id", "UUID")
    attribute_uuid.set("value", uuid)
    attribute_uuid.set("type", "22")

    return module

def generator2():

    # mods_dictionary returns 2 lists of dictionaries:
    info = mods_dictionary(a1)

    # info[0] contains a list of dictionaries.
    # Each dictionary contains information of each mod pulled from meta.lsx file inside each pak
    data_list = info[0]
    # error_list = info[1] # Not needed

    # ModOrderTree = element tree object @ <node id="Module">
    ModOrderTree = tree.xpath('//node[@id="ModOrder"]')[0]

    # ModOrder = element tree object @ <children>   
    ModOrder = ModOrderTree.find('children')

    # For each dictionary inside data_list
    for mods in data_list:
        order = new_module(mods["UUID"],ModOrder)
        desc = new_moduleshortdesc(mods["Name"], mods["Author"], mods["Version"], mods["UUID"], mods["Folder"])

    # Then write to file:    
    tree.write('testwrite.xml')

generator2()

Вопросы Есть ли способ достичь того, чего я хочу?

Пожалуйста, имейте в виду, что я новичок в программировании, очень много еще учусь, поэтому я уверен, что есть более питонские способы написания кода более эффективно.Не стесняйтесь делать предложения, если я сделал что-то нубистское, чтобы беспокоить вас: p

Пытались: 

    t1 = et.tostring(tree, encoding="unicode",method="xml",pretty_print=True)
    with open(test_file,'w') as f:
        f.write(t1)

tree.write('testwrite.xml', pretty_print=True).

1 Ответ

0 голосов
/ 23 сентября 2019

Решение : (Благодаря woodm1979 Красивый XML-принтер Python с lxml )

Просто удалите каждый бит пробела из всего документа, а затем переформатируйте анализаторэто правильно: 

def reformat(file):
    generator2()

    parser = et.XMLParser(remove_blank_text=True)
    tree = et.parse(test_file,parser)
    tree.write(test_file, encoding='utf-8',pretty_print=True,xml_declaration=True)
...