"""Run the sealed rule on ~/data/stats19/daily.csv and write results.json."""
import datetime as dt
import json
import os

import pandas as pd

import rule

df = pd.read_csv(os.path.expanduser("~/data/stats19/daily.csv"))
df["date"] = [dt.date.fromisoformat(x) for x in df["date"]]
out = rule.analyse(df)
# Printed only; cannot move the verdict.
out["fatal_serious_main"] = rule.analyse(df, "fatal_serious")["main"]
json.dump(out, open(os.path.join(os.path.dirname(os.path.abspath(__file__)), "results.json"), "w"), indent=1, default=float)
print(json.dumps(out, indent=1, default=float))