diff --git a/app/markdown/__init__.py b/app/markdown/__init__.py index 2a1389a2..1692c9ce 100644 --- a/app/markdown/__init__.py +++ b/app/markdown/__init__.py @@ -24,6 +24,7 @@ from markdown_it.token import Token from markdown_it.presets import gfm_like from pygments import highlight from pygments.lexers import get_lexer_by_name +from pygments.util import ClassNotFound from pygments.formatters.html import HtmlFormatter from .cleaner import clean_html @@ -31,10 +32,11 @@ from .mention import init_mention def highlight_code(code, name, attrs): - if name == "": + try: + lexer = get_lexer_by_name(name) + except ClassNotFound: return None - lexer = get_lexer_by_name(name) formatter = HtmlFormatter() return highlight(code, lexer, formatter)