view piecrust/dataproviders/base.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 d231a10d18f9
children
line wrap: on
line source

from piecrust.configuration import ConfigurationError


class DataProvider:
    """ The base class for a data provider.
    """
    PROVIDER_NAME = None

    debug_render_dynamic = []
    debug_render_invoke_dynamic = []

    def __init__(self, source, page):
        self._sources = [source]
        self._page = page
        self._app = source.app

    def _addSource(self, source):
        self._sources.append(source)


def build_data_provider(provider_type, source, page):
    if not provider_type:
        raise Exception("No data provider type specified.")

    for p in page.app.plugin_loader.getDataProviders():
        if p.PROVIDER_NAME == provider_type:
            pclass = p
            break
    else:
        raise ConfigurationError("Unknown data provider type: %s" %
                                 provider_type)

    return pclass(source, page)