rapid-django/src/rapid/templatetags/rapid_filters.py

# coding=utf-8
__author__ = 'marcos.medeiros'

from django import template
from django.utils.safestring import mark_safe
from django.template import loader, Context
from rapid import filters

register = template.Library()

_base = 'rapid/filters/'


@register.inclusion_tag(_base+'model_filters.html', takes_context=True)
def model_filters(context, model):
    ff = filters.FilterSet.from_request(model, context.request)
    return {
        'filters': mark_safe(ff.render_filters(context.request)),
        'has_filters': ff.has_filters(),
        'selectors': mark_safe(ff.render_selectors(context.request)),
    }


# noinspection PyUnusedLocal
@register.inclusion_tag(_base+'register.html', takes_context=True)
def register_filters(context):
    return {}


@register.simple_tag(takes_context=False)
def filter_icon(field):
    if filters.FilterSet.can_filter(field):
        c = Context({'f': field})
        t = loader.get_template(_base+'icon.html')
        return mark_safe(t.render(c))
    return ''