Fix handling of nullable argument

aubergine-developers · Jun 25, 2018 · aca99fe · aca99fe
1 parent 5714abd
commit aca99fe
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/nadia/common.py b/nadia/common.py
@@ -24,9 +24,8 @@ def translate_args(spec, **kwargs):
         :return: A mapping containing keyword arguments used for constructing marshmallow objects.
         :rtype: dict
         """
-
         return {
-            'allow_none': spec.get('nullable', False),
+            'allow_none': kwargs.get('nullable', False),
             'required': kwargs.get('required', False)
         }
 

diff --git a/nadia/object.py b/nadia/object.py
@@ -36,7 +36,10 @@ def construct_attributes_schemas(self, spec):
             attr_type = prop_content['type']
             attr_schemas_builder = self.builder_provider.get_builder(attr_type)
             attr_required = prop_name in spec.get('required', [])
-            attr_schemas[prop_name] = attr_schemas_builder.build_schema(prop_content, required=attr_required)
+            attr_schemas[prop_name] = attr_schemas_builder.build_schema(
+                prop_content,
+                required=attr_required,
+                nullable=prop_content.get('nullable', False))
         return attr_schemas
 
     @staticmethod