Add some failover rendering options

1 files changed, 39 insertions, 16 deletions

diff --git a/src/md.py b/src/md.py
index 14d7930..365dbd0 100755
--- a/src/md.py
+++ b/src/md.py
@@ -2,6 +2,9 @@
 import os
 import os.path
 import sys
+import traceback
+import logging
+
 
 
 #from jinja2 import Environment, PackageLoader, Template, FileSystemLoader
@@ -37,24 +40,39 @@ class TocAnchorRenderer(mistune.HTMLRenderer):
 		return html
 
 	def block_code(self, code: str, info=None) -> str:
+		render_failed = True
+		render_html = "<pre><code>No render!</code></pre>"
 		if ENABLE_CODE_HL:
 			if info:
-				lexer = get_lexer_by_name(info, stripall=True)
-				#formatter = html.HtmlFormatter()
-				formatter = html.HtmlFormatter(linenos=1)
-				return highlight(code, lexer, formatter)
-			return '<pre><code>' + mistune.escape(code) + '</code></pre>'
-		else:
+				try:
+					lexer = get_lexer_by_name(info, stripall=True)
+					#formatter = html.HtmlFormatter()
+					formatter = html.HtmlFormatter(linenos=1)
+					render_html = highlight(code, lexer, formatter)
+					render_failed = False
+				except Exception as e:
+
+					#print(traceback.format_exc())
+					pass
+			else:
+				render_html = '<pre><code'
+				if info is not None:
+					info = safe_entity(info.strip())
+				if info:
+					lang = info.split(None, 1)[0]
+					render_html += ' class="language-' + lang + '"'
+				render_html = render_html + '>' + escape_text(code) + '</code></pre>\n'
+				render_failed = False
+		if render_failed:
 			#default behaviout from the renderer
-			#html = '<pre><code'
-			#if info is not None:
-			#    info = safe_entity(info.strip())
-			#if info:
-			#    lang = info.split(None, 1)[0]
-			#    html += ' class="language-' + lang + '"'
-			#return html + '>' + escape_text(code) + '</code></pre>\n'
-
-			return ""
+			render_html = '<pre><code'
+			if info is not None:
+			    info = safe_entity(info.strip())
+			if info:
+				lang = info.split(None, 1)[0]
+				render_html += ' class="language-' + lang + '"'
+			render_html = render_html + '>' + escape_text(code) + '</code></pre>\n'
+		return render_html
 
 
 ################################################################################
@@ -84,7 +102,12 @@ def generate_toc(tags):
 		if tag["type"] == "heading":
 			attrs = tag["attrs"]
 			#attrs["id"] = "toc-none"
-			headers_list.append({"header":tag["children"][0]["raw"], "level":attrs["level"]})
+			try:
+				headers_list.append({"header":tag["children"][0]["raw"], "level":attrs["level"]})
+			except Exception as e:
+				print("Cant add tag:",tag)
+			#except:
+			#	print("ERROR:",tag)
 	#print(headers_list)
 	############################
 	#generate toc