diff --git a/nupes/scripts/import_resultats_2024.py b/nupes/scripts/import_resultats_2024.py index 2597888..9ec33b9 100644 --- a/nupes/scripts/import_resultats_2024.py +++ b/nupes/scripts/import_resultats_2024.py @@ -170,8 +170,9 @@ def importer_resultats_circo(engine: Engine, verbose: bool = False) -> None: for row in tqdm(reader, desc="Circonscription", disable=not verbose): circo_code = row['Code circonscription législative'] dpt_code, circo_code = circo_code[:-2], circo_code[-2:] + circo_id = f"{dpt_code.zfill(2)}-{circo_code.zfill(2)}" - resultats_circo = session.execute(select(ResultatsCirconscription).filter_by(circo_id=circo_code)) \ + resultats_circo = session.execute(select(ResultatsCirconscription).filter_by(circo_id=circo_id)) \ .scalar_one_or_none() if not resultats_circo: resultats_dpt = session.execute(select(ResultatsDepartement) @@ -202,10 +203,10 @@ def importer_resultats_circo(engine: Engine, verbose: bool = False) -> None: resultats_dpt = session.execute(select(ResultatsDepartement) .filter_by(dpt_id=row['Code département'])).scalar_one() - resultats_circo = ResultatsCirconscription(circo_id=circo_code, + resultats_circo = ResultatsCirconscription(circo_id=circo_id, resultats_departement_id=resultats_dpt.id) session.add(resultats_circo) - resultats_circo = session.execute(select(ResultatsCirconscription).filter_by(circo_id=circo_code)) \ + resultats_circo = session.execute(select(ResultatsCirconscription).filter_by(circo_id=circo_id)) \ .scalar_one() resultats_circo.inscrits = int(row['Inscrits'])