aboutsummaryrefslogtreecommitdiffstats
path: root/python/fatcat_web/graphics.py
blob: 96c3531a8432fd8d962cac0688136026bc34a8d8 (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
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

from typing import List, Tuple, Dict

import pygal
from pygal.style import CleanStyle

def ia_coverage_histogram(rows: List[Tuple]) -> pygal.Graph:
    """
    Note: this returns a raw pygal chart; it does not render it to SVG/PNG
    """

    raw_years = [int(r[0]) for r in rows]
    years = dict()
    if raw_years:
        for y in range(min(raw_years), max(raw_years)+1):
            years[int(y)] = dict(year=int(y), available=0, missing=0)
        for r in rows:
            if r[1]:
                years[int(r[0])]['available'] = r[2]
            else:
                years[int(r[0])]['missing'] = r[2]

    years = sorted(years.values(), key=lambda x: x['year'])

    CleanStyle.colors = ("green", "purple")
    label_count = len(years)
    if len(years) > 20:
        label_count = 10
    chart = pygal.StackedBar(dynamic_print_values=True, style=CleanStyle,
        width=1000, height=500, x_labels_major_count=label_count,
        show_minor_x_labels=False)
    #chart.title = "Perpetual Access Coverage"
    chart.x_title = "Year"
    #chart.y_title = "Releases"
    chart.x_labels = [str(y['year']) for y in years]
    chart.add('via Fatcat', [y['available'] for y in years])
    chart.add('Missing', [y['missing'] for y in years])
    return chart

def preservation_by_year_histogram(rows: List[Dict]) -> pygal.Graph:
    """
    Note: this returns a raw pygal chart; it does not render it to SVG/PNG
    """

    years = sorted(rows, key=lambda x: x['year'])

    CleanStyle.colors = ("red", "darkred", "darkolivegreen", "limegreen")
    label_count = len(years)
    if len(years) > 30:
        label_count = 10
    chart = pygal.StackedBar(dynamic_print_values=True, style=CleanStyle,
        width=1000, height=500, x_labels_major_count=label_count,
        show_minor_x_labels=False, x_label_rotation=20)
    #chart.title = "Preservation by Year"
    chart.x_title = "Year"
    #chart.y_title = "Count"
    chart.x_labels = [str(y['year']) for y in years]
    chart.add('None', [y['none'] for y in years])
    chart.add('Shadow', [y['shadows_only'] for y in years])
    chart.add('Dark', [y['dark'] for y in years])
    chart.add('Bright', [y['bright'] for y in years])
    return chart

def preservation_by_volume_histogram(rows: List[Dict]) -> pygal.Graph:
    """
    Note: this returns a raw pygal chart; it does not render it to SVG/PNG
    """

    volumes = sorted(rows, key=lambda x: x['volume'])

    CleanStyle.colors = ("red", "darkred", "darkolivegreen", "limegreen")
    label_count = len(volumes)
    if len(volumes) >= 30:
        label_count = 10
    chart = pygal.StackedBar(dynamic_print_values=True, style=CleanStyle,
        width=1000, height=500, x_labels_major_count=label_count,
        show_minor_x_labels=False, x_label_rotation=20)
    #chart.title = "Preservation by Year"
    chart.x_title = "Volume"
    #chart.y_title = "Count"
    chart.x_labels = [str(y['volume']) for y in volumes]
    chart.add('None', [y['none'] for y in volumes])
    chart.add('Shadow', [y['shadows_only'] for y in volumes])
    chart.add('Dark', [y['dark'] for y in volumes])
    chart.add('Bright', [y['bright'] for y in volumes])
    return chart