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