archive/pillar
0
0
Fork 0
Code Issues 42 Pull Requests Projects Releases Wiki Activity

{iframe} shortcode no longer requires cap=xxx

Browse Source
This commit is contained in:
Sybren A. Stüvel 2018-04-04 15:44:52 +02:00
parent 289dc39e50
commit c9789f46db
2 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pillar/shortcodes.py
View File

@ -142,16 +142,14 @@ def iframe(context: typing.Any,
import xml.etree.ElementTree as ET
from pillar.auth import current_user
cap = kwargs.pop('cap', None)
if not cap:
return html_module.escape('{iframe missing cap="somecap"}')
nocap = kwargs.pop('nocap', '')
if not current_user.has_cap(cap):
if not nocap:
return ''
html = html_module.escape(nocap)
return f'<p class="shortcode nocap">{html}</p>'
cap = kwargs.pop('cap', '')
if cap:
nocap = kwargs.pop('nocap', '')
if not current_user.has_cap(cap):
if not nocap:
return ''
html = html_module.escape(nocap)
return f'<p class="shortcode nocap">{html}</p>'
kwargs['class'] = f'shortcode {kwargs.get("class", "")}'.strip()
element = ET.Element('iframe', kwargs)

4
tests/test_shortcodes.py
View File

@ -109,7 +109,9 @@ class IFrameTest(AbstractPillarTest):
def test_missing_cap(self):
from pillar.shortcodes import render
self.assertEqual('{iframe missing cap=&quot;somecap&quot;}', render('{iframe}'))
md = '{iframe src="https://docs.python.org/3/library/"}'
expect = '<iframe class="shortcode" src="https://docs.python.org/3/library/"></iframe>'
self.assertEqual(expect, render(md))
def test_user_no_cap(self):
from pillar.shortcodes import render