aboutsummaryrefslogtreecommitdiff
path: root/mammoth/html_generation.py
blob: cf765008dbe6f65599b6f68c60ab45091a6f92af (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import cgi


class HtmlGenerator(object):
    def __init__(self):
        self._stack = []
        self._fragments = []
    
    def text(self, text):
        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))
    
    def end_all(self):
        while self._stack:
            self.end()
    
    def html_string(self):
        return "".join(self._fragments)


class _Element(object):
    def __init__(self, name):
        self.name = name


def _escape_html(text):
    return cgi.escape(text, quote=True)