Inline primitive models in preprocessing

mbohlool · mbohlool · commit fbdff46af084 · 2017-04-17T16:35:57.000-07:00
diff --git a/scripts/preprocess_spec.py b/scripts/preprocess_spec.py
@@ -117,6 +117,8 @@ def process_swagger(spec):
 
     remove_model_prefixes(spec)
 
+    inline_primitive_models(spec)
+
     # TODO: Kubernetes does not set a version for OpenAPI spec yet,
     # remove this when that is fixed.
     spec['info']['version'] = SPEC_VERSION
@@ -191,6 +193,40 @@ def remove_model_prefixes(spec):
         rename_model(spec, k, v["new_name"])
 
 
+def find_replace_ref_recursive(root, ref_name, replace_map):
+    if isinstance(root, list):
+        for r in root:
+            find_replace_ref_recursive(r, ref_name, replace_map)
+    if isinstance(root, dict):
+        if "$ref" in root:
+            if root["$ref"] == ref_name:
+                del root["$ref"]
+                for k, v in replace_map.iteritems():
+                    if k in root:
+                        if k != "description":
+                            raise PreprocessingException(
+                                "Cannot inline model %s because of "
+                                "conflicting key %s." % (ref_name, k))
+                        continue
+                    root[k] = v
+        for k, v in root.iteritems():
+            find_replace_ref_recursive(v, ref_name, replace_map)
+
+
+def inline_primitive_models(spec):
+    to_remove_models = []
+    for k, v in spec['definitions'].iteritems():
+        if "properties" not in v:
+            print("Making primitive mode `%s` inline ..." % k)
+            if "type" not in v:
+                v["type"] = "object"
+            find_replace_ref_recursive(spec, "#/definitions/" + k, v)
+            to_remove_models.append(k)
+
+    for k in to_remove_models:
+        del spec['definitions'][k]
+
+
 def main():
     pool = urllib3.PoolManager()
     with pool.request('GET', SPEC_URL, preload_content=False) as response:
