Function perform_codegen

codegen/__init__.py:111–330

Generate code (and possibly reformat).

(outdir, noformat=False)

def perform_codegen(outdir, noformat=False):
    """Generate code (and possibly reformat)."""

    # Get paths
    validators_dir, graph_objs_dir, graph_objects_path = make_paths(outdir)

    # Delete prior codegen output
    if opath.exists(validators_dir):
        shutil.rmtree(validators_dir)
    if opath.exists(graph_objs_dir):
        shutil.rmtree(graph_objs_dir)

    # Load plotly schema
    project_root = opath.dirname(outdir)
    plot_schema_path = opath.join(
        project_root, "codegen", "resources", "plot-schema.json"
    )

    with open(plot_schema_path, "r") as f:
        plotly_schema = json.load(f)

    # Preprocess Schema
    preprocess_schema(plotly_schema)

    # Build node lists

    # TraceNode
    base_traces_node = TraceNode(plotly_schema)
    all_trace_nodes = PlotlyNode.get_all_datatype_nodes(plotly_schema, TraceNode)

    # LayoutNode
    compound_layout_nodes = PlotlyNode.get_all_compound_datatype_nodes(
        plotly_schema, LayoutNode
    )
    layout_node = compound_layout_nodes[0]
    all_layout_nodes = PlotlyNode.get_all_datatype_nodes(plotly_schema, LayoutNode)

    subplot_nodes = [
        node
        for node in layout_node.child_compound_datatypes
        if node.node_data.get("_isSubplotObj", False)
    ]

    layout_array_nodes = [
        node
        for node in layout_node.child_compound_datatypes
        if node.is_array_element and node.has_child("xref") and node.has_child("yref")
    ]

    # FrameNode
    compound_frame_nodes = PlotlyNode.get_all_compound_datatype_nodes(
        plotly_schema, FrameNode
    )
    frame_node = compound_frame_nodes[0]
    all_frame_nodes = PlotlyNode.get_all_datatype_nodes(plotly_schema, FrameNode)

    # All nodes
    all_datatype_nodes = all_trace_nodes + all_layout_nodes + all_frame_nodes