refactor: pass event data into plotting functions

Author: oesteban

README.md

@@ -30,3 +30,12 @@ python src/run.py --help
 ```
 
 for a description of all available options.
+
+Running:
+
+```bash
+python src/run.py plot
+```
+
+will generate the performance and version stream plots using the records stored
+in MongoDB.

src/run.py

@@ -28,6 +28,7 @@
 
 import click
 from api import parallel_fetch, ISSUES, DEFAULT_MAX_ERRORS
+from viz import load_event, plot_performance, plot_version_stream
 
 DEFAULT_DAYS_WINDOW = 90
 DEFAULT_CHUNK_DAYS = 1
@@ -149,6 +150,23 @@ def get(event, start_date, end_date, days, chunk_days, jobs, max_errors, cached_
     click.echo(f"{datetime.now(timezone.utc):%Y-%m-%d %H:%M:%S} [Finished]")
 
 
+@cli.command()
+@click.option("-o", "--output-dir", type=click.Path(file_okay=False, dir_okay=True, writable=True), default=".")
+@click.option("--drop-cutoff", default=None, help="Ignore versions older than this")
+def plot(output_dir, drop_cutoff):
+    """Generate plots using records stored in MongoDB."""
+    today = datetime.now().date().strftime("%Y%m%d")
+    out_perf = os.path.join(output_dir, f"{today}_weekly.png")
+    out_ver = os.path.join(output_dir, f"{today}_versionstream.png")
+
+    unique_started = load_event("started")
+    unique_success = load_event("success")
+
+    plot_performance(unique_started, unique_success, drop_cutoff=drop_cutoff, out_file=out_perf)
+    plot_version_stream(unique_started, unique_success, drop_cutoff=drop_cutoff, out_file=out_ver)
+    click.echo(f"Saved plots to {output_dir}")
+
+
 if __name__ == "__main__":
     """ Install entry-point """
     cli()