Changeset 348

Timestamp:

06/21/08 22:18:21 (4 years ago)

Author:

simon

Message:

Add pygments syntax highlighters.

Files:

• hodgestar/Talks/PythonObjects/slides.py (modified) (2 diffs)

hodgestar/Talks/PythonObjects/slides.py

@@ -21 +21 @@
 #   Version 1.0.2
 #   Minor changes to move prev/next links to bottom. Clean-up tabs.
+#   Convert to new-style clases.
+#   Add syntax highlighters based on the pygments module.
 # 
 # 2002, Moshe Zadka and Itamar Shtull-Traurin
@@ -249 +251 @@
         return text.toHTML()
 
+#
+# Syntax highlighting for code using pygments.
+#
+
+class CODE(object):
+    """Syntax highlight code using pygments.
+
+       Imports pygments inside methods so that slides
+       can function without pygments.
+       """
+
+    def __init__(self, code, lexer="python", formatter="html", style="default"):
+        self.code = self._strip_indent(code)
+        self.lexer = self._lexer(lexer)
+        self.formatter = self._formatter(formatter, style)
+
+    def toHTML(self):
+        import pygments
+        return pygments.highlight(self.code, self.lexer, self.formatter)
+
+    def _strip_indent(self, code):
+        """Remove lead and trailing blank lines and strip indents."""
+        itr = iter(code.split("\n"))
+        for line in itr:
+            if line.strip():
+                break
+        indent = len(line) - len(line.lstrip())
+        lines = [line[indent:]]
+        for line in itr:
+            lines.append(line[indent:])
+        while lines and not lines[-1]:
+            del lines[-1]
+        return "\n".join(lines)
+
+    def _lexer(self, lexer_name):
+        """Fetch a pygments lexer by name."""
+        from pygments.lexers import get_lexer_by_name
+        return get_lexer_by_name(lexer_name)
+
+    def _formatter(self, formatter_name, style_name):
+        """Fetch a pygments formatter using formatter and style names."""
+        from pygments.formatters import get_formatter_by_name
+        return get_formatter_by_name(formatter_name, style=style_name)
+
+class PYCODE(CODE):
+    """Syntax hightlight Python code."""
+    def __init__(self, code, formatter="html", style="default"):
+        super(PYCODE, self).__init__(code, "python", formatter, style)
+
+class PYCON(CODE):
+    """Syntax highlight Python console sessions."""
+    def __init__(self, code, formatter="html", style="default"):
+        super(PYCON, self).__init__(code, "pycon", formatter, style)
 
 __all__ = ["Lecture", "Slide", "NumSlide", "PRE", "URL", "Bullet",
            "SubBullet", "toHTML", "Center", "Heading", "Image", "TitleSlide"]
+
+try:
+    import pygments
+    __all__.extend(["CODE", "PYCODE", "PYCON"])
+except ImportError:
+    pass