generate_html.py

"""Takes the data and generates html with nice Plotly graphs
Usage:
    generate_html.py data.csv index.html
"""

import sys
import datetime
import pandas as pd
import plotly.express as px

# number of days to look back
DAYS = 180

def fetch_csv(csv_path: str) -> pd.DataFrame:
    """Fetch CSV"""
    return pd.read_csv(csv_path)


def plot_cumulative_state(df: pd.DataFrame, outfile: str):
    """Plot cumulative states"""
    fig = px.line(
        df,
        x="timestamp",
        y="relative duration",
        title="NuGet restore time",
        color='solution',
        facet_col="scenario",
        facet_col_wrap=1,
        markers=True,
        hover_data=[
            "solution",
            "scenario",
            "duration",
            "timestamp",
            "version",
            "base version"])

    fig.update_traces(marker={'size': 4})
    fig.update_layout(title_x=0.5)
    fig.update_yaxes(ticksuffix="%", title='', rangemode="tozero")
    fig.update_xaxes(title='')
    fig.for_each_annotation(lambda a: a.update(text=a.text.split("=")[-1]))

    fig.write_html(outfile, include_plotlyjs='cdn')

if __name__ == "__main__":
    data = fetch_csv(sys.argv[1])
    data["timestamp"] = pd.to_datetime(data["timestamp"])
    data = data.loc[data["scenario"] != 'warmup']

    now = datetime.datetime.now(datetime.timezone.utc)
    cutoff_date = now - datetime.timedelta(days=DAYS)
    data = data.loc[data["timestamp"] > cutoff_date]

    plot_cumulative_state(data, sys.argv[2])