Mercurial > piecrust2
view piecrust/data/providersdata.py @ 473:eb3ace870708 2.0.0a13
bake: Fix a bug with copying assets when `pretty_urls` are disabled.
Added unit tests.
author | Ludovic Chabant <ludovic@chabant.com> |
---|---|
date | Tue, 14 Jul 2015 23:59:08 -0700 |
parents | 32c7c2d219d2 |
children | aca04e175488 |
line wrap: on
line source
import re import collections.abc re_endpoint_sep = re.compile(r'[\/\.]') class DataProvidersData(collections.abc.Mapping): def __init__(self, page): self._page = page self._dict = None def __getitem__(self, name): self._load() return self._dict[name] def __iter__(self): self._load() return iter(self._dict) def __len__(self): self._load() return len(self._dict) def _load(self): if self._dict is not None: return self._dict = {} for source in self._page.app.sources: endpoint_bits = re_endpoint_sep.split(source.data_endpoint) endpoint = self._dict for e in endpoint_bits[:-1]: if e not in endpoint: endpoint[e] = {} endpoint = endpoint[e] override = endpoint.get(endpoint_bits[-1]) provider = source.buildDataProvider(self._page, override) endpoint[endpoint_bits[-1]] = provider