From 3228e053703df51454c23349082b8f65fffde985 Mon Sep 17 00:00:00 2001 From: Patrick Dengler Date: Sat, 19 Oct 2013 15:39:04 +0200 Subject: [PATCH] Use isinstance for checking widget types --- bootstrapform/templatetags/bootstrap.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/bootstrapform/templatetags/bootstrap.py b/bootstrapform/templatetags/bootstrap.py index 3ab5aac..cc87370 100644 --- a/bootstrapform/templatetags/bootstrap.py +++ b/bootstrapform/templatetags/bootstrap.py @@ -1,3 +1,4 @@ +from django import forms from django.template import Context from django.template.loader import get_template from django import template @@ -82,14 +83,14 @@ def render(element, markup_classes): @register.filter def is_checkbox(field): - return field.field.widget.__class__.__name__.lower() == "checkboxinput" + return isinstance(field.field.widget, forms.CheckboxInput) @register.filter def is_multiple_checkbox(field): - return field.field.widget.__class__.__name__.lower() == "checkboxselectmultiple" + return isinstance(field.field.widget, forms.CheckboxSelectMultiple) @register.filter def is_radio(field): - return field.field.widget.__class__.__name__.lower() == "radioselect" + return isinstance(field.field.widget, forms.RadioSelect)