Source code for json2xml.json2xml

from typing import Any, Dict, Optional

from defusedxml.minidom import parseString
from pyexpat import ExpatError

from json2xml import dicttoxml

from .utils import InvalidDataError




class Json2xml:
    """
    Wrapper class to convert the data to xml
    """
    def __init__(
        self,
        data: dict[str, Any] | None = None,
        wrapper: str = "all",
        root: bool = True,
        pretty: bool = True,
        attr_type: bool = True,
        item_wrap: bool = True,
    ):
        self.data = data
        self.pretty = pretty
        self.wrapper = wrapper
        self.attr_type = attr_type
        self.root = root
        self.item_wrap = item_wrap



    def to_xml(self) -> Any | None:
        """
        Convert to xml using dicttoxml.dicttoxml and then pretty print it.
        """
        if self.data:
            xml_data = dicttoxml.dicttoxml(
                self.data,
                root=self.root,
                custom_root=self.wrapper,
                attr_type=self.attr_type,
                item_wrap=self.item_wrap,
            )
            if self.pretty:
                try:
                    result = parseString(xml_data).toprettyxml()
                except ExpatError:
                    raise InvalidDataError
                return result
            return xml_data
        return None