-
Notifications
You must be signed in to change notification settings - Fork 162
/
smartImg.py
93 lines (81 loc) · 3.5 KB
/
smartImg.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
#! /usr/bin/env python
import markdown
from markdown.treeprocessors import Treeprocessor
from markdown.blockprocessors import BlockProcessor
import re
from markdown import util
class SmartImgProcessor(Treeprocessor):
def __init__(self, parser, configs):
Treeprocessor.__init__(self, parser)
self.configs = configs
def run(self, root):
elemsToInspect = [root]
while len(elemsToInspect) > 0:
elem = elemsToInspect.pop()
for nelem in elem:
if nelem.tag in self.configs["PARENTS"]:
elemsToInspect.append(nelem)
elif nelem.tag == "p" and len(list(nelem.itertext())) == 0 :
lelems = list(nelem.iter())
if (len(lelems) == 1 or (len(lelems)==2 and lelems[0] is nelem)) \
and lelems[-1].tag == "img" \
and lelems[-1].attrib["alt"] != "" \
and not (lelems[-1].attrib["src"] in self.configs["IGNORING_IMG"]):
oldImg = lelems[-1]
nelem.remove(oldImg)
nFig = util.etree.Element("figure")
nFigCaption = util.etree.Element("figurecaption")
nFigCaption.text = oldImg.attrib["alt"]
oldImg.attrib["alt"]=""
nFig.append(oldImg)
nFig.append(nFigCaption)
nelem.append(nFig)
return root
class FigureProcessor(BlockProcessor):
def __init__(self, parser, md):
BlockProcessor.__init__(self, parser)
self.md = md
self.imPro = md.inlinePatterns["image_link"]
self.RE = re.compile(r'\n(Figure[ ]{0,1})*\:(?P<txtlegend>.*?)(\n|$)')
def test(self, parent, block):
mImg = self.imPro.getCompiledRegExp().match(block)
if mImg is None:
return False
mLeg = self.RE.search(block)
if mLeg is None:
return False
return mImg.start(11) == mLeg.start()
def run(self, parent, blocks):
block = blocks.pop(0)
mImg = self.imPro.getCompiledRegExp().match(block)
mLeg = self.RE.search(block)
before = block[:mImg.end(1)]
after = block[mLeg.end():]
content = block[mImg.start():mLeg.end()]
contentImg = block[mImg.end(1):mLeg.start()]
if before:
self.parser.parseBlocks(parent, [before])
nFig = util.etree.Element("figure")
nFigCaption = util.etree.Element("figurecaption")
self.parser.parseChunk(nFigCaption, mLeg.group('txtlegend'))
self.parser.parseChunk(nFig, contentImg)
nFig.append(nFigCaption)
parent.append(nFig)
if after:
blocks.insert(0,after)
class SmartImgExtension(markdown.extensions.Extension):
def __init__(self, configs={}):
self.configs = {
"IGNORING_IMG" : [],
"PARENTS" : [],
}
for key, value in configs.iteritems():
self.configs[key] = value
if "div" not in self.configs["PARENTS"]:
self.configs["PARENTS"].append("div")
def extendMarkdown(self, md, md_globals):
md.registerExtension(self)
md.treeprocessors.add('smart-img', SmartImgProcessor(md.parser,self.configs),"_end")
md.parser.blockprocessors.add('figure-processor', FigureProcessor(md.parser,md),"_begin")
def makeExtension(configs={}):
return SmartImgExtension(configs=configs)