def test_extract_titles(self):
""" Test the ``extract_titles`` utility. """
document_tree = RootTreeNode()
document_tree.new_child(None, TextTreeNode)
a1 = document_tree.new_child('title', generate_title_cls(1))
document_tree.new_child(None, TextTreeNode)
a2 = document_tree.new_child('title', generate_title_cls(1))
document_tree.new_child(None, TextTreeNode)
a3 = document_tree.new_child('title', generate_title_cls(1))
document_tree.new_child(None, TextTreeNode)
a4 = document_tree.new_child('title', generate_title_cls(1))
titles = extract_titles(document_tree)
self.assertEqual([a1, a2, a3, a4], titles)
def test_extract_titles(self):
""" Test the ``extract_titles`` utility. """
document_tree = RootTreeNode()
document_tree.new_child(None, TextTreeNode)
a1 = document_tree.new_child('title', generate_title_cls(1))
document_tree.new_child(None, TextTreeNode)
a2 = document_tree.new_child('title', generate_title_cls(1))
document_tree.new_child(None, TextTreeNode)
a3 = document_tree.new_child('title', generate_title_cls(1))
document_tree.new_child(None, TextTreeNode)
a4 = document_tree.new_child('title', generate_title_cls(1))
titles = extract_titles(document_tree)
self.assertEqual([a1, a2, a3, a4], titles)
def home_page(request,
template_name='home/home.html',
test_input_form=TestSkCodeInputForm,
extra_context=None):
"""
PySkCode tester home page with form for testing the parser.
:param request: The current request.
:param template_name: The template name to be used.
:param test_input_form: The test input form class to be used.
:param extra_context: Any extra template context information.
:return: TemplateResponse
"""
# Default values
output_content_html = ''
output_content_text = ''
summary_content_html = ''
summary_content_text = ''
footnotes_content_html = ''
footnotes_content_text = ''
document_has_errors = False
# Handle the form
if request.method == "POST":
form = test_input_form(request.POST, request.FILES)
if form.is_valid():
# Parse the input text
newline_node_cls = HardNewlineTreeNode if form.cleaned_data['hard_newline'] else NewlineTreeNode
html_error_template = DEFAULT_ERROR_HTML_TEMPLATE if form.cleaned_data['preview_mode'] else SUPPRESS_ERROR_HTML_TEMPLATE
document = parse_skcode(form.cleaned_data['content'],
allow_tagvalue_attr=form.cleaned_data['allow_tagvalue_attr'],
allow_self_closing_tags=form.cleaned_data['allow_self_closing_tags'],
mark_unclosed_tags_as_erroneous=form.cleaned_data['mark_unclosed_tags'],
newline_node_cls=newline_node_cls)
document_has_errors = document.has_errors()
# Handle smileys and cosmetics
if form.cleaned_data['replace_cosmetics']:
setup_cosmetics_replacement(document)
if form.cleaned_data['replace_smileys']:
def _base_url(filename):
return static('images/smileys/' + filename)
setup_smileys_replacement(document, _base_url)
# Handle relative urls
if form.cleaned_data['convert_relative_url_to_absolute']:
current_site = get_current_site(request)
setup_relative_urls_conversion(document, 'http://%s/' % current_site.domain)
# Get requested render mode
rendering_mode = form.cleaned_data['rendering_mode']
# Apply paragraph utilities
if form.cleaned_data['make_paragraphs']:
make_paragraphs(document)
# Apply footnotes utilities
if form.cleaned_data['render_footnotes_html']:
# Extract all footnotes
footnotes = extract_footnotes(document)
# Render all footnotes
if rendering_mode == RENDERING_MODE_HTML:
footnotes_content_html = render_footnotes_html(footnotes,
html_error_template=html_error_template)
elif rendering_mode == RENDERING_MODE_TEXT:
footnotes_content_text = render_footnotes_text(footnotes)
# Apply titles utilities (part 1 of 2)
if form.cleaned_data['make_auto_title_ids']:
make_auto_title_ids(document)
# Apply titles utilities (part 2 of 2)
if form.cleaned_data['extract_titles']:
# Extract all titles
titles = extract_titles(document)
# Turn the titles list into a hierarchy
titles_hierarchy = list(make_titles_hierarchy(titles))
# Render the output
if rendering_mode == RENDERING_MODE_HTML:
summary_content_html = render_titles_hierarchy_html(titles_hierarchy)
elif rendering_mode == RENDERING_MODE_TEXT:
summary_content_text = render_titles_hierarchy_text(titles_hierarchy)
# Render the document
if rendering_mode == RENDERING_MODE_HTML:
output_content_html = render_to_html(document, html_error_template=html_error_template)
elif rendering_mode == RENDERING_MODE_TEXT:
output_content_text = render_to_text(document)
else:
form = test_input_form()
print(output_content_html)
# Render the template
context = {
'form': form,
'document_has_errors': document_has_errors,
'output_content_html': output_content_html,
'output_content_text': output_content_text,
'summary_content_html': summary_content_html,
'summary_content_text': summary_content_text,
'footnotes_content_html': footnotes_content_html,
'footnotes_content_text': footnotes_content_text,
'title': _('Home page'),
}
if extra_context is not None:
context.update(extra_context)
return TemplateResponse(request, template_name, context)
def test_extract_titles_no_titles(self):
""" Test the ``extract_titles`` utility with no title. """
document_tree = RootTreeNode()
titles = extract_titles(document_tree)
self.assertEqual([], titles)
def test_extract_titles_no_titles(self):
""" Test the ``extract_titles`` utility with no title. """
document_tree = RootTreeNode()
titles = extract_titles(document_tree)
self.assertEqual([], titles)