site nav

aircox_web/renderer.py

@@ -1,16 +1,31 @@
 from django.utils.html import format_html, mark_safe
 from feincms3.renderer import TemplatePluginRenderer
 
-from .models import Page, RichText, Image
+from .models import *
 
 
-renderer = TemplatePluginRenderer()
-renderer.register_string_renderer(
-    RichText,
+site_renderer = TemplatePluginRenderer()
+site_renderer.register_string_renderer(
+    SiteRichText,
     lambda plugin: mark_safe(plugin.text),
 )
-renderer.register_string_renderer(
-    Image,
+site_renderer.register_string_renderer(
+    SiteImage,
+    lambda plugin: format_html(
+        '<figure><img src="{}" alt=""/><figcaption>{}</figcaption></figure>',
+        plugin.image.url,
+        plugin.caption,
+    ),
+)
+
+
+page_renderer = TemplatePluginRenderer()
+page_renderer.register_string_renderer(
+    PageRichText,
+    lambda plugin: mark_safe(plugin.text),
+)
+page_renderer.register_string_renderer(
+    PageImage,
     lambda plugin: format_html(
         '<figure><img src="{}" alt=""/><figcaption>{}</figcaption></figure>',
         plugin.image.url,