-
Notifications
You must be signed in to change notification settings - Fork 0
/
myextension.py
107 lines (91 loc) · 4.47 KB
/
myextension.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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# existing markdown inlinePatterns
# https://github.com/Python-Markdown/markdown/blob/2.6/markdown/inlinepatterns.py
from markdown.extensions import Extension
from markdown.inlinepatterns import SimpleTagPattern, Pattern
from markdown.inlinepatterns import SubstituteTagPattern
from markdown.util import etree
from markdown import util
DEL_RE = r'(~)(.*?)~' # Strikeout in slack
INS_RE = r'(__)(.*?)__' # not slack ;-)
STRONG_RE = r'(\*)(.*?)\*' # Bold in slack
EMPH_RE = r'(_)(.*?)_' # Italics in slack
CODE_RE = r'(`)(.*?)`' # code in slack
PREFORMATTED_RE = r'(```)(.*?)```' # preformatted in slack
# NEWLINE_RE = r'\n' # newline in slack
USERNAME_RE = r'(<@)(.*?)>' # username tag
CHANNEL_RE = r'(<#.+?\|)(.*?)>' # username tag
CHANNEL_2_RE = r'(<#)(.*?)>' # username tag
class MyExtension(Extension):
def __init__(self, *args, **kwargs):
# Define config options and defaults
self.config = {
'data_for_replacing_text': ['it shall be a list', 'To provide data_for_replacing_text data']
}
# Call the parent class's __init__ method to configure options
super(MyExtension, self).__init__(*args, **kwargs)
def extendMarkdown(self, md, md_globals):
data_for_replacing_text = self.getConfig('data_for_replacing_text')
del_tag = SimpleTagPattern(DEL_RE, 'del')
md.inlinePatterns.add('del', del_tag, '>not_strong')
ins_tag = SimpleTagPattern(INS_RE, 'ins')
md.inlinePatterns.add('ins', ins_tag, '>del')
strong_tag = SimpleTagPattern(STRONG_RE, 'strong')
md.inlinePatterns['strong'] = strong_tag
emph_tag = SimpleTagPattern(EMPH_RE, 'em')
md.inlinePatterns.add('em', emph_tag, '>del')
preformatted_tag = SimpleTagPattern(PREFORMATTED_RE, 'pre')
md.inlinePatterns.add('preformatted', preformatted_tag, '<backtick')
# newline_tag = SubstituteTagPattern(NEWLINE_RE, 'br')
# md.inlinePatterns.add('linebreak2', newline_tag, '>linebreak')
if isinstance(data_for_replacing_text, list):
username_tag = SimpleTagPatternWithClassOptionsAndData(USERNAME_RE, 'span', 'username', data_for_replacing_text)
md.inlinePatterns.add('username', username_tag, '<link')
channel_tag = SimpleTagPatternWithClassOptionsAndData(CHANNEL_RE, 'span', 'channel', data_for_replacing_text)
md.inlinePatterns.add('channel', channel_tag, '<username')
channel_2_tag = SimpleTagPatternWithClassOptionsAndData(CHANNEL_2_RE, 'span', 'channel', data_for_replacing_text)
md.inlinePatterns.add('channel_2', channel_2_tag, '>channel')
else:
username_tag = SimpleTagPatternWithClassOptions(USERNAME_RE, 'span', 'username')
md.inlinePatterns.add('username', username_tag, '<link')
channel_tag = SimpleTagPatternWithClassOptions(CHANNEL_RE, 'span', 'channel')
md.inlinePatterns.add('channel', channel_tag, '<username')
channel_2_tag = SimpleTagPatternWithClassOptions(CHANNEL_2_RE, 'span', 'channel')
md.inlinePatterns.add('channel_2', channel_2_tag, '>channel')
class SimpleTagPatternWithClassOptions(Pattern):
"""
Return element of type `tag` with a text attribute of group(3)
of a Pattern.
"""
def __init__(self, pattern, tag, class_name_in_html):
Pattern.__init__(self, pattern)
self.tag = tag
self.class_name_in_html = class_name_in_html
def handleMatch(self, m):
el = util.etree.Element(self.tag)
el.text = m.group(3)
el.set('class', self.class_name_in_html)
return el
class SimpleTagPatternWithClassOptionsAndData(Pattern):
"""
Return element of type `tag` with input text
of a Pattern.
"""
def __init__(self, pattern, tag, class_name_in_html, data_for_replacing_text):
Pattern.__init__(self, pattern)
self.tag = tag
self.class_name_in_html = class_name_in_html
self.data_for_replacing_text = data_for_replacing_text
def handleMatch(self, m):
el = util.etree.Element(self.tag)
data_id = m.group(3)
datum_for_replacing_text_name = self.get_datum_text(self.data_for_replacing_text, data_id)
el.text = datum_for_replacing_text_name
el.set('class', self.class_name_in_html)
return el
def get_datum_text(self, data_for_replacing_text, data_id):
datum_for_replacing_text_name = data_id
for datum_for_replacing_text in data_for_replacing_text:
if datum_for_replacing_text.get('data_id') == data_id:
datum_for_replacing_text_name = datum_for_replacing_text.get('text')
break
return datum_for_replacing_text_name