1# Copyright (c) 2012 The Chromium Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5import sys
6try:
7  import pygments
8  from pygments.lexers import CppLexer
9  from pygments.formatters import HtmlFormatter
10  PYGMENTS_IMPORTED = True
11except ImportError:
12  print('It appears that Pygments is not installed. '
13    'Can be installed using easy_install Pygments or from http://pygments.org.')
14  PYGMENTS_IMPORTED = False
15
16class PygmentsHighlighter(object):
17  def __init__(self):
18    if not PYGMENTS_IMPORTED:
19      raise ImportError('Pygments not installed')
20
21  """Highlighter that uses the python pygments library to highlight code.
22  """
23  def GetCSS(self, style):
24    formatter = HtmlFormatter(linenos=True,
25        style=pygments.styles.get_style_by_name(style))
26    return formatter.get_style_defs('.highlight')
27
28  def GetCodeElement(self, code, style):
29    formatter = HtmlFormatter(linenos=True,
30            style=pygments.styles.get_style_by_name(style))
31    return pygments.highlight(code, CppLexer(), formatter)
32
33  def DisplayName(self):
34    return 'pygments' + ('' if PYGMENTS_IMPORTED else ' (not installed)')
35
36  def GetStyles(self):
37    return list(pygments.styles.get_all_styles())
38