blob: bd379b98de0eb20c6516728aebc5bcd53919953b (
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
47
|
import sys
from django import template
from django.utils.encoding import smart_unicode
from django.utils.html import conditional_escape
from django.utils.safestring import mark_safe
from Bcfg2.Compat import u_str
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:
value = conditional_escape(value)
arg = conditional_escape(arg)
if colorize:
try:
output = u_str('<style type="text/css">') \
+ smart_unicode(HtmlFormatter().get_style_defs('.highlight')) \
+ u_str('</style>')
lexer = get_lexer_by_name(arg)
output += highlight(value, lexer, HtmlFormatter())
return mark_safe(output)
except:
return value
else:
return mark_safe(u_str('<div class="note-box">Tip: Install pygments '
'for highlighting</div><pre>%s</pre>') % value)
syntaxhilight.needs_autoescape = True
|