lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix pygments latest version

Browse source
This commit is contained in:
cuom1999 2022-09-15 12:01:29 -05:00
parent 979b9862fb
commit 1c0686e30a
1 changed files with 3 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
judge/highlight_code.py
View file

@ -7,17 +7,10 @@ def _make_pre_code(code):
return mark_safe("<pre>" + escape(code) + "</pre>")
def _wrap_code(inner):
yield 0, "<code>"
for tup in inner:
yield tup
yield 0, "</code>"
try:
import pygments
import pygments.lexers
import pygments.formatters.html
import pygments.formatters
import pygments.util
except ImportError:
@ -25,11 +18,6 @@ except ImportError:
return _make_pre_code(code)
else:
class HtmlCodeFormatter(pygments.formatters.HtmlFormatter):
def wrap(self, source, outfile):
return self._wrap_div(self._wrap_pre(_wrap_code(source)))
def highlight_code(code, language, cssclass="codehilite", linenos=True):
try:
lexer = pygments.lexers.get_lexer_by_name(language)
@ -39,9 +27,9 @@ else:
if linenos:
return mark_safe(
pygments.highlight(
code, lexer, HtmlCodeFormatter(cssclass=cssclass, linenos="table")
code, lexer, HtmlCodeFormatter(cssclass=cssclass, linenos="table", wrapcode=True)
)
)
return mark_safe(
pygments.highlight(code, lexer, HtmlCodeFormatter(cssclass=cssclass))
pygments.highlight(code, lexer, HtmlCodeFormatter(cssclass=cssclass, wrapcode=True))
)