pretty_xml/pretty_xml.py - master - ilya.petrash/SimplePyScripts

pretty_xml/pretty_xml.py
67 строк2 KB
gil9red
Refactoring. pretty_xml/pretty_xml.py
12 апр 2025, 23:38
12 апр 2025, 23:3849e2631
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = "ipetrash"


from xml.dom.minidom import parseString
from lxml import etree


def process_xml_string(xml_string: str) -> str:
    """
    Функция из текста выдирает строку с xml --
    она должна начинаться на < и заканчиваться >

    """

    start = xml_string.index("<")
    end = xml_string.rindex(">")
    return xml_string[start : end + 1]


def pretty_xml_minidom(xml_string: str, indent: int = 4) -> str:
    """Функция принимает строку xml и выводит xml с отступами."""

    xml_string: str = process_xml_string(xml_string)

    xml_utf8 = parseString(xml_string).toprettyxml(
        indent=" " * indent,
        encoding="utf-8",
    )
    return xml_utf8.decode("utf-8")


def pretty_xml_lxml(xml_string: str) -> str:
    """Функция принимает строку xml и выводит xml с отступами."""

    xml_string: str = process_xml_string(xml_string)

    root = etree.fromstring(
        text=xml_string,
        parser=etree.XMLParser(
            remove_blank_text=True,
            # NOTE: Для обработки больших XML
            huge_tree=True,
        ),
    )

    return etree.tostring(
        root,
        pretty_print=True,
        encoding="utf-8",
    ).decode("utf-8")


if __name__ == "__main__":
    xml = """
        <Recipe name="хлеб" preptime="5min" cooktime="180min"><Title>Простой хлеб</Title>
        <Composition><Ingredient amount="3" unit="стакан">Мука</Ingredient><Ingredient amount="0.25" 
        unit="грамм">Дрожжи</Ingredient><Ingredient amount="1.5" unit="стакан">Тёплая вода</Ingredient>
        </Composition></Recipe>
    """

    # TODO: Плохо работает
    print(pretty_xml_minidom(xml))

    print(pretty_xml_lxml(xml))