diff --git a/nupes/scripts/legislatives2024/import_resultats.py b/nupes/scripts/legislatives2024/import_resultats.py index 5766ccc..49d31b2 100644 --- a/nupes/scripts/legislatives2024/import_resultats.py +++ b/nupes/scripts/legislatives2024/import_resultats.py @@ -173,6 +173,8 @@ def importer_resultats_commune(engine: Engine, verbose: bool = False) -> None: session.commit() + com_codes = {commune.code_insee for commune in session.execute(select(Commune)).scalars().all()} + for tour, file_url in tours: file = get_file(file_url, f"resultats-legislatives-2024-t{tour}-par-commune.csv") @@ -197,6 +199,10 @@ def importer_resultats_commune(engine: Engine, verbose: bool = False) -> None: # 3 royaumes à Wallis-et-Futuna, mais un seul résultat de commune com_code = "98611" + if com_code not in com_codes: + print(f"Commune {com_code} non trouvée") + continue + resultats_commune = session.execute(select(ResultatsCommuneLegislatives2024) .filter_by(commune_id=com_code)).scalar_one_or_none() if not resultats_commune: @@ -534,7 +540,7 @@ def importer_resultats_region(engine: Engine, verbose: bool = False) -> None: elif tour == 2: voix_nuance_reg.voix_t2 = voix - session.commit() + session.commit() def importer_resultats_france(engine: Engine, verbose: bool = False) -> None: @@ -599,5 +605,5 @@ def run(engine: Engine, verbose: bool = False) -> None: importer_resultats_region(engine, verbose) importer_resultats_departement(engine, verbose) importer_resultats_circo(engine, verbose) - # importer_resultats_commune(engine, verbose) + importer_resultats_commune(engine, verbose) importer_resultats_bv(engine, verbose)