tcms.core.templatetags.extra_filters

Complexity

Total Complexity

4

Size/Duplication

Total Lines	54
Duplicated Lines	0 %

Importance

Changes

0

3 Functions

Rating	Name	Duplication	Size	Complexity
A	message_icon()	0	13	1
A	is_list()	0	3	1
A	markdown2html()	0	18	2

"""
    Custom template tag filters.
"""

import bleach
import markdown
from bleach_whitelist import markdown_tags, markdown_attrs, print_tags

from django import template
from django.utils.safestring import mark_safe
from django.contrib.messages import constants as messages

register = template.Library()


@register.filter(name='is_list')
def is_list(variable):
    return isinstance(variable, list)


@register.filter(name='markdown2html')
def markdown2html(md_str):
    """
        Returns markdown string as HTML.
    """
    if md_str is None:
        md_str = ''

    rendered_md = markdown.markdown(md_str,
                                    extensions=[
                                        'markdown.extensions.fenced_code',
                                        'markdown.extensions.nl2br',
                                        'markdown.extensions.tables',
                                    ])
    html = bleach.clean(rendered_md,
                        markdown_tags + print_tags,
                        markdown_attrs)
    return mark_safe(html)  # nosec:B308:blacklist


@register.filter(name='message_icon')
def message_icon(msg):
    """
        Returns the string class name of a message icon
        which feeds directly into Patternfly.
    """
    icons = {
        messages.ERROR: 'error-circle-o',
        messages.WARNING: 'warning-triangle-o',
        messages.SUCCESS: 'ok',
        messages.INFO: 'info',
    }
    return 'pficon-' + icons[msg.level]