Mercurial > piecrust2

annotate piecrust/data/providersdata.py @ 912:cd236a6af9f6

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
data: Add access to route metadata in the templating data. It's available under `route`. Also, rename `slug` to `rel_url`, because it's confusing when `slug` could be the route metadata of the same name as provided by the default page source. The old `slug` is still there for backwards compatibility though.
author Ludovic Chabant <ludovic@chabant.com>
date Fri, 28 Apr 2017 10:05:14 -0700
parents 71c4f43d8fc1
children 08e02c2a2a1a
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
440
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
1 import re
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
2 import collections.abc
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
3
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
4
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
5 re_endpoint_sep = re.compile(r'[\/\.]')
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
6
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
7
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
8 class DataProvidersData(collections.abc.Mapping):
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
9 def __init__(self, page):
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
10 self._page = page
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
11 self._dict = None
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
12
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
13 def __getitem__(self, name):
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
14 self._load()
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
15 return self._dict[name]
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
16
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
17 def __iter__(self):
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
18 self._load()
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
19 return iter(self._dict)
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
20
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
21 def __len__(self):
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
22 self._load()
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
23 return len(self._dict)
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
24
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
25 def _load(self):
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
26 if self._dict is not None:
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
27 return
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
28
32c7c2d219d2 performance: Refactor how data is managed to reduce copying.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
29 self._dict = {}
814
aca04e175488 data: Allow page generators to have an associated data provider
Ben Artin <ben@artins.org>
parents: 440
diff changeset
30 for source in self._page.app.sources + self._page.app.generators:
aca04e175488 data: Allow page generators to have an associated data provider
Ben Artin <ben@artins.org>
parents: 440
diff changeset
31 if source.data_endpoint:
aca04e175488 data: Allow page generators to have an associated data provider
Ben Artin <ben@artins.org>
parents: 440
diff changeset
32 endpoint_bits = re_endpoint_sep.split(source.data_endpoint)
aca04e175488 data: Allow page generators to have an associated data provider
Ben Artin <ben@artins.org>
parents: 440
diff changeset
33 endpoint = self._dict
aca04e175488 data: Allow page generators to have an associated data provider
Ben Artin <ben@artins.org>
parents: 440
diff changeset
34 for e in endpoint_bits[:-1]:
aca04e175488 data: Allow page generators to have an associated data provider
Ben Artin <ben@artins.org>
parents: 440
diff changeset
35 if e not in endpoint:
aca04e175488 data: Allow page generators to have an associated data provider
Ben Artin <ben@artins.org>
parents: 440
diff changeset
36 endpoint[e] = {}
aca04e175488 data: Allow page generators to have an associated data provider
Ben Artin <ben@artins.org>
parents: 440
diff changeset
37 endpoint = endpoint[e]
aca04e175488 data: Allow page generators to have an associated data provider
Ben Artin <ben@artins.org>
parents: 440
diff changeset
38 override = endpoint.get(endpoint_bits[-1])
aca04e175488 data: Allow page generators to have an associated data provider
Ben Artin <ben@artins.org>
parents: 440
diff changeset
39 provider = source.buildDataProvider(self._page, override)
847
71c4f43d8fc1 data: Don't add route functions or data providers that happen to be null.
Ludovic Chabant <ludovic@chabant.com>
parents: 814
diff changeset
40 if provider is not None:
71c4f43d8fc1 data: Don't add route functions or data providers that happen to be null.
Ludovic Chabant <ludovic@chabant.com>
parents: 814
diff changeset
41 endpoint[endpoint_bits[-1]] = provider