diff --git a/nupes/scripts/import_resultats_web_2024.py b/nupes/scripts/import_resultats_web_2024.py index 0c8db11..fe3013b 100644 --- a/nupes/scripts/import_resultats_web_2024.py +++ b/nupes/scripts/import_resultats_web_2024.py @@ -48,7 +48,8 @@ def importer_resultats_france(engine: Engine, debug: bool = False) -> None: voix_liste_france.voix = voix_liste["voix"] else: liste = session.execute(select(Liste).filter_by(nom_majuscules=voix_liste["nom"].upper())).scalar_one() - voix_liste_france = VoixListeFrance(liste_id=liste.id, voix=voix_liste["voix"]) + voix_liste_france = VoixListeFrance(liste_id=liste.id, resultats_france_id=resultats_france.id, + voix=voix_liste["voix"]) session.add(voix_liste_france) session.commit() @@ -103,7 +104,8 @@ def importer_resultats_regions(engine: Engine, debug: bool = False) -> None: else: liste = session.execute(select(Liste).filter_by(nom_majuscules=voix_liste["nom"].upper())) \ .scalar_one() - voix_liste_region = VoixListeRegion(liste_id=liste.id, voix=voix_liste["voix"]) + voix_liste_region = VoixListeRegion(liste_id=liste.id, resultats_region_id=resultats_region.id, + voix=voix_liste["voix"]) session.add(voix_liste_region) session.commit() @@ -166,7 +168,9 @@ def importer_resultats_departements(engine: Engine, debug: bool = False) -> None else: liste = session.execute(select(Liste).filter_by(nom_majuscules=voix_liste["nom"].upper())) \ .scalar_one() - voix_liste_departement = VoixListeDepartement(liste_id=liste.id, voix=voix_liste["voix"]) + voix_liste_departement = VoixListeDepartement(liste_id=liste.id, + resultats_departement_id=resultats_departement.id, + voix=voix_liste["voix"]) session.add(voix_liste_departement) session.commit() @@ -240,7 +244,8 @@ def importer_resultats_communes(engine: Engine, debug: bool = False) -> None: else: liste = session.execute(select(Liste).filter_by(nom_majuscules=voix_liste["nom"].upper())) \ .scalar_one() - voix_liste_commune = VoixListeCommune(liste_id=liste.id, voix=voix_liste["voix"]) + voix_liste_commune = VoixListeCommune(liste_id=liste.id, resultats_commune_id=resultats_commune.id, + voix=voix_liste["voix"]) session.add(voix_liste_commune) session.commit() @@ -266,7 +271,7 @@ def analyser_resultats(file) -> dict: liste = {} resultats_par_liste.append(liste) - liste['nom'] = cells[0].string + liste['nom'] = cells[0].text.replace("\n", "").replace("\u202f", " ").strip() liste['voix'] = int(cells[2].string.replace("\n", "").replace(" ", "").replace("\u202f", "").strip()) elif table.find('td', text=re.compile("Inscrits")): tbody = table.tbody