def render(self, context):
page = context['self']
fragment = [f for f in page.fragments.all() if f.name == self.fragment_name]
if fragment:
return richtext(fragment[0].fragment)
else:
return u''
def render(self, context):
page = context['self']
fragment = [
f for f in page.fragments.all() if f.name == self.fragment_name
]
if fragment:
return richtext(fragment[0].fragment)
else:
return u''
def get_context(self, context, **kwargs):
header = context.get('header')
if not header:
page = context.get('page')
if page:
return {
'heading': page.seo_title or page.title,
'lead':
richtext(page.lead) if hasattr(page, 'lead') else '',
'noimage': True,
}
return {
'heading': header.get('title'),
'heading_iconsrc': header.get('iconsrc'),
'lead':
richtext(header.get('content')) if header.get('content') else '',
'noimage': not header.get('images'),
}
def richtext_or_stream(value):
""" A safer alternative to the wagtail filter |richtext, which will render a richtext if it got passed one,
or just render a streamfield with its innate function if it is one such. """
if value and isinstance(value, six.string_types):
return richtext(value)
return value
def test_call_with_none(self):
result = richtext(None)
self.assertEqual(result, '<div class="rich-text"></div>')
def test_call_with_text(self):
result = richtext("Hello world!")
self.assertEqual(result, '<div class="rich-text">Hello world!</div>')
self.assertIsInstance(result, SafeString)
def richtext(self):
return richtext(self.body)
def test_call_with_none(self):
result = richtext(None)
self.assertEqual(result, '<div class="rich-text"></div>')
def test_call_with_text(self):
result = richtext("Hello world!")
self.assertEqual(result, '<div class="rich-text">Hello world!</div>')
self.assertIsInstance(result, SafeText)
