steffim/plot.py at master · schicho/steffim · GitHub

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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
import json
import os
from datetime import datetime

import matplotlib.pyplot as plt
from matplotlib.ticker import MultipleLocator

plt.rcParams["svg.fonttype"] = "none"
plt.rcParams["pdf.use14corefonts"] = True

STEFFIM_PLOT_STYLE = [
    "seaborn-v0_8-darkgrid",
    {
        "font.family": "monospace",
        "font.size": 10,
        "axes.prop_cycle": plt.cycler("color", ["#ff8800"]),
    },
]


def plot_by_chair():
    with plt.style.context(STEFFIM_PLOT_STYLE):
        _plot_by_chair()


def _plot_by_chair():
    fig, ax = plt.subplots(figsize=(10, 10), constrained_layout=True)
    fig.set_figwidth(8)
    fig.set_figheight(5)

    plt.setp(ax.get_xticklabels(), rotation=45, ha="right")

    ax.set_title("#Stef by Chair")

    # force integer y-axis
    ax.yaxis.get_major_locator().set_params(integer=True)
    ax.xaxis.get_major_locator().set_params(integer=True)

    # get latest historic file
    historic_files = [f for f in os.listdir("historic") if f.endswith(".json")]
    historic_files.sort()
    with open(f"historic/{historic_files[-1]}", "r") as f:
        chair_json = json.load(f)

    # leave out empty chairs
    chair_data = [chair for chair in chair_json["data"] if len(chair["stef_list"]) > 0]

    # list of tuples (chair_name, stef_count)
    chair_and_stef_count = [
        (chair["chair_name"], len(chair["stef_list"])) for chair in chair_data
    ]

    # sort by stef count and then by name
    chair_names, stef_counts = zip(
        *sorted(chair_and_stef_count, key=lambda x: (x[1], x[0]))
    )

    rects = ax.bar(chair_names, stef_counts)
    ax.bar_label(
        rects, [f"{count}" for count in stef_counts], padding=-32, fontweight="bold"
    )

    fig.savefig("generated/stef_by_chair.svg")


def plot_over_time():
    with plt.style.context(STEFFIM_PLOT_STYLE):
        _plot_over_time()


def _plot_over_time():
    fig, ax = plt.subplots(figsize=(10, 10), constrained_layout=True)
    fig.set_figwidth(8)
    fig.set_figheight(5)

    plt.xticks(rotation=90)

    ax.set_title("#Stef Over Time")

    # limit x ticks, may need to be increased as time goes on and the crawler still works
    ax.xaxis.set_major_locator(MultipleLocator(2))

    # force integer axis
    ax.yaxis.get_major_locator().set_params(integer=True)

    # get list of historic files
    historic_files = [f for f in os.listdir("historic") if f.endswith(".json")]
    historic_files.sort()

    # get data from historic files
    historic_data = []
    for historic_file in historic_files:
        with open(f"historic/{historic_file}", "r") as f:
            historic_data.append(json.load(f))

    xData = []
    yData = []

    for historic in historic_data:
        # convert timestamp to datetime YYYY-MM-DD
        timestamp = datetime.fromtimestamp(float(historic["timestamp"]))

        xData.append(timestamp.strftime("%b '%y"))

        # summarize stef count over all chairs
        stef_count = 0
        for chair in historic["data"]:
            stef_count += len(chair["stef_list"])

        yData.append(stef_count)

    ax.plot(xData, yData)
    fig.savefig("generated/stef_over_time.svg")