mirror of
https://gitlab.gnome.org/jwestman/blueprint-compiler.git
synced 2025-05-04 15:59:08 -04:00
63 lines
2 KiB
Python
63 lines
2 KiB
Python
# xml_emitter.py
|
|
#
|
|
# Copyright 2021 James Westman <james@jwestman.net>
|
|
#
|
|
# This file is free software; you can redistribute it and/or modify it
|
|
# under the terms of the GNU Lesser General Public License as
|
|
# published by the Free Software Foundation; either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This file is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public
|
|
# License along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# SPDX-License-Identifier: LGPL-3.0-or-later
|
|
|
|
|
|
from xml.sax import saxutils
|
|
|
|
|
|
class XmlEmitter:
|
|
def __init__(self, indent=2):
|
|
self.indent = indent
|
|
self.result = '<?xml version="1.0" encoding="UTF-8"?>'
|
|
self._tag_stack = []
|
|
self._needs_newline = False
|
|
|
|
def start_tag(self, tag, **attrs):
|
|
self._indent()
|
|
self.result += f"<{tag}"
|
|
for key, val in attrs.items():
|
|
if val is not None:
|
|
self.result += f' {key}="{saxutils.escape(str(val))}"'
|
|
self.result += ">"
|
|
self._tag_stack.append(tag)
|
|
self._needs_newline = False
|
|
|
|
def put_self_closing(self, tag, **attrs):
|
|
self._indent()
|
|
self.result += f"<{tag}"
|
|
for key, val in attrs.items():
|
|
if val is not None:
|
|
self.result += f' {key}="{saxutils.escape(str(val))}"'
|
|
self.result += "/>"
|
|
self._needs_newline = True
|
|
|
|
def end_tag(self):
|
|
tag = self._tag_stack.pop()
|
|
if self._needs_newline:
|
|
self._indent()
|
|
self.result += f"</{tag}>"
|
|
self._needs_newline = True
|
|
|
|
def put_text(self, text):
|
|
self.result += saxutils.escape(str(text))
|
|
self._needs_newline = False
|
|
|
|
def _indent(self):
|
|
if self.indent is not None:
|
|
self.result += "\n" + " " * (self.indent * len(self._tag_stack))
|