Added RadioField to model_form

Adding {“radio”: True} to the field’s form_args when you create the
form, instead of a SelectField, the field will render a RadioField

Author: Garito

AUTHORS

@@ -27,3 +27,4 @@ that much better:
 * Peter D. Gray
 * Massimo Santini
 * Len Buckens - https://github.com/buckensl
+* Garito - https://github.com/garito

flask_mongoengine/wtf/orm.py

@@ -68,6 +68,8 @@ def convert(self, model, field, field_args):
                 kwargs["coerce"] = self.coerce(ftype)
             if kwargs.pop('multiple', False):
                 return f.SelectMultipleField(**kwargs)
+            if kwargs.pop('radio', False):
+                return f.RadioField(**kwargs)
             return f.SelectField(**kwargs)
 
         ftype = type(field).__name__

tests/test_forms.py

@@ -290,6 +290,23 @@ class DogOwner(db.Document):
             self.assertEqual(len(choices), 2)
             self.assertFalse(choices[0].checked)
             self.assertFalse(choices[1].checked)
+            
+    def test_modelradiofield(self):
+        with self.app.test_request_context('/'):
+            db = self.db
+        
+            choices = (('male', 'Male'), ('female', 'Female'), ('other', 'Other'))
+        
+            class Poll(db.Document):
+                answer = db.StringField(choices=choices)
+    
+            PollForm = model_form(Poll, field_args={'answer': {'radio': True}})
+        
+            form = PollForm(answer=None)
+            self.assertTrue(form.validate())
+        
+            self.assertEqual(form.answer.type, 'RadioField')
+            self.assertEqual(form.answer.choices, choices)
 
     def test_passwordfield(self):
         with self.app.test_request_context('/'):