| Left: | ||
| Right: |
| OLD | NEW |
|---|---|
| (Empty) | |
| 1 import re | |
| 2 from jinja2 import contextfunction | |
| 3 | |
| 4 | |
| 5 @contextfunction | |
| 6 def get_pages_metadata(context, filters=None): | |
| 7 if not isinstance(filters, dict) and filters: | |
| 8 raise TypeError('Filters are not a dictionary') | |
| 9 | |
| 10 return_data = [] | |
| 11 for page_name, _format in context['source'].list_pages(): | |
| 12 data, filename = context['source'].read_page(page_name, _format) | |
| 13 page_data = parse_page_metadata(data, page_name) | |
| 14 | |
| 15 if filter_metadata(filters, page_data) is True: | |
| 16 return_data.append(page_data) | |
| 17 | |
| 18 return return_data | |
| 19 | |
| 20 | |
| 21 def parse_page_metadata(data, page): | |
| 22 page_metadata = {'page': page} | |
| 23 lines = data.splitlines(True) | |
| 24 for i, line in enumerate(lines): | |
| 25 if not re.search(r'^\s*[\w\-]+\s*=', line): | |
| 26 break | |
| 27 name, value = line.split('=', 1) | |
| 28 value = value.strip(' ').strip('\n') | |
|
Vasily Kuznetsov
2017/03/08 17:07:41
Can we just strip everything? Or is there a specia
Jon Sonesen
2017/03/08 17:47:17
Nah, we should just strip everything, will do
| |
| 29 if value.startswith('['): | |
|
Vasily Kuznetsov
2017/03/08 17:07:41
Maybe add "and value.endswith(']')" here to be a b
Jon Sonesen
2017/03/08 17:47:16
Agreed.
| |
| 30 value = value[1:-1].split(',') | |
|
Vasily Kuznetsov
2017/03/08 17:07:41
We should probably filter the split value through
Jon Sonesen
2017/03/08 17:47:16
no problem :)
| |
| 31 page_metadata[name.strip()] = value | |
| 32 return page_metadata | |
| 33 | |
| 34 | |
| 35 def filter_metadata(filters, metadata): | |
| 36 if filters is None: | |
| 37 return True | |
| 38 for filter_name, filter_value in filters.items(): | |
| 39 if filter_name not in metadata: | |
| 40 return False | |
| 41 for option in filter_value.split(','): | |
| 42 if option not in metadata[filter_name]: | |
| 43 return False | |
| 44 return True | |
| OLD | NEW |