nupes-elections/nupes/scripts/export_resultats_2024.py

265 lines
11 KiB
Python
Raw Normal View History

2024-06-08 21:06:41 +00:00
import json
from pathlib import Path
from sqlalchemy import Engine, select
from sqlalchemy.orm import Session
from tqdm import tqdm
2024-06-08 21:06:41 +00:00
from nupes.models.geographie import Region, Departement, Commune
from nupes.models.europeennes2024 import Bloc, Nuance, Liste, \
ResultatsFrance, ResultatsRegion, ResultatsDepartement, ResultatsCommune
DATA_DIR = Path(__file__).parent.parent.parent / 'data'
2024-06-08 21:06:41 +00:00
def exporter_listes(engine: Engine, debug: bool = False) -> None:
2024-06-08 21:06:41 +00:00
with Session(engine) as session:
blocs = session.execute(select(Bloc)).scalars().all()
blocs_json = []
for bloc in blocs:
bloc_json = {'id': bloc.id, 'nom': bloc.nom, 'couleur': bloc.couleur}
blocs_json.append(bloc_json)
file = DATA_DIR / "resultats" / "europeennes2024" / "blocs.json"
if not file.parent.is_dir():
file.parent.mkdir(parents=True)
with file.open('w') as f:
json.dump(blocs_json, f)
nuances = session.execute(select(Nuance)).scalars().all()
nuances_json = []
for nuance in nuances:
nuance_json = {'code': nuance.code, 'nom': nuance.nom, 'couleur': nuance.couleur}
nuances_json.append(nuance_json)
file = DATA_DIR / "resultats" / "europeennes2024" / "nuances.json"
if not file.parent.is_dir():
file.parent.mkdir(parents=True)
with file.open('w') as f:
json.dump(nuances_json, f)
listes = session.execute(select(Liste)).scalars().all()
listes_json = []
for liste in listes:
candidats = [
{'ordre': candidat.ordre, 'nom': candidat.nom, 'prenom': candidat.prenom, 'sexe': candidat.sexe.value,
'date_naissance': candidat.date_naissance.isoformat(), 'profession': candidat.profession,
'code_personnalite': candidat.code_personnalite.value, 'sortant': candidat.sortant}
for candidat in liste.candidats
]
liste_json = {'numero': liste.numero, 'nom': liste.nom, 'nuance': liste.nuance_id,
'bloc': liste.bloc.nom, 'candidats': candidats}
listes_json.append(liste_json)
file = DATA_DIR / "resultats" / "europeennes2024" / "listes.json"
if not file.parent.is_dir():
file.parent.mkdir(parents=True)
with file.open('w') as f:
json.dump(listes_json, f)
def exporter_resultats_france(engine: Engine, debug: bool = False) -> None:
2024-06-08 21:06:41 +00:00
with Session(engine) as session:
resultats_france = session.execute(select(ResultatsFrance)).scalar_one_or_none()
if not resultats_france:
resultats_france = ResultatsFrance()
session.add(resultats_france)
resultats_dict = {
"inscrits": resultats_france.inscrits,
"votants": resultats_france.votants,
"abstentions": resultats_france.abstentions,
2024-06-08 21:06:41 +00:00
"exprimes": resultats_france.exprimes,
"blancs": resultats_france.blancs,
"nuls": resultats_france.nuls,
'geometry': {},
}
resultats_listes = {}
resultats_dict['voix_listes'] = resultats_listes
for voix_liste in resultats_france.voix_listes:
resultats_listes[voix_liste.liste.numero] = voix_liste.voix
file = DATA_DIR / "resultats" / "europeennes2024" / "france.json"
if not file.parent.is_dir():
file.parent.mkdir(parents=True)
with file.open('w') as f:
json.dump(resultats_dict, f)
session.commit()
def exporter_resultats_regions(engine: Engine, debug: bool = False) -> None:
2024-06-08 21:06:41 +00:00
with Session(engine) as session:
regions = session.execute(select(Region)).scalars().all()
regions_json = []
regions_iterator = tqdm(regions, desc="Régions") if debug else regions
for region in regions_iterator:
2024-06-08 21:06:41 +00:00
region_json = {'code_insee': region.code_insee, 'nom': region.libelle,
'departements': [dpt.code_insee for dpt in region.departements]}
regions_json.append(region_json)
resultats_region = session.execute(select(ResultatsRegion).filter_by(region_id=region.code_insee)) \
.scalar_one_or_none()
if not resultats_region:
resultats_france = session.execute(select(ResultatsFrance)).scalar_one()
resultats_region = ResultatsRegion(region_id=region.code_insee, resultats_france_id=resultats_france.id)
session.add(resultats_region)
resultats_dict = {
"inscrits": resultats_region.inscrits,
"votants": resultats_region.votants,
"abstentions": resultats_region.abstentions,
2024-06-08 21:06:41 +00:00
"exprimes": resultats_region.exprimes,
"blancs": resultats_region.blancs,
"nuls": resultats_region.nuls,
'region': region_json,
'geometry': region.geometry,
}
resultats_listes = {}
resultats_dict['voix_listes'] = resultats_listes
for voix_liste in resultats_region.voix_listes:
resultats_listes[voix_liste.liste.numero] = voix_liste.voix
file = DATA_DIR / "resultats" / "europeennes2024" / "regions" / f"{region.code_insee}.json"
if not file.parent.is_dir():
file.parent.mkdir(parents=True)
with file.open('w') as f:
json.dump(resultats_dict, f)
session.commit()
regions_file = DATA_DIR / "resultats" / "europeennes2024" / "regions" / "regions.json"
if not regions_file.parent.is_dir():
regions_file.parent.mkdir(parents=True)
with regions_file.open('w') as f:
json.dump(regions_json, f)
def exporter_resultats_departements(engine: Engine, debug: bool = False) -> None:
2024-06-08 21:06:41 +00:00
with Session(engine) as session:
departements = session.execute(select(Departement)).scalars().all()
departements_json = []
iterator = tqdm(departements, desc="Départements") if debug else departements
for departement in iterator:
2024-06-08 21:06:41 +00:00
departement_json = {'code_insee': departement.code_insee, 'nom': departement.libelle,
'region': departement.region_code,
'communes': [commune.code_insee for commune in departement.communes]}
departements_json.append(departement_json)
resultats_departement = session.execute(
select(ResultatsDepartement).filter_by(dpt_id=departement.code_insee)).scalar_one_or_none()
if not resultats_departement:
resultats_region = session.execute(select(ResultatsRegion)
.filter_by(region_id=departement.region_code)).scalar_one()
resultats_departement = ResultatsDepartement(dpt_id=departement.code_insee,
resultats_region_id=resultats_region.id)
session.add(resultats_departement)
resultats_dict = {
"inscrits": resultats_departement.inscrits,
"votants": resultats_departement.votants,
"abstentions": resultats_departement.abstentions,
2024-06-08 21:06:41 +00:00
"exprimes": resultats_departement.exprimes,
"blancs": resultats_departement.blancs,
"nuls": resultats_departement.nuls,
'departement': departement_json,
'geometry': departement.geometry,
}
resultats_listes = {}
resultats_dict['voix_listes'] = resultats_listes
for voix_liste in resultats_departement.voix_listes:
resultats_listes[voix_liste.liste.numero] = voix_liste.voix
file = DATA_DIR / "resultats" / "europeennes2024" / "departements" / f"{departement.code_insee}.json"
if not file.parent.is_dir():
file.parent.mkdir(parents=True)
with file.open('w') as f:
json.dump(resultats_dict, f)
session.commit()
departements_file = DATA_DIR / "resultats" / "europeennes2024" / "departements" / "departements.json"
if not departements_file.parent.is_dir():
departements_file.parent.mkdir(parents=True)
with departements_file.open('w') as f:
json.dump(departements_json, f)
def exporter_resultats_communes(engine: Engine, debug: bool = False) -> None:
2024-06-08 21:06:41 +00:00
with Session(engine) as session:
communes = session.execute(select(Commune)).scalars().all()
2024-06-08 21:06:41 +00:00
communes_json = []
iterator = tqdm(communes, desc="Communes") if debug else communes
for commune in iterator:
2024-06-08 21:06:41 +00:00
commune_json = {'code_insee': commune.code_insee, 'nom': commune.libelle,
'departement': commune.departement_code,
'bureaux_vote': [bv.id for bv in commune.bureaux_vote]}
communes_json.append(commune_json)
resultats_commune = session.execute(
select(ResultatsCommune).filter_by(commune_id=commune.code_insee)).scalar_one_or_none()
if not resultats_commune:
resultats_departement = session.execute(select(ResultatsDepartement)
.filter_by(dpt_id=commune.departement_code)).scalar_one()
resultats_commune = ResultatsCommune(commune_id=commune.code_insee,
resultats_dpt_id=resultats_departement.id)
session.add(resultats_commune)
resultats_dict = {
"inscrits": resultats_commune.inscrits,
"votants": resultats_commune.votants,
"abstentions": resultats_commune.abstentions,
2024-06-08 21:06:41 +00:00
"exprimes": resultats_commune.exprimes,
"blancs": resultats_commune.blancs,
"nuls": resultats_commune.nuls,
'commune': commune_json,
'geometry': commune.geometry,
}
resultats_listes = {}
resultats_dict['voix_listes'] = resultats_listes
for voix_liste in resultats_commune.voix_listes:
resultats_listes[voix_liste.liste.numero] = voix_liste.voix
file = DATA_DIR / "resultats" / "europeennes2024" / "communes" / f"{commune.code_insee}.json"
if not file.parent.is_dir():
file.parent.mkdir(parents=True)
with file.open('w') as f:
json.dump(resultats_dict, f)
session.commit()
communes_file = DATA_DIR / "resultats" / "europeennes2024" / "communes" / "communes.json"
if not communes_file.parent.is_dir():
communes_file.parent.mkdir(parents=True)
with communes_file.open('w') as f:
json.dump(communes_json, f)
def run(engine: Engine, debug: bool = False) -> None:
exporter_listes(engine, debug)
exporter_resultats_france(engine, debug)
exporter_resultats_regions(engine, debug)
exporter_resultats_departements(engine, debug)
# FIXME Les communes prennent trop de temps
# exporter_resultats_communes(engine, debug)