Mercurial > piecrust2
view piecrust/processing/pygments_style.py @ 1188:a7c43131d871
bake: Fix file write flushing problem with Python 3.8+
Writing the cache files fails in Python 3.8 because it looks like flushing
behaviour has changed. We need to explicitly flush. And even then, in very
rare occurrences, it looks like it can still run into racing conditions,
so we do a very hacky and ugly "retry" loop when fetching cached data :(
author | Ludovic Chabant <ludovic@chabant.com> |
---|---|
date | Tue, 15 Jun 2021 22:36:23 -0700 |
parents | 727110ea112a |
children |
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.safe_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