Support oneOf
@@ -286,7 +286,9 @@ def get_json_schema_object_fields(obj, enforce_title=False):
def process_data_type(prop, required=False, enforce_title=True):
prop = inherit_parents(prop)
- prop_type = prop['type']
+ prop_type = prop.get('oneOf', prop.get('type', []))
+ assert prop_type
+
tables = []
enum_desc = None
is_object = False
@@ -317,6 +319,15 @@ def process_data_type(prop, required=False, enforce_title=True):
tables = nested["tables"]
enum_desc = nested["enum_desc"]
+ elif isinstance(prop_type, list):
+ prop_title = []
+ for t in prop_type:
+ if isinstance(t, dict):
+ nested = process_data_type(t)
+ tables.extend(nested['tables'])
+ prop_title.append(nested['title'])
+ else:
+ prop_title.append(t)
else:
prop_title = prop_type
