blob: 291528e2e4093564f128617fdda7ee20dc88bd9d (
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
48
|
import sys
from django import template
from django.utils.encoding import smart_unicode, smart_str
from django.utils.html import conditional_escape
from django.utils.safestring import mark_safe
register = template.Library()
try:
from pygments import highlight
from pygments.lexers import get_lexer_by_name
from pygments.formatters import HtmlFormatter
colorize = True
except:
colorize = False
def u_str(string):
if sys.hexversion >= 0x03000000:
return string
else:
return unicode(string)
@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
|