diff options
Diffstat (limited to 'mammoth')
| -rw-r--r-- | mammoth/html_generation.py | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/mammoth/html_generation.py b/mammoth/html_generation.py index cf76500..b771b00 100644 --- a/mammoth/html_generation.py +++ b/mammoth/html_generation.py @@ -7,20 +7,30 @@ class HtmlGenerator(object): self._fragments = [] def text(self, text): + self._write_all() self._fragments.append(_escape_html(text)) def start(self, name): self._stack.append(_Element(name)) - self._fragments.append("<{0}>".format(name)) def end(self): element = self._stack.pop() - self._fragments.append("</{0}>".format(element.name)) + if element.written: + self._fragments.append("</{0}>".format(element.name)) def end_all(self): while self._stack: self.end() + def _write_all(self): + for element in self._stack: + if not element.written: + element.written = True + self._write_element(element) + + def _write_element(self, element): + self._fragments.append("<{0}>".format(element.name)) + def html_string(self): return "".join(self._fragments) @@ -28,6 +38,7 @@ class HtmlGenerator(object): class _Element(object): def __init__(self, name): self.name = name + self.written = False def _escape_html(text): |