| Index: cms/sources.py |
| =================================================================== |
| --- a/cms/sources.py |
| +++ b/cms/sources.py |
| @@ -298,8 +298,115 @@ |
| result.append(relpath + filename) |
| elif os.path.isdir(path): |
| do_list(path, relpath + filename + '/') |
| do_list(self.get_path(subdir), '') |
| return result |
| def get_cache_dir(self): |
| return os.path.join(self._dir, 'cache') |
| + |
| + |
| +class MultiSource(Source): |
| + """A source that combines the contents of multiple other sources.""" |
| + |
| + def __init__(self, base_sources): |
| + self._bases = base_sources |
| + |
| + @property |
| + def version(self): |
| + return self._bases[0].version |
| + |
| + def get_cache_dir(self): |
| + return self._bases[0].get_cache_dir() |
| + |
| + def __enter__(self): |
| + for base in self._bases: |
| + base.__enter__() |
| + return self |
| + |
| + def __exit__(self, exc_type, exc_value, tb): |
| + return any(base.__exit__(exc_type, exc_value, tb) |
| + for base in self._bases) |
| + |
| + def close(self): |
| + for base in self._bases: |
| + base.close() |
| + |
| + def has_file(self, filename): |
| + return any(base.has_file(filename) for base in self._bases) |
| + |
| + def read_file(self, filename, binary=False): |
| + for base in self._bases: |
| + if base.has_file(filename): |
| + return base.read_file(filename, binary) |
| + raise KeyError('File not found {}'.format(filename)) |
| + |
| + def list_files(self, subdir): |
| + return {f for base in self._bases for f in base.list_files(subdir)} |
| + |
| + |
| +def _memoize(func): |
| + """Cache results of functions calls.""" |
| + memoized = {} |
| + |
| + def wrapper(*args): |
| + try: |
| + return memoized[args] |
| + except KeyError: |
| + return memoized.setdefault(args, func(*args)) |
| + wrapper.cache_clear = memoized.clear |
| + return wrapper |
| + |
| + |
| +def create_source(path, static=False, revision='default'): |
|
Wladimir Palant
2017/09/29 08:50:44
The default revision should be 'master', not 'defa
Vasily Kuznetsov
2017/09/29 10:49:09
Changed to `None`, following your next comment. Do
|
| + """Create a source from path and optional revision. |
| + |
| + `static` flag indicates that the website source under `path` can be assumed |
| + to not change after the source was created and any changes can be safely |
| + ignored. |
| + - If `static` is `True`, no interim changes are expected and caching will |
| + be used to optimize performance (source directory is also assumed to be a |
| + Mercurial repository, and `revision` parameter can be used to select a |
| + source revision from which the files will be read). |
| + - If `static` is `False`, changes on the filesystem are expected and will |
| + be reflected in the outputs of the calls. This mode can be used for live |
| + preview. |
| + |
| + If `settings.ini` in the source contains `[paths]` section with an |
| + `additional-paths` key that contains the list of additional root folders, |
| + `MultiSource` will be instantiated and its bases will be the original |
| + source plus an additional source for each additional root folder. |
| + `MultiSource` looks up files in its base sources in the order they are |
| + provided, so the files in the additional folders will only be used if the |
| + original source doesn't contain that file. |
| + """ |
| + if static: |
|
Wladimir Palant
2017/09/29 08:50:44
Should this flag really determine which type of so
Vasily Kuznetsov
2017/09/29 10:49:10
Yeah, you're right. `static` parameter should cont
|
| + source = MercurialSource(path, revision) |
| + else: |
| + source = FileSource(path) |
| + |
| + config = source.read_config() |
| + try: |
| + ap = config.get('paths', 'additional-paths').strip() |
| + additional_paths = filter(None, ap.split()) |
| + except ConfigParser.NoSectionError: |
|
Wladimir Palant
2017/09/29 08:50:44
What about NoOptionError?
Vasily Kuznetsov
2017/09/29 10:49:09
Thank you. Done.
|
| + additional_paths = [] |
| + |
| + if additional_paths: |
| + additional_sources = [ |
| + create_source(os.path.join(path, p)) |
|
Sebastian Noack
2017/09/28 20:48:21
This seems to be equivalent to FileSource(os.path.
Wladimir Palant
2017/09/29 08:50:44
Not quite equivalent. The directory we link to mig
Vasily Kuznetsov
2017/09/29 10:49:09
Yes, nested dependencies are the reason why I made
Sebastian Noack
2017/09/29 15:55:07
I agree with Wladimir. His suggestion seems to be
|
| + for p in additional_paths |
| + ] |
| + source = MultiSource([source] + additional_sources) |
| + |
| + if static: |
| + for fname in [ |
| + 'resolve_link', |
| + 'read_config', |
| + 'read_template', |
| + 'read_locale', |
| + 'read_include', |
| + 'exec_file', |
| + ]: |
| + setattr(source, fname, _memoize(getattr(source, fname))) |
| + |
| + return source |