41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
import os
|
|
|
|
from pelican import signals
|
|
from pelican import contents
|
|
|
|
from jinja2 import Environment, ChoiceLoader, FileSystemLoader
|
|
|
|
|
|
def execjinja2(instance):
|
|
if type(instance) in (contents.Article, contents.Page):
|
|
base_path = os.path.dirname(os.path.abspath(__file__))
|
|
jinja2_env = Environment(
|
|
trim_blocks=True,
|
|
lstrip_blocks=True,
|
|
loader=ChoiceLoader([
|
|
FileSystemLoader(
|
|
os.path.join(base_path, instance.settings['THEME'], 'templates')
|
|
),
|
|
FileSystemLoader(
|
|
os.path.join(base_path, instance.settings['PATH'])
|
|
)
|
|
]),
|
|
extensions=instance.settings['JINJA_EXTENSIONS'],
|
|
)
|
|
|
|
if 'JINJA_FILTERS' in instance.settings:
|
|
jinja2_env.filters.update(instance.settings['JINJA_FILTERS'])
|
|
|
|
jinja2_template = jinja2_env.from_string(instance._content)
|
|
|
|
kwargs = instance._context
|
|
if type(instance) is contents.Article:
|
|
kwargs['article'] = instance
|
|
elif type(instance) is contents.Page:
|
|
kwargs['page'] = instance
|
|
|
|
instance._content = jinja2_template.render(**kwargs)
|
|
|
|
|
|
def register():
|
|
signals.content_object_init.connect(execjinja2)
|