Coverage for cookbook/helper/mdx_attributes.py: 88%

1import markdown 

2from markdown.treeprocessors import Treeprocessor 

3 

4 

5class StyleTreeprocessor(Treeprocessor): 

6 

7 def run_processor(self, node): 

8 for child in node: 

9 if child.tag == "table": 

10 child.set("class", "table table-bordered") 

11 if child.tag == "img": 

12 child.set("class", "img-fluid") 

13 self.run_processor(child) 

14 return node 

15 

16 def run(self, root): 

17 self.run_processor(root) 

18 return root 

19 

20 

21class MarkdownFormatExtension(markdown.Extension): 

22 # md_ globals deprecated - see here: 

23 def extendMarkdown(self, md): 

24 md.treeprocessors.register( 

25 StyleTreeprocessor(), 

26 'StyleTreeprocessor', 

27 10 

28 )