Conditional decorator to escape strings

gistfile1.py

from functools import wraps

from django.utils.safestring import mark_safe


def raw_html(func_or_str):
    '''
    Call me a string or decorator a function with me to output HTML in your template.
    '''
    if callable(func_or_str):
        @wraps(func_or_str)
        def decorated(*args, **kwargs):
            return mark_safe(func_or_str(*args, **kwargs))
        return decorated
    else:
        return mark_safe(func_or_str)