diff options
Diffstat (limited to 'src/md.py')
| -rwxr-xr-x | src/md.py | 178 |
1 files changed, 76 insertions, 102 deletions
@@ -12,103 +12,17 @@ import mistune #global name for templates article = {}# -#class TocMixin(): -# """TOC mixin for Renderer, mix this with Renderer::# - -# class TocRenderer(TocMixin, Renderer): -# pass# - -# toc = TocRenderer() -# md = mistune.Markdown(renderer=toc)# - -# # required in this order -# toc.reset_toc() # initial the status -# md.parse(text) # parse for headers -# toc.render_toc(level=3) # render TOC HTML -# """# - -# def reset_toc(self): -# self.toc_tree = [] -# self.toc_count = 0 -# self.title = None -# self.img_count = -1# - -# def header(self, text, level, raw=None): -# rv = "" -# if level != 1: -# rv = '<h%d id="toc-%d">%s</h%d>\n' % ( -# level, self.toc_count, text, level -# ) -# else: -# self.title = text -# self.toc_tree.append((self.toc_count, text, level, raw)) -# self.toc_count += 1 -# return rv# - -# #Lets do previev image -# def image(self, src, title, alt_text): -# prev_img = src.split(".") -# prev_img = prev_img[0]+"_prev."+prev_img[1] -# self.img_count += 1 -# return "<a href="+src+"><img src=\""+prev_img+"\" alt=\""+alt_text+"\"></a>"# - -# def render_toc(self, level=3): -# """Render TOC to HTML.# - -# :param level: render toc to the given level -# """ -# return ''.join(self._iter_toc(level))# - -# def _iter_toc(self, level): -# first_level = None -# last_level = None -# cnt = 0# - -# if (self.title != None): -# yield '<h1>%s</h1>\n' % (self.title)# - -# yield '<ul id="table-of-content">\n'# - -# for toc in self.toc_tree: -# index, text, l, raw = toc -# index = cnt -# # - -# if l > level: -# # ignore this level -# continue# - -# if first_level is None: -# # based on first level -# first_level = l -# last_level = l -# #yield '<li><a href="#toc-%d">%s</a>' % (cnt, text) -# yield '' -# elif last_level == l: -# yield '</li>\n<li><a href="#toc-%d">%s</a>' % (cnt, text) -# elif last_level == l - 1: -# last_level = l -# yield '<ul>\n<li><a href="#toc-%d">%s</a>' % (cnt, text) -# elif last_level > l: -# # close indention -# yield '</li>' -# while last_level > l: -# yield '</ul>\n</li>\n' -# last_level -= 1 -# yield '<li><a href="#toc-%d">%s</a>' % (cnt, text) -# cnt = cnt + 1# - -# # close tags -# yield '</li>\n' -# while last_level > first_level: -# yield '</ul>\n<!--</li>-->\n' -# last_level -= 1# - -# yield '</ul>\n' -# self.toc_count = 0# - -#class TocRenderer(TocMixin, mistune.Renderer): -# pass +class TocAnchorRenderer(mistune.HTMLRenderer): + count = 1 + def heading(self, text, level, **attrs): + tag = 'h' + str(level) + html = '<' + tag + _id = attrs.get('id') + #if _id: + # html += ' id="toc-' + _id + '"' + html += ' id="toc-' + str(self.count) + '"' + self.count += 1 + return html + '>' + text + '</' + tag + '>\n' ################################################################################ #get first tags and use them to configure some bits @@ -129,6 +43,53 @@ def get_tags(data): return text ################################################################################ +# +def generate_toc(tags): + cnt = 3 + headers_list = [] + for tag in tags: + if tag["type"] == "heading": + attrs = tag["attrs"] + #attrs["id"] = "toc-none" + headers_list.append({"header":tag["children"][0]["raw"], "level":attrs["level"]}) + #print(headers_list) + ############################ + #generate toc + html_toc = '<ul id="table-of-content">\n' + first_level = None + last_level = None + cnt = 1 + for el in headers_list: + l = int(el["level"]) + + if first_level is None: + first_level = l + last_level = l + #same level + elif last_level == l: + html_toc += '</li>\n<li><a href="#toc-%d">%s</a>' % (cnt, el["header"] ) + # if last level smaller then open one more + elif last_level == l - 1: + last_level = l + html_toc += '<ul>\n<li><a href="#toc-%d">%s</a>' % (cnt, el["header"]) + elif last_level > l: + # close indention + html_toc += '</li>' + while last_level > l: + html_toc += '</ul>\n</li>\n' + last_level -= 1 + html_toc += '<li><a href="#toc-%d">%s</a>' % (cnt, el["header"]) + cnt += 1 + + html_toc += '</li>\n' + while last_level > first_level: + html_toc += '</ul>\n<!--</li>-->\n' + last_level -= 1 + + html_toc += '</ul>' + return html_toc + +################################################################################ #check if there is input file if len(sys.argv) < 2: print("ERROR: Exit . Need more arguments") @@ -138,9 +99,9 @@ md_fn = sys.argv[1] #template loader loader = FileSystemLoader( "/home/fam/downloads/source/repos/md-site/src/templ" ) -templ_env = Environment( loader = loader )# +templ_env = Environment( loader = loader ) -t = templ_env.get_template("main.thtml")# +t = templ_env.get_template("main.thtml") ##get md file f = open( md_fn, "r" ) @@ -148,13 +109,26 @@ data = f.read() data = get_tags(data) #toc = TocRenderer() ## Create AST rendered +markdown_raw = mistune.create_markdown(renderer=None) +raw_tags = markdown_raw(data) +html_toc = generate_toc(raw_tags) +########################### ## Create AST -> Markdown renderer - -## Mistune generate MTL -md_rend = mistune.html(data) +#markdown_ast = mistune.create_markdown(renderer='ast') +#print(markdown_ast(raw_tags)) + +custom_render = True +if not custom_render: + ## Mistune generate MTL + md_rend = html_toc + mistune.html(data) +else: + ## Mistune with custom renderer + markdown = mistune.create_markdown(renderer=TocAnchorRenderer()) + md_rend = html_toc + markdown(data) #print t.render( article=article, block = md_rend ) print(t.render( article=article, block = md_rend )) + f.close()
\ No newline at end of file |