~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to loggerhead/highlight.py

• browse files at revision 458
• compare with another revision
• download diff
• download tarball
• view history from revision 458

• reverse the comparison (458 to 142.1.2)
• stop comparing with revision 142.1.2

Show diffs side-by-side

added

removed

loggerhead/highlight.py

 

1

#

 

2

# Copyright (C) 2009  Peter Bui <pnutzh4x0r@gmail.com>

 

3

#

 

4

# This program is free software; you can redistribute it and/or modify

 

5

# it under the terms of the GNU General Public License as published by

 

6

# the Free Software Foundation; either version 2 of the License, or

 

7

# (at your option) any later version.

 

8

#

 

9

# This program is distributed in the hope that it will be useful,

 

10

# but WITHOUT ANY WARRANTY; without even the implied warranty of

 

11

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 

12

# GNU General Public License for more details.

 

13

#

 

14

# You should have received a copy of the GNU General Public License

 

15

# along with this program; if not, write to the Free Software

 

16

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 

17

#

 

18

 

 

19

import bzrlib.osutils

 

20

import cgi

 

21

 

 

22

from pygments import highlight as _highlight_func

 

23

from pygments.lexers import guess_lexer, guess_lexer_for_filename, TextLexer

 

24

from pygments.formatters import HtmlFormatter

 

25

from pygments.util import ClassNotFound

 

26

 

 

27

DEFAULT_PYGMENT_STYLE = 'colorful'

 

28

 

 

29

# Trying to highlight very large files using pygments was killing

 

30

# loggerhead on launchpad.net, because pygments isn't very fast.

 

31

# So we only highlight files if they're 512K or smaller.

 

32

MAX_HIGHLIGHT_SIZE = 512000;

 

33

 

 

34

def highlight(path, text, encoding, style=DEFAULT_PYGMENT_STYLE):

 

35

    """

 

36

    Returns a list of highlighted (i.e. HTML formatted) strings.

 

37

    """

 

38

 

 

39

    if len(text) > MAX_HIGHLIGHT_SIZE:

 

40

        return map(cgi.escape,  bzrlib.osutils.split_lines(text))

 

41

 

 

42

    formatter = HtmlFormatter(style=style, nowrap=True, classprefix='pyg-')

 

43

 

 

44

    try:

 

45

        lexer = guess_lexer_for_filename(path, text[:1024], encoding=encoding)

 

46

    except (ClassNotFound, ValueError):

 

47

        try:

 

48

            lexer = guess_lexer(text[:1024], encoding=encoding)

 

49

        except (ClassNotFound, ValueError):

 

50

            lexer = TextLexer(encoding=encoding)

 

51

 

 

52

    hl_lines = _highlight_func(text, lexer, formatter)

 

53

    hl_lines = bzrlib.osutils.split_lines(hl_lines)

 

54

 

 

55

    return hl_lines

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1