summaryrefslogtreecommitdiff
path: root/display.py
blob: cfd7041a0d584186b7b332f8556a0ac5e4bdfb1f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#!/usr/bin/python3
# displays stats as graph. assumes csv is in chronological order, with no
# overlapping sessions. as initial test: make a bar chart for time spent per
# tag, for every day
import pygal
import csv
import argparse
from collections import OrderedDict
import formatting as f
import os

parser = argparse.ArgumentParser()
parser.add_argument("logfile", help="log file to display chart of",
                    type=argparse.FileType('r'))
args = parser.parse_args()

day_times = f.per_tag_per_day(args.logfile)

cache_dir = os.environ["HOME"] + "/.cache/chronologgy"
try:
    os.makedirs(cache_dir)
except FileExistsError:
    pass
except PermissionError:
    print("permission denied: unable to create " + config_dir)

for day in day_times:
    chart = pygal.Bar()
    pie = pygal.Pie()
    for key, value in day.items():
        if key == "date":
            continue
        chart.add(key, round(value/60/60, 1))
        pie.add(key, round(value/60/60, 1))
    chart.render_to_file(str(cache_dir + "/" + day["date"]+"-bar.svg"))
    pie.render_to_file(str(cache_dir + "/" + day["date"]+"-pie.svg"))