plugin.py

from collections.abc import Iterable, Mapping
from typing import Any

import xmltodict

from jinjanator_plugins import (
    Formats,
    plugin_formats_hook,
)


class XMLFormat:
    name = "xml"
    suffixes: Iterable[str] | None = ".xml"
    option_names: Iterable[str] | None = "process-namespaces"

    def __init__(self, options: Iterable[str] | None) -> None:
        self.process_namespaces = False
        if options:
            for _option in options:
                self.process_namespaces = True

    def parse(
        self,
        data_string: str,
    ) -> Mapping[str, Any]:
        return xmltodict.parse(data_string, process_namespaces=self.process_namespaces)


@plugin_formats_hook
def plugin_formats() -> Formats:
    return {XMLFormat.name: XMLFormat}