Chiffre les personnages par année avec heatmap

lib/rebrickable/minifig_characters.py

@@ -56,10 +56,11 @@ def aggregate_presence_by_year(
     sets_years: Dict[str, str],
     excluded_characters: Sequence[str] | None = None,
 ) -> List[dict]:
-    """Construit la présence binaire des personnages par année (hors figurants)."""
+    """Compte le nombre total de minifigs par personnage et par année (hors figurants)."""
     excluded = set(excluded_characters or [])
-    presence: set[tuple[str, int]] = set()
+    counts: Dict[tuple[str, int], int] = defaultdict(int)
     years_all = {int(year) for year in sets_years.values()}
+    characters_all: Set[str] = set()
     for row in minifigs_rows:
         character = row["known_character"].strip()
         fig_num = row["fig_num"].strip()
@@ -70,17 +71,20 @@ def aggregate_presence_by_year(
         year = sets_years.get(row["set_num"])
         if year is None:
             continue
-        presence.add((character, int(year)))
+        year_int = int(year)
+        counts[(character, year_int)] += 1
+        characters_all.add(character)
     years = sorted(years_all)
-    characters = sorted({character for character, _ in presence})
+    characters = sorted(characters_all)
     results: List[dict] = []
     for character in characters:
         for year in years:
+            count = counts.get((character, year), 0)
             results.append(
                 {
                     "known_character": character,
                     "year": str(year),
-                    "present": "1" if (character, year) in presence else "0",
+                    "minifig_count": str(count),
                 }
             )
     return results
@@ -89,7 +93,7 @@ def aggregate_presence_by_year(
 def write_presence_by_year(path: Path, rows: Sequence[dict]) -> None:
     """Écrit la matrice présence binaire année/personnage."""
     ensure_parent_dir(path)
-    fieldnames = ["known_character", "year", "present"]
+    fieldnames = ["known_character", "year", "minifig_count"]
     with path.open("w", newline="") as csv_file:
         writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
         writer.writeheader()