Set widget types before render the form

moccu · Feb 8, 2017 · 9ffa83e · 9ffa83e
1 parent d0e9dba
commit 9ffa83e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/barbeque/forms/mixins.py b/barbeque/forms/mixins.py
@@ -67,11 +67,14 @@ class FloppyformsLayoutMixin(object):
     row_classname = 'form-row'
     div_template_name = 'barbeque/forms/layout/div.html'
 
-    def __init__(self, *args, **kwargs):
-        super(FloppyformsLayoutMixin, self).__init__(*args, **kwargs)
+    def set_widget_types(self):
         for name, field in self.fields.items():
             widget = self.fields[name].widget
             widget.widget_type = widget.__class__.__name__.lower()
 
+    def _render_as(self, *args, **kwargs):
+        self.set_widget_types()
+        return super()._render_as(*args, **kwargs)
+
     def as_div(self):
         return self._render_as(self.div_template_name)
diff --git a/barbeque/tests/test_forms_mixins.py b/barbeque/tests/test_forms_mixins.py
@@ -124,6 +124,7 @@ class TestFloppyformsLayoutMixin:
 
     def test_widget_type(self):
         form = FloppyformsLayoutForm()
+        form.set_widget_types()
         assert form.fields['name'].widget.widget_type == 'textinput'
 
     @mock.patch('barbeque.tests.test_forms_mixins.FloppyformsLayoutForm._render_as')