Mercurial > piecrust2
view piecrust/processing/pygments_style.py @ 1140:8e1a0b4d08c5 3.2.0
cm: Regenerate the CHANGELOG.
author | Ludovic Chabant <ludovic@chabant.com> |
---|---|
date | Sun, 20 May 2018 22:37:48 -0700 |
parents | ee3fe63cc51d |
children | 727110ea112a |
line wrap: on
line source
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