diff --git a/homepage/pelicanconf.py b/homepage/pelicanconf.py index 5fd4540..6fa44b5 100644 --- a/homepage/pelicanconf.py +++ b/homepage/pelicanconf.py @@ -39,8 +39,12 @@ DIRECT_TEMPLATES = ['sitemap'] SITEMAP_SAVE_AS = 'sitemap.xml' PLUGIN_PATHS = ['./plugins'] -PLUGINS = ['assets'] +PLUGINS = ['decorate_content', 'assets'] +DECORATE_CONTENT = { + # maps any CSS selector to a list of classes to be added + # 'p': ['pv0', 'dim'] +} GITHUB_ORG = 'https://github.com/offen' CONTACT_EMAIL = 'hioffen@posteo.de' diff --git a/homepage/plugins/decorate_content/__init__.py b/homepage/plugins/decorate_content/__init__.py new file mode 100644 index 0000000..1dd5d68 --- /dev/null +++ b/homepage/plugins/decorate_content/__init__.py @@ -0,0 +1 @@ +from .decorate_content import * diff --git a/homepage/plugins/decorate_content/decorate_content.py b/homepage/plugins/decorate_content/decorate_content.py new file mode 100644 index 0000000..21c0f4e --- /dev/null +++ b/homepage/plugins/decorate_content/decorate_content.py @@ -0,0 +1,33 @@ +from json import loads + +from pelican import signals +from pelican.contents import Article, Page, Static +from bs4 import BeautifulSoup + + +def content_object_init(instance): + if isinstance(instance, (Article, Page, Static)): + plugin_settings = instance.settings.get("DECORATE_CONTENT", {}) + content_overrides = ( + instance.metadata.get("decorate_content", None) + if instance.metadata is not None + else None + ) + + settings = plugin_settings.copy() + settings.update( + loads(content_overrides) if content_overrides is not None else {} + ) + + soup = BeautifulSoup(instance._content, "html.parser") + + for selector, class_names in settings.items(): + elems = soup.find_all(selector) + for elem in elems: + elem["class"] = elem.get("class", []) + class_names + + instance._content = soup.decode() + + +def register(): + signals.content_object_init.connect(content_object_init) diff --git a/homepage/requirements.txt b/homepage/requirements.txt index 401a5fb..5a8b71c 100644 --- a/homepage/requirements.txt +++ b/homepage/requirements.txt @@ -3,3 +3,4 @@ markdown==3.1.1 git+git://github.com/miracle2k/webassets#d1f3455e383446ca4ab0c644f326ee937e68e809 cssmin==0.2.0 libsass==0.19.3 +beautifulsoup4==4.8.1