diff piecrust/processing/pygments_style.py @ 507:ee3fe63cc51d

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
children 727110ea112a
line wrap: on
line diff
--- /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
+