Changeset 1436

Timestamp:

Oct 22, 2006, 10:18:02 PM (17 years ago)

Author:

Matt Good

Message:

TracPygmentsPlugin:

use the Trac CSS classes for code coloring

File:

• tracpygmentsplugin/0.11/tracpygments/__init__.py (modified) (2 diffs)

tracpygmentsplugin/0.11/tracpygments/__init__.py

@@ -108 +108 @@
     import pygments
     lexer = pygments.get_lexer_by_name(lang)
-    formatter = pygments.get_formatter_by_name('html', noclasses=True,
-                    cssclass = not annotate and 'code' or '')
+    formatter = TracHtmlFormatter(cssclass = not annotate and 'code' or '')
     html = pygments.highlight(content, lexer, formatter).rstrip('\n')
     if annotate:
@@ -138 +137 @@
     def render_macro(self, req, name, content):
         return _format(name, content)
+
+try:
+    import pygments
+except ImportError:
+    pass
+else:
+    from pygments.formatters.html import HtmlFormatter
+    from pygments.token import *
+
+    def _issubtoken(token, base):
+        while token is not None:
+            if token == base:
+                return True
+            token = token.parent
+        return False
+
+    class TracHtmlFormatter(HtmlFormatter):
+        token_classes = [
+            (Comment, 'code-comment'),
+            (Name.Function, 'code-func'),
+            (Name.Variable, 'code-var'),
+            (Name, 'code-lang'),
+            (String, 'code-string'),
+            (Keyword.Type, 'code-type'),
+            (Keyword, 'code-keyword'),
+            (Token, None),
+        ]
+
+        def _get_css_class(self, ttype):
+            try:
+                return self._class_cache[ttype]
+            except KeyError:
+                pass
+            for token, css_class in self.token_classes:
+                if _issubtoken(ttype, token):
+                    break
+            else:
+                css_class = None
+            if css_class is not None:
+                css_class = self.classprefix + css_class
+            self._class_cache[ttype] = css_class
+            return css_class