From ab163f6b3dc4e8d319e47ccce2b5a3c81a2cf27c Mon Sep 17 00:00:00 2001 From: Ben Kurtovic Date: Sat, 5 Jul 2014 21:34:55 -0400 Subject: [PATCH] Fix lexer selection a bit. --- bitshift/codelet.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/bitshift/codelet.py b/bitshift/codelet.py index 5fbf164..664599c 100644 --- a/bitshift/codelet.py +++ b/bitshift/codelet.py @@ -1,7 +1,7 @@ from operator import concat from pygments import highlight -from pygments.lexers import get_lexer_by_name +from pygments.lexers import find_lexer_class, get_lexer_by_name from pygments.formatters.html import HtmlFormatter from .languages import LANGS @@ -88,11 +88,12 @@ class Codelet(object): lang = LANGS[self.language] code = self.code if highlight_code: + lexer = find_lexer_class(lang) or get_lexer_by_name("text") symbols = reduce(concat, self.symbols.values(), []) lines = reduce(concat, [[loc[0] for loc in sym[1] + sym[2]] for sym in symbols], []) formatter = HtmlFormatter(linenos=True, hl_lines=lines) - code = highlight(code, get_lexer_by_name(lang.lower()), formatter) + code = highlight(code, lexer, formatter) return { "name": self.name, "code": code, "lang": lang,