Mercurial > piecrust2

changeset 507:ee3fe63cc51d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
bake: Add a processor to generate a Pygments style CSS file.
author Ludovic Chabant <ludovic@chabant.com>
date Sat, 25 Jul 2015 22:00:27 -0700
parents 6ca1b6fb9964
children 79f03b5c1b8f
files piecrust/plugins/builtin.py piecrust/processing/pygments_style.py
diffstat 2 files changed, 26 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- a/piecrust/plugins/builtin.py	Sat Jul 25 21:55:26 2015 -0700
+++ b/piecrust/plugins/builtin.py	Sat Jul 25 22:00:27 2015 -0700
@@ -27,6 +27,7 @@
 from piecrust.processing.compressors import (
         CleanCssProcessor, UglifyJSProcessor)
 from piecrust.processing.less import LessProcessor
+from piecrust.processing.pygments_style import PygmentsStyleProcessor
 from piecrust.processing.requirejs import RequireJSProcessor
 from piecrust.processing.sass import SassProcessor
 from piecrust.processing.sitemap import SitemapProcessor
@@ -99,6 +100,7 @@
         return [
                 CopyFileProcessor(),
                 ConcatProcessor(),
+                PygmentsStyleProcessor(),
                 CompassProcessor(),
                 LessProcessor(),
                 SassProcessor(),
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/piecrust/processing/pygments_style.py	Sat Jul 25 22:00:27 2015 -0700
@@ -0,0 +1,24 @@
+import yaml
+from pygments.formatters import HtmlFormatter
+from piecrust.processing.base import SimpleFileProcessor
+
+
+class PygmentsStyleProcessor(SimpleFileProcessor):
+    PROCESSOR_NAME = 'pygments_style'
+
+    def __init__(self):
+        super(PygmentsStyleProcessor, self).__init__({'pygstyle': 'css'})
+
+    def _doProcess(self, in_path, out_path):
+        with open(in_path, 'r') as fp:
+            config = yaml.load(fp)
+
+        style_name = config.get('style', 'default')
+        class_name = config.get('class', '.highlight')
+        fmt = HtmlFormatter(style=style_name).get_style_defs(class_name)
+
+        with open(out_path, 'w') as fp:
+            fp.write(fmt)
+
+        return True
+