Fix crash on unknown code language
This commit is contained in:
@@ -24,6 +24,7 @@ from markdown_it.token import Token
|
|||||||
from markdown_it.presets import gfm_like
|
from markdown_it.presets import gfm_like
|
||||||
from pygments import highlight
|
from pygments import highlight
|
||||||
from pygments.lexers import get_lexer_by_name
|
from pygments.lexers import get_lexer_by_name
|
||||||
|
from pygments.util import ClassNotFound
|
||||||
from pygments.formatters.html import HtmlFormatter
|
from pygments.formatters.html import HtmlFormatter
|
||||||
|
|
||||||
from .cleaner import clean_html
|
from .cleaner import clean_html
|
||||||
@@ -31,10 +32,11 @@ from .mention import init_mention
|
|||||||
|
|
||||||
|
|
||||||
def highlight_code(code, name, attrs):
|
def highlight_code(code, name, attrs):
|
||||||
if name == "":
|
try:
|
||||||
|
lexer = get_lexer_by_name(name)
|
||||||
|
except ClassNotFound:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
lexer = get_lexer_by_name(name)
|
|
||||||
formatter = HtmlFormatter()
|
formatter = HtmlFormatter()
|
||||||
|
|
||||||
return highlight(code, lexer, formatter)
|
return highlight(code, lexer, formatter)
|
||||||
|
|||||||
Reference in New Issue
Block a user