-
Notifications
You must be signed in to change notification settings - Fork 6
/
mkdcomments.py
65 lines (47 loc) · 1.79 KB
/
mkdcomments.py
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
60
61
62
63
64
65
import re
from markdown.preprocessors import Preprocessor
from markdown.postprocessors import Postprocessor
from markdown.extensions import Extension
PREFIX_PLACEHOLDER = "OMtxTKldR2f1LZ5Q"
class CommentsExtension(Extension):
def extendMarkdown(self, md):
md.registerExtension(self)
md.preprocessors.register(
CommentMunger(md), "comment_munger", 21)
md.preprocessors.register(
CommentRemover(md), "comment_remover", 19)
md.postprocessors.register(
RawCommentReplacer(md), "raw_comment_replacer", 29)
class CommentMunger(Preprocessor):
def run(self, lines):
return [re.sub(r'<!---', PREFIX_PLACEHOLDER, line) for line in lines]
class CommentRemover(Preprocessor):
def run(self, lines):
new_lines = []
is_multi = False
for line in lines:
if not is_multi:
new_line, is_multi = self._uncommenter(line)
else:
new_line, is_multi = self._unmultiliner(line)
new_lines.append(new_line)
return new_lines
def _uncommenter(self, line):
# inline
line = re.sub(r'\s*' + PREFIX_PLACEHOLDER + r'.*?-->', '', line)
# start multiline
line, count = re.subn(r'\s*' + PREFIX_PLACEHOLDER + r'.*', '', line)
return line, bool(count)
def _unmultiliner(self, line):
new_line, count = re.subn(r'.*?-->', '', line, count=1)
# end multiline
if count > 0:
return self._uncommenter(new_line)
# continue multiline
else:
return ('', True)
class RawCommentReplacer(Postprocessor):
def run(self, text):
return re.sub(PREFIX_PLACEHOLDER, '<!---', text)
def makeExtension(*args, **kwargs):
return CommentsExtension(*args, **kwargs)