from typing import Any, Dict, List, Tuple 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 Rows are tuples of: (year: float or int, in_ia: bool, count: int) """ raw_years = [int(r[0]) for r in rows] years_dict = dict() if raw_years: for y in range(min(raw_years), max(raw_years) + 1): years_dict[int(y)] = dict(year=int(y), available=0, missing=0) for r in rows: if r[1]: years_dict[int(r[0])]["available"] = r[2] else: years_dict[int(r[0])]["missing"] = r[2] years: List[Dict[str, Any]] = sorted(years_dict.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(v["year"]) for v in years] chart.add("via Fatcat", [v["available"] for v in years]) chart.add("Missing", [v["missing"] for v in years]) return chart def preservation_by_year_histogram( rows: List[Dict], merge_shadows: bool = False ) -> pygal.Graph: """ Note: this returns a raw pygal chart; it does not render it to SVG/PNG Rows are dict with keys as preservation types and values as counts (int). There is also a 'year' key with float/int value. """ years = sorted(rows, key=lambda x: x["year"]) if merge_shadows: CleanStyle.colors = ("red", "darkolivegreen", "limegreen") else: 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] if merge_shadows: chart.add("None", [y["none"] + y["shadows_only"] for y in years]) else: 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_date_histogram( rows: List[Dict], merge_shadows: bool = False ) -> pygal.Graph: """ Note: this returns a raw pygal chart; it does not render it to SVG/PNG Rows are dict with keys as preservation types and values as counts (int). There is also a 'date' key with str value. """ dates = sorted(rows, key=lambda x: x["date"]) if merge_shadows: CleanStyle.colors = ("red", "darkolivegreen", "limegreen") else: CleanStyle.colors = ("red", "darkred", "darkolivegreen", "limegreen") label_count = len(dates) if len(dates) > 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 Date" chart.x_title = "Date" # chart.y_title = "Count" chart.x_labels = [str(y["date"]) for y in dates] if merge_shadows: chart.add("None", [y["none"] + y["shadows_only"] for y in dates]) else: chart.add("None", [y["none"] for y in dates]) chart.add("Shadow", [y["shadows_only"] for y in dates]) chart.add("Dark", [y["dark"] for y in dates]) chart.add("Bright", [y["bright"] for y in dates]) return chart def preservation_by_volume_histogram( rows: List[Dict], merge_shadows: bool = False ) -> pygal.Graph: """ Note: this returns a raw pygal chart; it does not render it to SVG/PNG Rows are dict with keys as preservation types and values as counts (int). There is also a 'volume' key with str value. """ volumes = sorted(rows, key=lambda x: x["volume"]) if merge_shadows: CleanStyle.colors = ("red", "darkolivegreen", "limegreen") else: 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 Volume" chart.x_title = "Volume" # chart.y_title = "Count" chart.x_labels = [str(y["volume"]) for y in volumes] if merge_shadows: chart.add("None", [y["none"] + y["shadows_only"] for y in volumes]) else: 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