__init__.py

import json
import os.path as opath
import shutil

from codegen.graph_objs import (delete_graph_objs_non_generated_lines,
                                append_graph_objs_generated_lines)


def perform_codegen():
    outdir = 'plotly/'

    # Delete non-generated lines from graph_objs.py
    # ---------------------------------------------
    delete_graph_objs_non_generated_lines()

    # Perform codegen imports
    # -----------------------
    # this must happend after deleting non-generated lines above
    from codegen.datatypes import (build_datatypes_py, write_datatypes_py,
                                   append_figure_class)
    from codegen.utils import TraceNode, PlotlyNode, LayoutNode, FrameNode
    from codegen.validators import (write_validator_py,
                                    append_traces_validator_py)

    # Load plotly schema
    # ------------------
    with open('plotly/package_data/default-schema.json', 'r') as f:
        plotly_schema = json.load(f)

    # Compute property paths
    # ----------------------
    base_traces_node = TraceNode(plotly_schema)
    compound_trace_nodes = PlotlyNode.get_all_compound_datatype_nodes(plotly_schema, TraceNode)
    compound_layout_nodes = PlotlyNode.get_all_compound_datatype_nodes(plotly_schema, LayoutNode)
    compound_frame_nodes = PlotlyNode.get_all_compound_datatype_nodes(plotly_schema, FrameNode)

    extra_layout_nodes = PlotlyNode.get_all_trace_layout_nodes(plotly_schema)

    # Write out validators
    # --------------------
    validators_pkgdir = opath.join(outdir, 'validators')
    if opath.exists(validators_pkgdir):
        shutil.rmtree(validators_pkgdir)

    # ### Layout ###
    for node in compound_layout_nodes:
        write_validator_py(outdir, node, extra_layout_nodes)

    # ### Trace ###
    for node in compound_trace_nodes:
        write_validator_py(outdir, node)

    # Write out datatypes
    # -------------------
    datatypes_pkgdir = opath.join(outdir, 'datatypes')
    if opath.exists(datatypes_pkgdir):
        shutil.rmtree(datatypes_pkgdir)

    # ### Layout ###
    for node in compound_layout_nodes:
        write_datatypes_py(outdir, node, extra_layout_nodes)

    # ### Trace ###
    for node in compound_trace_nodes:
        write_datatypes_py(outdir, node)

    # Append traces validator class
    # -----------------------------
    append_traces_validator_py(validators_pkgdir, base_traces_node)

    # Add Frames
    # ----------
    # ### Validator ###
    for node in compound_frame_nodes:
        write_validator_py(outdir, node)

    # ### Datatypes ###
    for node in compound_frame_nodes:
        write_datatypes_py(outdir, node)

    # Append figure class to datatypes
    # --------------------------------
    append_figure_class(datatypes_pkgdir, base_traces_node)

    # Append generated lines to graph_objs.py
    # ------------------------------------
    append_graph_objs_generated_lines()


if __name__ == '__main__':
    perform_codegen()