Account for allow_none.

When allow_none is set to True in the marshmallow schema, we have to add 'null' to the list of possible types. Other wise you get a None is not a Number error.
fuhrysteve · Sep 10, 2016 · ee17c8d · ee17c8d
1 parent 6f51771
commit ee17c8d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/marshmallow_jsonschema/base.py b/marshmallow_jsonschema/base.py
@@ -107,7 +107,9 @@ def get_properties(self, obj):
                     schema = FIELD_VALIDATORS[validator.__class__](
                         schema, field, validator, obj
                     )
-
+            if field.allow_none:
+                schema['type'] = [schema['type'], 'null']
+
             properties[field.name] = schema
 
         return properties