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
49
50
51
52
53
|
#
# Copyright (C) 2009 Peter Bui <pnutzh4x0r@gmail.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
import cgi
from pygments import highlight as _highlight_func
from pygments.lexers import guess_lexer, guess_lexer_for_filename, TextLexer
from pygments.formatters import HtmlFormatter
from pygments.util import ClassNotFound
DEFAULT_PYGMENT_STYLE = 'colorful'
# Trying to highlight very large files using pygments was killing
# loggerhead on launchpad.net, because pygments isn't very fast.
# So we only highlight files if they're 512K or smaller.
MAX_HIGHLIGHT_SIZE = 512000;
def highlight(path, text, encoding, style=DEFAULT_PYGMENT_STYLE):
"""
Returns a list of highlighted (i.e. HTML formatted) strings.
"""
if len(text) > MAX_HIGHLIGHT_SIZE:
return map(cgi.escape, text.split('\n'))
formatter = HtmlFormatter(style=style, nowrap=True, classprefix='pyg-')
try:
lexer = guess_lexer_for_filename(path, text[:1024], encoding=encoding)
except (ClassNotFound, ValueError):
try:
lexer = guess_lexer(text[:1024], encoding=encoding)
except (ClassNotFound, ValueError):
lexer = TextLexer(encoding=encoding)
hl_lines = _highlight_func(text, lexer, formatter).split('\n')
return hl_lines
|