imankulov/coverage-plot

import os

import click

from coverage_plot import import_json, import_xml, plot_sunburst, plot_treemap
from coverage_plot.importance_filesize import FileSizeImportance
from coverage_plot.importance_recency import GitImportance


@click.command()
@click.option(
    "--plot-type",
    default="treemap",
    type=click.Choice(["sunburst", "treemap"]),
    help="Set the plot type",
)
@click.option(
    "--importance-type",
    default="size",
    type=click.Choice(["size", "recency"]),
    help="Set the importance type",
)
@click.option("--show/--no-show", default=True, help="Show the plot in a browser")
@click.option(
    "--save",
    type=click.Path(file_okay=True, dir_okay=False, writable=True, allow_dash=False),
    default=None,
    help="Save the plot in the HTML file",
)
@click.argument("coverage_file", type=click.File(mode="rt"))
def coverage_plot(plot_type, importance_type, show, save, coverage_file):
    """
    Display a summary coverage plot from the coverage.json file.
    """
    importers = {".json": import_json, ".xml": import_xml}
    plotters = {"sunburst": plot_sunburst, "treemap": plot_treemap}

    _, ext = os.path.splitext(coverage_file.name)
    report = importers[ext](coverage_file.read())

    importances = {
        "size": lambda: FileSizeImportance(report),
        "recency": lambda: GitImportance(os.path.dirname(coverage_file.name)),
    }
    importance = importances[importance_type]()
    fig = plotters[plot_type](report, importance)
    if show:
        fig.show()
    if save:
        fig.write_html(save)