Allow single-field schemas

labthings · Jan 17, 2020 · 8ac445d · 8ac445d
1 parent c4ddb0a
commit 8ac445d
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/labthings/server/spec/__init__.py b/labthings/server/spec/__init__.py
@@ -134,6 +134,8 @@ def convert_schema(schema, spec: APISpec):
         return schema
     elif isinstance(schema, Mapping):
         return map2properties(schema, spec)
+    elif isinstance(schema, Field):
+        return field2property(schema, spec)
     else:
         raise TypeError(
             "Unsupported schema type. Ensure schema is a Schema class, or dictionary of Field objects"
@@ -156,3 +158,12 @@ def map2properties(schema, spec: APISpec):
             d[k] = v
 
     return {"properties": d}
+
+
+def field2property(field, spec: APISpec):
+    marshmallow_plugin = next(
+        plugin for plugin in spec.plugins if isinstance(plugin, MarshmallowPlugin)
+    )
+    converter = marshmallow_plugin.converter
+
+    return converter.field2property(field)