view piecrust/dataproviders/base.py @ 911:f2b75e4be981

internal: Use pickle for caching things on disk. This is just easier and lets us use proper classes instead of converting to/from dictionaries.
author Ludovic Chabant <ludovic@chabant.com>
date Sun, 23 Jul 2017 18:03:21 -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)