blob: 22700689fa862e6ab57d1a98b1e7d7ffe3a43dfd (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
from django import template
from django.utils.encoding import smart_str
from django.utils.html import conditional_escape
from django.utils.safestring import mark_safe
register = template.Library()
# pylint: disable=E0611
try:
from pygments import highlight
from pygments.lexers import get_lexer_by_name
from pygments.formatters import HtmlFormatter
colorize = True
except:
colorize = False
# pylint: enable=E0611
@register.filter
def syntaxhilight(value, arg="diff", autoescape=None):
"""
Returns a syntax-hilighted version of Code;
requires code/language arguments
"""
if autoescape:
# Seems to cause a double escape
#value = conditional_escape(value)
arg = conditional_escape(arg)
if colorize:
try:
output = smart_str('<style type="text/css">') \
+ smart_str(HtmlFormatter().get_style_defs('.highlight')) \
+ smart_str('</style>')
lexer = get_lexer_by_name(arg)
output += highlight(value, lexer, HtmlFormatter())
return mark_safe(output)
except:
return value
else:
return mark_safe(smart_str(
'<div class="note-box">Tip: Install pygments '
'for highlighting</div><pre>%s</pre>') % value)
syntaxhilight.needs_autoescape = True
|