Ndtyjky

Is it possible in python to pretty print the root's attributes?

I used etree to extend the attributes of the child tag and then I had overwritten the existing file with the new content. However during the first generation of the XML, we were using a template where the attributes of the root tag were listed one per line and now with the etree I don't manage to achieve the same result.

I found similar questions but they were all referring to the tutorial of etree, which I find incomplete.

Hopefully someone has found a solution for this using etree.

EDIT: This is for custom XML so HTML Tidy (which was proposed in the comments), doesn't work for this.

Thanks!

generated_descriptors = list_generated_files(generated_descriptors_folder)

counter = 0

for g in generated_descriptors:

    if counter % 20 == 0:

        print "Extending Descriptor # %s out of %s" % (counter, len(descriptor_attributes))



    with open(generated_descriptors_folder + "\" + g, 'r+b') as descriptor:

        root = etree.XML(descriptor.read(), parser=parser)



        # Go through every ContextObject to check if the block is mandatory

        for context_object in root.findall('ContextObject'):

            for attribs in descriptor_attributes:

                if attribs['descriptor_name'] == g[:-11] and context_object.attrib['name'] in attribs['attributes']['mandatoryobjects']:

                    context_object.set('allow-null', 'false')

                elif attribs['descriptor_name'] == g[:-11] and context_object.attrib['name'] not in attribs['attributes']['mandatoryobjects']:

                    context_object.set('allow-null', 'true')



        # Sort the ContextObjects based on allow-null and their name

        context_objects = root.findall('ContextObject')

        context_objects_sorted = sorted(context_objects, key=lambda c: (c.attrib['allow-null'], c.attrib['name']))



        root[:] = context_objects_sorted



        # Remove mandatoryobjects from Descriptor attributes and pretty print

        root.attrib.pop("mandatoryobjects", None)

        # paste new line here





        # Convert to string in order to write the enhanced descriptor

        xml = etree.tostring(root, pretty_print=True, encoding="UTF-8", xml_declaration=True)



        # Write the enhanced descriptor

        descriptor.seek(0)  # Set cursor at beginning of the file

        descriptor.truncate(0)  # Make sure that file is empty

        descriptor.write(xml)



        descriptor.close()



    counter+=1