tools/json_schema_compiler/highlighters/pygments_highlighter.py

   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
   5 import sys
   6 try:
   7   import pygments
   8   from pygments.lexers import CppLexer
   9   from pygments.formatters import HtmlFormatter
  10   PYGMENTS_IMPORTED = True
  11 except 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
  16 class 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())