OLD | NEW |
| (Empty) |
1 """ | |
2 ========================= FOOTNOTES ================================= | |
3 | |
4 This section adds footnote handling to markdown. It can be used as | |
5 an example for extending python-markdown with relatively complex | |
6 functionality. While in this case the extension is included inside | |
7 the module itself, it could just as easily be added from outside the | |
8 module. Not that all markdown classes above are ignorant about | |
9 footnotes. All footnote functionality is provided separately and | |
10 then added to the markdown instance at the run time. | |
11 | |
12 Footnote functionality is attached by calling extendMarkdown() | |
13 method of FootnoteExtension. The method also registers the | |
14 extension to allow it's state to be reset by a call to reset() | |
15 method. | |
16 | |
17 Example: | |
18 Footnotes[^1] have a label[^label] and a definition[^!DEF]. | |
19 | |
20 [^1]: This is a footnote | |
21 [^label]: A footnote on "label" | |
22 [^!DEF]: The footnote for definition | |
23 | |
24 """ | |
25 | |
26 from __future__ import absolute_import | |
27 from __future__ import unicode_literals | |
28 from . import Extension | |
29 from ..preprocessors import Preprocessor | |
30 from ..inlinepatterns import Pattern | |
31 from ..treeprocessors import Treeprocessor | |
32 from ..postprocessors import Postprocessor | |
33 from ..util import etree, text_type | |
34 from ..odict import OrderedDict | |
35 import re | |
36 | |
37 FN_BACKLINK_TEXT = "zz1337820767766393qq" | |
38 NBSP_PLACEHOLDER = "qq3936677670287331zz" | |
39 DEF_RE = re.compile(r'[ ]{0,3}\[\^([^\]]*)\]:\s*(.*)') | |
40 TABBED_RE = re.compile(r'((\t)|( ))(.*)') | |
41 | |
42 class FootnoteExtension(Extension): | |
43 """ Footnote Extension. """ | |
44 | |
45 def __init__ (self, configs): | |
46 """ Setup configs. """ | |
47 self.config = {'PLACE_MARKER': | |
48 ["///Footnotes Go Here///", | |
49 "The text string that marks where the footnotes go"], | |
50 'UNIQUE_IDS': | |
51 [False, | |
52 "Avoid name collisions across " | |
53 "multiple calls to reset()."], | |
54 "BACKLINK_TEXT": | |
55 ["↩", | |
56 "The text string that links from the footnote to the rea
der's place."] | |
57 } | |
58 | |
59 for key, value in configs: | |
60 self.config[key][0] = value | |
61 | |
62 # In multiple invocations, emit links that don't get tangled. | |
63 self.unique_prefix = 0 | |
64 | |
65 self.reset() | |
66 | |
67 def extendMarkdown(self, md, md_globals): | |
68 """ Add pieces to Markdown. """ | |
69 md.registerExtension(self) | |
70 self.parser = md.parser | |
71 self.md = md | |
72 self.sep = ':' | |
73 if self.md.output_format in ['html5', 'xhtml5']: | |
74 self.sep = '-' | |
75 # Insert a preprocessor before ReferencePreprocessor | |
76 md.preprocessors.add("footnote", FootnotePreprocessor(self), | |
77 "<reference") | |
78 # Insert an inline pattern before ImageReferencePattern | |
79 FOOTNOTE_RE = r'\[\^([^\]]*)\]' # blah blah [^1] blah | |
80 md.inlinePatterns.add("footnote", FootnotePattern(FOOTNOTE_RE, self), | |
81 "<reference") | |
82 # Insert a tree-processor that would actually add the footnote div | |
83 # This must be before all other treeprocessors (i.e., inline and | |
84 # codehilite) so they can run on the the contents of the div. | |
85 md.treeprocessors.add("footnote", FootnoteTreeprocessor(self), | |
86 "_begin") | |
87 # Insert a postprocessor after amp_substitute oricessor | |
88 md.postprocessors.add("footnote", FootnotePostprocessor(self), | |
89 ">amp_substitute") | |
90 | |
91 def reset(self): | |
92 """ Clear the footnotes on reset, and prepare for a distinct document. "
"" | |
93 self.footnotes = OrderedDict() | |
94 self.unique_prefix += 1 | |
95 | |
96 def findFootnotesPlaceholder(self, root): | |
97 """ Return ElementTree Element that contains Footnote placeholder. """ | |
98 def finder(element): | |
99 for child in element: | |
100 if child.text: | |
101 if child.text.find(self.getConfig("PLACE_MARKER")) > -1: | |
102 return child, element, True | |
103 if child.tail: | |
104 if child.tail.find(self.getConfig("PLACE_MARKER")) > -1: | |
105 return child, element, False | |
106 finder(child) | |
107 return None | |
108 | |
109 res = finder(root) | |
110 return res | |
111 | |
112 def setFootnote(self, id, text): | |
113 """ Store a footnote for later retrieval. """ | |
114 self.footnotes[id] = text | |
115 | |
116 def makeFootnoteId(self, id): | |
117 """ Return footnote link id. """ | |
118 if self.getConfig("UNIQUE_IDS"): | |
119 return 'fn%s%d-%s' % (self.sep, self.unique_prefix, id) | |
120 else: | |
121 return 'fn%s%s' % (self.sep, id) | |
122 | |
123 def makeFootnoteRefId(self, id): | |
124 """ Return footnote back-link id. """ | |
125 if self.getConfig("UNIQUE_IDS"): | |
126 return 'fnref%s%d-%s' % (self.sep, self.unique_prefix, id) | |
127 else: | |
128 return 'fnref%s%s' % (self.sep, id) | |
129 | |
130 def makeFootnotesDiv(self, root): | |
131 """ Return div of footnotes as et Element. """ | |
132 | |
133 if not list(self.footnotes.keys()): | |
134 return None | |
135 | |
136 div = etree.Element("div") | |
137 div.set('class', 'footnote') | |
138 etree.SubElement(div, "hr") | |
139 ol = etree.SubElement(div, "ol") | |
140 | |
141 for id in self.footnotes.keys(): | |
142 li = etree.SubElement(ol, "li") | |
143 li.set("id", self.makeFootnoteId(id)) | |
144 self.parser.parseChunk(li, self.footnotes[id]) | |
145 backlink = etree.Element("a") | |
146 backlink.set("href", "#" + self.makeFootnoteRefId(id)) | |
147 if self.md.output_format not in ['html5', 'xhtml5']: | |
148 backlink.set("rev", "footnote") # Invalid in HTML5 | |
149 backlink.set("class", "footnote-backref") | |
150 backlink.set("title", "Jump back to footnote %d in the text" % \ | |
151 (self.footnotes.index(id)+1)) | |
152 backlink.text = FN_BACKLINK_TEXT | |
153 | |
154 if li.getchildren(): | |
155 node = li[-1] | |
156 if node.tag == "p": | |
157 node.text = node.text + NBSP_PLACEHOLDER | |
158 node.append(backlink) | |
159 else: | |
160 p = etree.SubElement(li, "p") | |
161 p.append(backlink) | |
162 return div | |
163 | |
164 | |
165 class FootnotePreprocessor(Preprocessor): | |
166 """ Find all footnote references and store for later use. """ | |
167 | |
168 def __init__ (self, footnotes): | |
169 self.footnotes = footnotes | |
170 | |
171 def run(self, lines): | |
172 """ | |
173 Loop through lines and find, set, and remove footnote definitions. | |
174 | |
175 Keywords: | |
176 | |
177 * lines: A list of lines of text | |
178 | |
179 Return: A list of lines of text with footnote definitions removed. | |
180 | |
181 """ | |
182 newlines = [] | |
183 i = 0 | |
184 while True: | |
185 m = DEF_RE.match(lines[i]) | |
186 if m: | |
187 fn, _i = self.detectTabbed(lines[i+1:]) | |
188 fn.insert(0, m.group(2)) | |
189 i += _i-1 # skip past footnote | |
190 self.footnotes.setFootnote(m.group(1), "\n".join(fn)) | |
191 else: | |
192 newlines.append(lines[i]) | |
193 if len(lines) > i+1: | |
194 i += 1 | |
195 else: | |
196 break | |
197 return newlines | |
198 | |
199 def detectTabbed(self, lines): | |
200 """ Find indented text and remove indent before further proccesing. | |
201 | |
202 Keyword arguments: | |
203 | |
204 * lines: an array of strings | |
205 | |
206 Returns: a list of post processed items and the index of last line. | |
207 | |
208 """ | |
209 items = [] | |
210 blank_line = False # have we encountered a blank line yet? | |
211 i = 0 # to keep track of where we are | |
212 | |
213 def detab(line): | |
214 match = TABBED_RE.match(line) | |
215 if match: | |
216 return match.group(4) | |
217 | |
218 for line in lines: | |
219 if line.strip(): # Non-blank line | |
220 detabbed_line = detab(line) | |
221 if detabbed_line: | |
222 items.append(detabbed_line) | |
223 i += 1 | |
224 continue | |
225 elif not blank_line and not DEF_RE.match(line): | |
226 # not tabbed but still part of first par. | |
227 items.append(line) | |
228 i += 1 | |
229 continue | |
230 else: | |
231 return items, i+1 | |
232 | |
233 else: # Blank line: _maybe_ we are done. | |
234 blank_line = True | |
235 i += 1 # advance | |
236 | |
237 # Find the next non-blank line | |
238 for j in range(i, len(lines)): | |
239 if lines[j].strip(): | |
240 next_line = lines[j]; break | |
241 else: | |
242 break # There is no more text; we are done. | |
243 | |
244 # Check if the next non-blank line is tabbed | |
245 if detab(next_line): # Yes, more work to do. | |
246 items.append("") | |
247 continue | |
248 else: | |
249 break # No, we are done. | |
250 else: | |
251 i += 1 | |
252 | |
253 return items, i | |
254 | |
255 | |
256 class FootnotePattern(Pattern): | |
257 """ InlinePattern for footnote markers in a document's body text. """ | |
258 | |
259 def __init__(self, pattern, footnotes): | |
260 super(FootnotePattern, self).__init__(pattern) | |
261 self.footnotes = footnotes | |
262 | |
263 def handleMatch(self, m): | |
264 id = m.group(2) | |
265 if id in self.footnotes.footnotes.keys(): | |
266 sup = etree.Element("sup") | |
267 a = etree.SubElement(sup, "a") | |
268 sup.set('id', self.footnotes.makeFootnoteRefId(id)) | |
269 a.set('href', '#' + self.footnotes.makeFootnoteId(id)) | |
270 if self.footnotes.md.output_format not in ['html5', 'xhtml5']: | |
271 a.set('rel', 'footnote') # invalid in HTML5 | |
272 a.set('class', 'footnote-ref') | |
273 a.text = text_type(self.footnotes.footnotes.index(id) + 1) | |
274 return sup | |
275 else: | |
276 return None | |
277 | |
278 | |
279 class FootnoteTreeprocessor(Treeprocessor): | |
280 """ Build and append footnote div to end of document. """ | |
281 | |
282 def __init__ (self, footnotes): | |
283 self.footnotes = footnotes | |
284 | |
285 def run(self, root): | |
286 footnotesDiv = self.footnotes.makeFootnotesDiv(root) | |
287 if footnotesDiv: | |
288 result = self.footnotes.findFootnotesPlaceholder(root) | |
289 if result: | |
290 child, parent, isText = result | |
291 ind = parent.getchildren().index(child) | |
292 if isText: | |
293 parent.remove(child) | |
294 parent.insert(ind, footnotesDiv) | |
295 else: | |
296 parent.insert(ind + 1, footnotesDiv) | |
297 child.tail = None | |
298 else: | |
299 root.append(footnotesDiv) | |
300 | |
301 class FootnotePostprocessor(Postprocessor): | |
302 """ Replace placeholders with html entities. """ | |
303 def __init__(self, footnotes): | |
304 self.footnotes = footnotes | |
305 | |
306 def run(self, text): | |
307 text = text.replace(FN_BACKLINK_TEXT, self.footnotes.getConfig("BACKLINK
_TEXT")) | |
308 return text.replace(NBSP_PLACEHOLDER, " ") | |
309 | |
310 def makeExtension(configs=[]): | |
311 """ Return an instance of the FootnoteExtension """ | |
312 return FootnoteExtension(configs=configs) | |
313 | |
OLD | NEW |