wagtail-birdsong/birdsong/blocks.py

16 lines
554 B
Python

from django.utils.html import mark_safe
from wagtail.core import blocks
class UnwrappedStreamBlock(blocks.StreamBlock):
"""Removes the surrounding divs around streamblocks."""
def render_basic(self, value, context=None):
return mark_safe('\n'.join(
child.render(context=context) for child in value))
class DefaultBlocks(UnwrappedStreamBlock):
rich_text = blocks.RichTextBlock(
template='birdsong/mail/blocks/richtext.html',
features=['h3', 'h4', 'bold', 'italic', 'link', 'ul', 'ol', 'document-link'])