Use schemas to dynamically display UI

pydatalab/src/pydatalab/routes/v0_1/info.py

@@ -11,7 +11,8 @@
 from pydatalab import __version__
 from pydatalab.blocks import BLOCK_TYPES
 from pydatalab.config import CONFIG, FEATURE_FLAGS, FeatureFlags
-from pydatalab.models import Person
+from pydatalab.models import Collection, Person
+from pydatalab.models.items import Item
 from pydatalab.mongo import flask_mongo
 
 from ._version import __api_version__
@@ -143,3 +144,50 @@ def list_block_types():
             ).json()
         )
     )
+
+
+def get_all_items_models():
+    return Item.__subclasses__()
+
+
+@INFO.route("/info/types", methods=["GET"])
+def list_supported_types():
+    """Returns a list of supported schemas."""
+    types = [cls.schema()["properties"]["type"]["default"] for cls in get_all_items_models()]
+    types.append(Collection.schema()["properties"]["type"]["default"])
+
+    return jsonify(types)
+
+
+for model_class in get_all_items_models():
+    model_type = model_class.schema()["properties"]["type"]["default"]
+
+    def make_route(model_class, model_type):
+        @INFO.route(
+            f"/info/types/{model_type}", methods=["GET"], endpoint=f"get_{model_type}_schema"
+        )
+        def get_model_schema():
+            """Returns the JSON schema for the model."""
+            schema = model_class.schema()
+
+            response = {
+                "data": {
+                    "schema": schema,
+                }
+            }
+            return jsonify(response)
+
+    make_route(model_class, model_type)
+
+
+@INFO.route("/info/types/collections", methods=["GET"])
+def get_collection_schema():
+    """Returns the JSON schema for the Collection type."""
+    schema = Collection.schema()
+
+    response = {
+        "data": {
+            "schema": schema,
+        }
+    }
+    return jsonify(response)