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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
from __future__ import unicode_literals
import re
class _Wrapped(object):
def __init__(self, start, end):
self._start = start
self._end = end
def __call__(self):
return self._start, self._end
def _init_writers():
writers = {
"p": _Wrapped("", "\n\n"),
}
for level in range(1, 7):
writers["h{0}".format(level)] = _Wrapped("#" * level + " ", "\n\n")
return writers
_writers = _init_writers()
_default_writer = lambda: ("", "")
class MarkdownWriter(object):
def __init__(self):
self._fragments = []
self._element_stack = []
def text(self, text):
self._fragments.append(_escape_markdown(text))
def start(self, name, attributes=None):
start, end = _writers.get(name, _default_writer)()
self._fragments.append(start)
self._element_stack.append(end)
def end(self, name):
end = self._element_stack.pop()
self._fragments.append(end)
def self_closing(self, name, attributes=None):
pass
def append(self, other):
self._fragments.append(other)
def as_string(self):
return "".join(self._fragments)
def _escape_markdown(value):
return re.sub(r"([\`\*_\{\}\[\]\(\)\#\+\-\.\!])", r"\\\1", re.sub("\\\\", "\\\\\\\\", value))
|
