1
etude_lego_jurassic_world/scripts/compute_global_minifig_heads.py

40 lines
1.3 KiB
Python

"""Agrégation globale des couleurs de têtes de minifigs (catalogue complet)."""
from pathlib import Path
from lib.rebrickable.global_minifig_heads import aggregate_global_heads_by_year, write_global_heads_by_year
INVENTORIES_PATH = Path("data/raw/inventories.csv")
INVENTORY_PARTS_PATH = Path("data/raw/inventory_parts.csv")
PARTS_PATH = Path("data/raw/parts.csv")
COLORS_PATH = Path("data/raw/colors.csv")
SETS_PATH = Path("data/raw/sets.csv")
DESTINATION_PATH = Path("data/intermediate/global_minifig_heads_by_year.csv")
DESTINATION_PRINTED_PATH = Path("data/intermediate/global_minifig_heads_printed_by_year.csv")
def main() -> None:
"""Construit l'agrégat mondial des têtes de minifigs par couleur et année."""
heads_by_year = aggregate_global_heads_by_year(
INVENTORIES_PATH,
INVENTORY_PARTS_PATH,
PARTS_PATH,
COLORS_PATH,
SETS_PATH,
)
write_global_heads_by_year(DESTINATION_PATH, heads_by_year)
heads_printed_by_year = aggregate_global_heads_by_year(
INVENTORIES_PATH,
INVENTORY_PARTS_PATH,
PARTS_PATH,
COLORS_PATH,
SETS_PATH,
require_printed=True,
)
write_global_heads_by_year(DESTINATION_PRINTED_PATH, heads_printed_by_year)
if __name__ == "__main__":
main()