From 2ddcbfb5ab778da5d8e9e2f85d44ee89ed7dd7e8 Mon Sep 17 00:00:00 2001 From: rubenwardy Date: Wed, 4 Jun 2025 18:13:59 +0100 Subject: [PATCH] Fix crash on unknown code language --- app/markdown/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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)