From ab9267cb13f7be756e8362d5fd1b0d3264c34253 Mon Sep 17 00:00:00 2001 From: Emmy D'Anello Date: Sat, 15 Jun 2024 20:39:42 +0200 Subject: [PATCH] =?UTF-8?q?Correction=20import=20r=C3=A9sultats=20Saint-Ma?= =?UTF-8?q?rtin/Saint-Barth=C3=A9l=C3=A9my?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- nupes/scripts/import_resultats_2024.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/nupes/scripts/import_resultats_2024.py b/nupes/scripts/import_resultats_2024.py index a8d2f3c..280a2e8 100644 --- a/nupes/scripts/import_resultats_2024.py +++ b/nupes/scripts/import_resultats_2024.py @@ -102,16 +102,20 @@ def importer_resultats_commune(engine: Engine, verbose: bool = False) -> None: with Session(engine) as session: for row in tqdm(reader, desc="Commune", disable=not verbose): com_code = str(row['Code commune']).zfill(5) + dpt_code = row['Code département'] + + if dpt_code == "ZX": + dpt_code = "977" resultats_commune = session.execute(select(ResultatsCommune).filter_by(commune_id=com_code)) \ .scalar_one_or_none() if not resultats_commune: resultats_dpt = session.execute(select(ResultatsDepartement) - .filter_by(dpt_id=row['Code département'])) \ + .filter_by(dpt_id=dpt_code)) \ .scalar_one_or_none() if not resultats_dpt: dpt = session.execute(select(Departement) - .filter_by(code_insee=f"{row['Code département'].zfill(2)}")) \ + .filter_by(code_insee=f"{dpt_code.zfill(2)}")) \ .scalar_one() resultats_reg = session.execute(select(ResultatsRegion) .filter_by(region_id=dpt.region_code)) \ @@ -128,11 +132,11 @@ def importer_resultats_commune(engine: Engine, verbose: bool = False) -> None: resultats_reg = session.execute(select(ResultatsRegion) .filter_by(region_id=dpt.region_code)).scalar_one() - resultats_dpt = ResultatsDepartement(dpt_id=row['Code département'], + resultats_dpt = ResultatsDepartement(dpt_id=dpt_code, resultats_region_id=resultats_reg.id) session.add(resultats_dpt) resultats_dpt = session.execute(select(ResultatsDepartement) - .filter_by(dpt_id=row['Code département'])).scalar_one() + .filter_by(dpt_id=dpt_code)).scalar_one() resultats_commune = ResultatsCommune(commune_id=com_code, resultats_dpt_id=resultats_dpt.id) session.add(resultats_commune)