Ajoute le graphique des personnages représentés

lib/rebrickable/minifig_characters.py

@@ -0,0 +1,41 @@
+"""Agrégation des minifigs par personnage représenté."""
+
+from collections import defaultdict
+from pathlib import Path
+from typing import Dict, Iterable, List, Sequence
+
+from lib.rebrickable.stats import read_rows
+from lib.filesystem import ensure_parent_dir
+import csv
+
+
+def load_minifigs_by_set(path: Path) -> List[dict]:
+    """Charge le CSV minifigs_by_set."""
+    return read_rows(path)
+
+
+def aggregate_by_character(rows: Iterable[dict]) -> List[dict]:
+    """Compte les minifigs distinctes par personnage (fig_num unique)."""
+    fig_nums_by_character: Dict[str, set] = defaultdict(set)
+    for row in rows:
+        character = row["known_character"].strip()
+        fig_num = row["fig_num"].strip()
+        if character == "" or fig_num == "":
+            continue
+        fig_nums_by_character[character].add(fig_num)
+    aggregates: List[dict] = []
+    for character, fig_nums in fig_nums_by_character.items():
+        aggregates.append({"known_character": character, "minifig_count": len(fig_nums)})
+    aggregates.sort(key=lambda r: (-r["minifig_count"], r["known_character"]))
+    return aggregates
+
+
+def write_character_counts(path: Path, rows: Sequence[dict]) -> None:
+    """Écrit le CSV des comptes par personnage."""
+    ensure_parent_dir(path)
+    fieldnames = ["known_character", "minifig_count"]
+    with path.open("w", newline="") as csv_file:
+        writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
+        writer.writeheader()
+        for row in rows:
+            writer.writerow(row)