Merge pull request #40 from ref-humbold/master

Handle classes for subwidgets in MultiWidget

Author: kalwalkden

materializecssform/templatetags/materializecss.py

@@ -8,6 +8,7 @@
 
 register = template.Library()
 
+
 @register.filter
 def materializecss(element, options={}):
     # Set default values if none of them are set
@@ -38,16 +39,22 @@ def materializecss(element, options={}):
     return render(element, markup_classes)
 
 
+def _add_input_classes_widget(widget, field_errors):
+    if _is_multi_widget(widget):
+        for subwidget in widget.widgets:
+            _add_input_classes_widget(subwidget, field_errors)
+    elif not _is_checkbox_widget(widget) and not _is_multiple_checkbox_widget(widget) \
+            and not _is_radio_widget(widget) and not _is_file_widget(widget):
+        classes = widget.attrs.get('class', '')
+        if config.MATERIALIZECSS_VALIDATION:
+            classes += ' validate'
+        if field_errors:
+            classes += ' invalid'
+        widget.attrs['class'] = classes
+
 
 def add_input_classes(field):
-    if not is_checkbox(field) and not is_multiple_checkbox(field) and not is_radio(field) \
-        and not is_file(field):
-        field_classes = field.field.widget.attrs.get('class', '')
-        if config.MATERIALIZECSS_VALIDATION:
-            field_classes += ' validate'
-        if field.errors:
-            field_classes+= ' invalid'
-        field.field.widget.attrs['class'] = field_classes
+    _add_input_classes_widget(field.field.widget, field.errors)
 
 
 def render(element, markup_classes):
@@ -79,10 +86,31 @@ def render(element, markup_classes):
     return template.render(context)
 
 
+def _is_checkbox_widget(widget):
+    return isinstance(widget, forms.CheckboxInput)
+
+
+def _is_multiple_checkbox_widget(widget):
+    return isinstance(widget, forms.CheckboxSelectMultiple)
+
+
+def _is_radio_widget(widget):
+    return isinstance(widget, forms.RadioSelect)
+
+
+def _is_file_widget(widget):
+    return isinstance(widget, forms.FileInput)
+
+
+def _is_multi_widget(widget):
+    return isinstance(widget, forms.MultiWidget)
+
+
 @register.filter
 def is_checkbox(field):
     return isinstance(field.field.widget, forms.CheckboxInput)
 
+
 @register.filter
 def is_textarea(field):
     return isinstance(field.field.widget, forms.Textarea)
@@ -97,10 +125,12 @@ def is_multiple_checkbox(field):
 def is_radio(field):
     return isinstance(field.field.widget, forms.RadioSelect)
 
+
 @register.filter
 def is_date_input(field):
     return isinstance(field.field, DateField)
 
+
 @register.filter
 def is_datetime_input(field):
     return isinstance(field.field, DateTimeField)