Fixed array and relation types to show its child correct type, instead of always string.

sneto · sneto · commit b0b95aa72cc9 · 2018-07-19T09:48:10.000-03:00
diff --git a/openapi_codec/encode.py b/openapi_codec/encode.py
@@ -118,14 +118,21 @@ def _get_field_type(field):
     if field.schema is None:
         return 'string'
 
+    return _get_field_type_name(field.schema.__class__)
+
+
+def _get_field_type_name(type):
+    if type is None:
+        return 'string'
+
     return {
         coreschema.String: 'string',
         coreschema.Integer: 'integer',
         coreschema.Number: 'number',
         coreschema.Boolean: 'boolean',
         coreschema.Array: 'array',
         coreschema.Object: 'object',
-    }.get(field.schema.__class__, 'string')
+    }.get(type, 'string')
 
 
 def _get_parameters(link, encoding):
@@ -162,7 +169,14 @@ def _get_parameters(link, encoding):
                     'type': field_type,
                 }
                 if field_type == 'array':
-                    schema_property['items'] = {'type': 'string'}
+                    type_string = 'string'
+                    if hasattr(field, "schema") and hasattr(field.schema, "items") and field.schema.items is not None:
+                        if isinstance(field.schema.items, type):
+                            model_type = field.schema.items
+                        else:
+                            model_type = field.schema.items.__class__
+                        type_string = _get_field_type_name(model_type)
+                    schema_property['items'] = {'type': type_string}
                 properties[field.name] = schema_property
                 if field.required:
                     required.append(field.name)
