~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to loggerhead/highlight.py

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

• reverse the comparison (146.1.2 to 460)
• stop comparing with revision 460

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