120 lines
5.9 KiB
Python
120 lines
5.9 KiB
Python
|
from csv import DictReader
|
||
|
|
|
||
|
|
from sqlalchemy import Engine, select
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
from tqdm import tqdm
|
||
|
|
|
||
|
|
from nupes.cache import get_file
|
||
|
|
from nupes.models import BureauVote, Departement
|
||
|
|
from nupes.models.europeennes2024 import ResultatsBureauVote, Liste, VoixListeBureauVote, ResultatsCommune, \
|
||
|
|
ResultatsDepartement, ResultatsRegion, ResultatsFrance
|
||
|
|
|
||
|
|
|
||
|
|
def importer_resultats_bv(engine: Engine, debug: bool = False) -> None:
|
||
|
|
file = get_file("https://www.data.gouv.fr/fr/datasets/r/937bb638-a487-40cd-9a0b-610d539a4207",
|
||
|
|
"resultats-temporaires-par-bureau-de-vote.csv")
|
||
|
|
|
||
|
|
with file.open('r') as f:
|
||
|
|
next(f) # On saute la première ligne
|
||
|
|
reader = DictReader(f)
|
||
|
|
with Session(engine) as session:
|
||
|
|
for row in tqdm(reader, desc="Bureau de vote", disable=not debug):
|
||
|
|
com_code = str(row['Code commune']).zfill(5)
|
||
|
|
bv_code = row['Code BV']
|
||
|
|
bv_id = f"{com_code}_{bv_code}"
|
||
|
|
|
||
|
|
if not session.execute(select(BureauVote).filter_by(id=bv_id)).scalar_one_or_none():
|
||
|
|
print(f"Bureau de vote {bv_id} non trouvé")
|
||
|
|
continue
|
||
|
|
|
||
|
|
resultats_bv = session.execute(select(ResultatsBureauVote).filter_by(bv_id=bv_id)).scalar_one_or_none()
|
||
|
|
if not resultats_bv:
|
||
|
|
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'])) \
|
||
|
|
.scalar_one_or_none()
|
||
|
|
if not resultats_dpt:
|
||
|
|
dpt = session.execute(select(Departement)
|
||
|
|
.filter_by(code_insee=f"{row['Code département'].zfill(2)}")) \
|
||
|
|
.scalar_one()
|
||
|
|
resultats_reg = session.execute(select(ResultatsRegion)
|
||
|
|
.filter_by(region_id=dpt.region_code)) \
|
||
|
|
.scalar_one_or_none()
|
||
|
|
if not resultats_reg:
|
||
|
|
resultats_france = session.execute(select(ResultatsFrance)).scalar_one_or_none()
|
||
|
|
if not resultats_france:
|
||
|
|
session.add(ResultatsFrance())
|
||
|
|
resultats_france = session.execute(select(ResultatsFrance)).scalar_one()
|
||
|
|
|
||
|
|
resultats_reg = ResultatsRegion(region_id=str(dpt.region_code),
|
||
|
|
resultats_france_id=resultats_france.id)
|
||
|
|
session.add(resultats_reg)
|
||
|
|
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_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()
|
||
|
|
|
||
|
|
resultats_commune = ResultatsCommune(commune_id=com_code, resultats_dpt_id=resultats_dpt.id)
|
||
|
|
session.add(resultats_commune)
|
||
|
|
resultats_commune = session.execute(select(ResultatsCommune).filter_by(commune_id=com_code)) \
|
||
|
|
.scalar_one()
|
||
|
|
|
||
|
|
resultats_bv = ResultatsBureauVote(bv_id=bv_id, resultats_commune_id=resultats_commune.id)
|
||
|
|
session.add(resultats_bv)
|
||
|
|
resultats_bv = session.execute(select(ResultatsBureauVote).filter_by(bv_id=bv_id)).scalar_one()
|
||
|
|
|
||
|
|
resultats_bv.inscrits = int(row['Inscrits'])
|
||
|
|
resultats_bv.votants = int(row['Votants'])
|
||
|
|
resultats_bv.abstentions = int(row['Abstentions'])
|
||
|
|
resultats_bv.exprimes = int(row['Exprimés'])
|
||
|
|
resultats_bv.blancs = int(row['Blancs'])
|
||
|
|
resultats_bv.nuls = int(row['Nuls'])
|
||
|
|
|
||
|
|
for liste in session.execute(select(Liste)).scalars().all():
|
||
|
|
voix_liste_bv = session.execute(select(VoixListeBureauVote)
|
||
|
|
.filter_by(resultats_bureau_vote_id=resultats_bv.id,
|
||
|
|
liste_id=liste.id)) \
|
||
|
|
.scalar_one_or_none()
|
||
|
|
if not voix_liste_bv:
|
||
|
|
voix_liste_bv = VoixListeBureauVote(resultats_bureau_vote_id=resultats_bv.id, liste_id=liste.id)
|
||
|
|
session.add(voix_liste_bv)
|
||
|
|
|
||
|
|
voix_liste_bv.voix = int(row[f"Voix {liste.id}"])
|
||
|
|
|
||
|
|
session.commit()
|
||
|
|
|
||
|
|
|
||
|
|
def calculer_resultats_commune(engine: Engine, debug: bool = False) -> None:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def calculer_resultats_circo(engine: Engine, debug: bool = False) -> None:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def calculer_resultats_departement(engine: Engine, debug: bool = False) -> None:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def calculer_resultats_region(engine: Engine, debug: bool = False) -> None:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def calculer_resultats_national(engine: Engine, debug: bool = False) -> None:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def run(engine: Engine, debug: bool = False) -> None:
|
||
|
|
importer_resultats_bv(engine, debug)
|
||
|
|
calculer_resultats_commune(engine, debug)
|
||
|
|
# calculer_resultats_circo(engine, debug)
|
||
|
|
calculer_resultats_departement(engine, debug)
|
||
|
|
calculer_resultats_region(engine, debug)
|
||
|
|
calculer_resultats_national(engine, debug)
|