Ajoute le genre des personnages et colore les graphiques

lib/rebrickable/minifigs_by_set.py

@@ -128,6 +128,16 @@ def load_aliases(path: Path) -> Dict[str, str]:
     return aliases
 
 
+def load_gender_overrides(path: Path) -> Dict[str, str]:
+    """Charge les correspondances personnage -> genre."""
+    overrides: Dict[str, str] = {}
+    with path.open() as gender_file:
+        reader = csv.DictReader(gender_file)
+        for row in reader:
+            overrides[row["known_character"].lower()] = row["gender"]
+    return overrides
+
+
 def normalize_known_character(raw_known: str, extracted_name: str, aliases: Dict[str, str]) -> str:
     """Nettoie et mappe un nom vers une version canonique."""
     base = raw_known or extracted_name
@@ -201,6 +211,7 @@ def aggregate_heads_by_set(
     minifig_heads: Dict[str, Set[str]],
     minifig_catalog: Dict[str, dict],
     aliases: Dict[str, str],
+    gender_overrides: Dict[str, str],
 ) -> List[dict]:
     """Agrège les têtes de minifigs par set en éliminant les rechanges et doublons."""
     seen: Set[Tuple[str, str]] = set()
@@ -228,12 +239,14 @@ def aggregate_heads_by_set(
         normalized = normalize_known_character(known_character, extracted, aliases)
         if matched_fig == "":
             continue
+        gender = gender_overrides.get(normalized.lower(), "")
         heads.append(
             {
                 "set_num": row["set_num"],
                 "part_num": row["part_num"],
                 "known_character": normalized,
                 "fig_num": matched_fig,
+                "gender": gender,
             }
         )
         seen.add(key)
@@ -244,7 +257,7 @@ def aggregate_heads_by_set(
 def write_heads_by_set(destination_path: Path, rows: Sequence[dict]) -> None:
     """Écrit le CSV intermédiaire listant les têtes de minifigs par set."""
     ensure_parent_dir(destination_path)
-    fieldnames = ["set_num", "part_num", "known_character", "fig_num"]
+    fieldnames = ["set_num", "part_num", "known_character", "fig_num", "gender"]
     with destination_path.open("w", newline="") as csv_file:
         writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
         writer.writeheader()
@@ -260,6 +273,7 @@ def build_minifigs_by_set(
     inventory_minifigs_path: Path,
     minifigs_path: Path,
     aliases_path: Path,
+    gender_overrides_path: Path,
     destination_path: Path,
 ) -> None:
     """Construit le CSV listant les têtes de minifigs présentes par set."""
@@ -271,6 +285,7 @@ def build_minifigs_by_set(
     minifig_heads = build_minifig_heads_lookup(minifig_catalog, latest_inventories, inventory_parts_path, head_parts)
     set_minifigs = build_set_minifigs_lookup(latest_inventories, inventory_minifigs_path)
     aliases = load_aliases(aliases_path)
+    gender_overrides = load_gender_overrides(gender_overrides_path)
     heads = aggregate_heads_by_set(
         parts_rows,
         parts_catalog,
@@ -279,5 +294,6 @@ def build_minifigs_by_set(
         minifig_heads,
         minifig_catalog,
         aliases,
+        gender_overrides,
     )
     write_heads_by_set(destination_path, heads)