write single json instead of python files

Author: bmaranville

codegen/__init__.py

@@ -22,6 +22,8 @@
 )
 from codegen.validators import (
     write_validator_py,
+    get_validator_params,
+    write_validator_json,
     write_data_validator_py,
     get_data_validator_instance,
 )
@@ -171,20 +173,26 @@ def perform_codegen(reformat=True):
         if node.is_compound and not isinstance(node, ElementDefaultsNode)
     ]
 
+    validator_params = {}
     # Write out validators
     # --------------------
     # # ### Layout ###
     for node in all_layout_nodes:
-        write_validator_py(outdir, node)
+        # write_validator_py(outdir, node)
+        get_validator_params(node, validator_params)
 
     # ### Trace ###
     for node in all_trace_nodes:
-        write_validator_py(outdir, node)
+        # write_validator_py(outdir, node)
+        get_validator_params(node, validator_params)
 
     # ### Frames ###
     for node in all_frame_nodes:
-        write_validator_py(outdir, node)
-
+        # write_validator_py(outdir, node)
+        get_validator_params(node, validator_params)
+        
+    os.makedirs(validators_pkgdir, exist_ok=True)
+    write_validator_json(outdir, validator_params)
     # ### Data (traces) validator ###
     write_data_validator_py(outdir, base_traces_node)
 
@@ -235,8 +243,8 @@ def perform_codegen(reformat=True):
 
     # Output validator __init__.py files
     validators_pkg = opath.join(outdir, "validators")
-    for path_parts, rel_classes in validator_rel_class_imports.items():
-        write_init_py(validators_pkg, path_parts, [], rel_classes)
+    # for path_parts, rel_classes in validator_rel_class_imports.items():
+        # write_init_py(validators_pkg, path_parts, [], rel_classes)
 
     # Write datatype __init__.py files
     # --------------------------------

codegen/validators.py

@@ -1,5 +1,6 @@
 import os.path as opath
 from io import StringIO
+import json
 
 import _plotly_utils.basevalidators
 from codegen.utils import CAVEAT, PlotlyNode, TraceNode, write_source_py
@@ -114,6 +115,58 @@ def write_validator_py(outdir, node: PlotlyNode):
 
     write_source_py(validator_source, filepath, leading_newlines=2)
 
+def get_validator_params(node: PlotlyNode, store: dict):
+    """
+    Write out the JSON schema for the validator
+
+    Parameters
+    ----------
+    node : PlotlyNode
+        The datatype node (node.is_datatype must evaluate to true) for which
+        to build a validator class
+    store : dict
+        Dictionary to store the JSON data for the validator
+    Returns
+    -------
+    None
+    """
+    assert isinstance(store, dict)    
+    assert node.is_datatype
+
+    raw_params = node.get_validator_params()
+    params = dict([(k, eval(v)) for k, v in raw_params.items()])
+    superclass_name = node.name_base_validator.split(".")[-1]
+
+    key = ".".join(node.parent_path_parts + (node.name_property,))
+    store[key] = {"params": params, "superclass": superclass_name}
+
+def write_validator_json(outdir, params: dict):
+    """
+    Write out the JSON schema for the validator
+
+    Parameters
+    ----------
+    outdir : str
+        Root outdir in which the validators package should reside
+    params : dict
+        Dictionary to store the JSON data for the validator
+    Returns
+    -------
+    None
+    """
+    import json
+    # Validate inputs
+    # ---------------
+    if not isinstance(params, dict):
+        raise ValueError("Expected params to be a dictionary")
+
+    # Write file
+    # ----------
+    filepath = opath.join(outdir, "validators", "_validators.json")
+    with open(filepath, "w") as f:
+        f.write(json.dumps(params, indent=4))
+        # f.write(str(params))
+
 
 def build_data_validator_params(base_trace_node: TraceNode):
     """