37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
from lektor.pluginsystem import Plugin
|
|
from typing import Generator
|
|
import re
|
|
from lektor_groupby import GroupBy, GroupByCallbackArgs
|
|
|
|
|
|
class AdvancedGroupByPlugin(Plugin):
|
|
def on_groupby_before_build_all(self, groupby: GroupBy, builder, **extra):
|
|
# load config
|
|
config = self.get_config()
|
|
regex = config.get('testC.pattern.match')
|
|
try:
|
|
regex = re.compile(regex)
|
|
except Exception as e:
|
|
print('inlinetags.regex not valid: ' + str(e))
|
|
return
|
|
|
|
# load config directly (which also tracks dependency)
|
|
watcher = groupby.add_watcher('testC', config)
|
|
|
|
@watcher.grouping()
|
|
def _replace(args: GroupByCallbackArgs) -> Generator[str, str, None]:
|
|
# args.field assumed to be Markdown
|
|
obj = args.field.source
|
|
slugify_map = {} # type Dict[str, str]
|
|
for match in regex.finditer(obj):
|
|
tag = match.group(1)
|
|
key = yield tag
|
|
print('[advanced] slugify:', tag, '->', key)
|
|
slugify_map[tag] = key
|
|
|
|
def _fn(match: re.Match) -> str:
|
|
tag = match.group(1)
|
|
return f'<a href="/advanced/{slugify_map[tag]}/">{tag}</a>'
|
|
args.field.source = regex.sub(_fn, obj)
|