nupes-elections/nupes/scripts/import_candidats_2024.py

174 lines
9.7 KiB
Python

import csv
import datetime
from sqlalchemy import Engine, select
from sqlalchemy.orm import Session
from nupes.cache import get_file
from nupes.models.europeennes2024 import Bloc, Nuance, Liste, Candidat
def creer_blocs(engine: Engine, debug: bool = False) -> None:
blocs = [
{"id": 1, "nom": "Gauche", "couleur": "#BB1840"},
{"id": 2, "nom": "Droite libérale", "couleur": "#FFEB00"},
{"id": 3, "nom": "Droite conservatrice", "couleur": "#0066CC"},
{"id": 4, "nom": "Extrême droite", "couleur": "#0D378A"},
{"id": 5, "nom": "Autres", "couleur": "#DCBFA3"}
]
with Session(engine) as session:
for bloc_dict in blocs:
if bloc := session.execute(select(Bloc).filter_by(id=bloc_dict["id"])).scalar_one_or_none():
bloc.nom = bloc_dict["nom"]
bloc.couleur = bloc_dict["couleur"]
else:
session.add(Bloc(**bloc_dict))
session.commit()
def creer_nuances(engine: Engine, debug: bool = False) -> None:
nuances = [
{"code": "LEXG", "nom": "Liste d'extrême gauche", "couleur": "#BB0000"},
{"code": "LCOM", "nom": "Liste du Parti communiste français", "couleur": "#DD0000"},
{"code": "LFI", "nom": "Liste de La France insoumise", "couleur": "#CC2443"},
{"code": "LUG", "nom": "Liste d'union de la gauche", "couleur": "#FF8080"},
{"code": "LVEC", "nom": "Liste Les Écologistes", "couleur": "#00C000"},
{"code": "LDVG", "nom": "Liste divers gauche", "couleur": "#FFC0C0"},
{"code": "LECO", "nom": "Liste écologiste", "couleur": "#77FF77"},
{"code": "LDIV", "nom": "Liste Divers", "couleur": "#DCDCDC"},
{"code": "LENS", "nom": "Liste Ensemble", "couleur": "#FFEB00"},
{"code": "LLR", "nom": "Liste des Républicains", "couleur": "#0066CC"},
{"code": "LDVD", "nom": "Liste divers droite", "couleur": "#26C4EC"},
{"code": "LRN", "nom": "Liste du Rassemblement national", "couleur": "#0D378A"},
{"code": "LREC", "nom": "Liste Reconquête !", "couleur": "#404040"},
{"code": "LEXD", "nom": "Liste d'extrême droite", "couleur": "#404040"},
]
with Session(engine) as session:
for nuance_dict in nuances:
if nuance := session.execute(select(Nuance).filter_by(code=nuance_dict["code"])).scalar_one_or_none():
nuance.nom = nuance_dict["nom"]
nuance.couleur = nuance_dict["couleur"]
else:
session.add(Nuance(**nuance_dict))
session.commit()
def importer_listes(engine: Engine, debug: bool = False) -> None:
listes = [
{"numero": 1, "nom": "Pour une humanité souveraine", "nuance_id": "LDIV", "bloc_id": 5},
{"numero": 2, "nom": "Pour une démocratie réelle : Décidons nous-mêmes !",
"nuance_id": "LDIV", "bloc_id": 5},
{"numero": 3, "nom": "La France fière, menée par Marion Maréchal et soutenue par Éric Zemmour",
"nuance_id": "LREC", "bloc_id": 4},
{"numero": 4, "nom": "La France insoumise - Union populaire", "nuance_id": "LFI", "bloc_id": 1},
{"numero": 5, "nom": "La France revient ! Avec Jordan Bardella et Marine Le Pen",
"nuance_id": "LRN", "bloc_id": 4},
{"numero": 6, "nom": "Europe Écologie", "nuance_id": "LVEC", "bloc_id": 1},
{"numero": 7, "nom": "Free Palestine", "nuance_id": "LDIV", "bloc_id": 1},
{"numero": 8, "nom": "Parti animaliste - Les animaux comptent, votre voix aussi",
"nuance_id": "LDIV", "bloc_id": 5},
{"numero": 9, "nom": "Parti révolutionnaire Communistes", "nuance_id": "LEXG", "bloc_id": 1},
{"numero": 10, "nom": "Parti pirate", "nuance_id": "LDIV", "bloc_id": 1},
{"numero": 11, "nom": "Besoin d'Europe", "nuance_id": "LENS", "bloc_id": 2},
{"numero": 12, "nom": "PACE - Parti des citoyens européens, pour l'armée européenne, "
"pour l'Europe sociale, pour la planète !", "nuance_id": "LDIV", "bloc_id": 1},
{"numero": 13, "nom": "Équinoxe : écologie pratique et renouveau démocratique",
"nuance_id": "LECO", "bloc_id": 5},
{"numero": 14, "nom": "Écologie positive et territoires", "nuance_id": "LECO", "bloc_id": 5},
{"numero": 15, "nom": "Liste Asselineau-Frexit, pour le pouvoir d'achat et pour la paix",
"nuance_id": "LDIV", "bloc_id": 4},
{"numero": 16, "nom": "Paix et Décroissance", "nuance_id": "LEXG", "bloc_id": 1},
{"numero": 17, "nom": "Pour une autre Europe", "nuance_id": "LDIV", "bloc_id": 5},
{"numero": 18, "nom": "La droite pour faire entendre la voix de la France en Europe",
"nuance_id": "LLR", "bloc_id": 4},
{"numero": 19, "nom": "Lutte ouvrière le camp des travailleurs", "nuance_id": "LEXG", "bloc_id": 1},
{"numero": 20, "nom": "Changer l'Europe", "nuance_id": "LDVG", "bloc_id": 1},
{"numero": 21, "nom": "Nous le peuple", "nuance_id": "LDIV", "bloc_id": 5},
{"numero": 22, "nom": "Pour un monde sans frontières ni patrons, urgence révolution !",
"nuance_id": "LEXG", "bloc_id": 1},
{"numero": 23, "nom": "\"Pour le pain, la paix, la liberté !\" présentée par le Parti des travailleurs",
"nuance_id": "LEXG", "bloc_id": 1},
{"numero": 24, "nom": "L'Europe ça suffit !", "nuance_id": "LEXD", "bloc_id": 4},
{"numero": 25, "nom": "Non ! Prenons-nous en mains", "nuance_id": "LDIV", "bloc_id": 4},
{"numero": 26, "nom": "Forteresse Europe — Liste d'unité nationaliste", "nuance_id": "LEXD", "bloc_id": 4},
{"numero": 27, "nom": "Réveiller l'Europe", "nuance_id": "LUG", "bloc_id": 1},
{"numero": 28, "nom": "Non à l'UE et à l'OTAN, communistes pour la paix et le progrès social",
"nuance_id": "LEXG", "bloc_id": 1},
{"numero": 29, "nom": "Alliance rurale", "nuance_id": "LDVD", "bloc_id": 3},
{"numero": 30, "nom": "France libre", "nuance_id": "LDIV", "bloc_id": 4},
{"numero": 31, "nom": "Europe Territoires Écologie", "nuance_id": "LDVG", "bloc_id": 1},
{"numero": 32, "nom": "La ruche citoyenne", "nuance_id": "LDIV", "bloc_id": 5},
{"numero": 33, "nom": "Gauche unie pour le monde du travail soutenue par Fabien Roussel",
"nuance_id": "LCOM", "bloc_id": 1},
{"numero": 34, "nom": "Défendre les enfants", "nuance_id": "LDIV", "bloc_id": 5},
{"numero": 35, "nom": "Écologie au centre", "nuance_id": "LECO", "bloc_id": 2},
{"numero": 36, "nom": "Démocratie représentative", "nuance_id": "LDIV", "bloc_id": 1},
{"numero": 37, "nom": "Espéranto langue commune", "nuance_id": "LDIV", "bloc_id": 5},
{"numero": 38, "nom": "Liberté démocratique française", "nuance_id": "LDIV", "bloc_id": 4},
]
with Session(engine) as session:
for liste_dict in listes:
if liste := session.execute(select(Liste).filter_by(numero=liste_dict["numero"])).scalar_one_or_none():
liste.nom = liste_dict["nom"]
liste.nuance_id = liste_dict["nuance_id"]
liste.bloc_id = liste_dict["bloc_id"]
else:
liste = Liste(numero=liste_dict['numero'], nom=liste_dict['nom'],
nom_majuscules=liste_dict['nom'].upper(), nuance_id=liste_dict['nuance_id'],
bloc_id=liste_dict['bloc_id'])
session.add(liste)
session.commit()
def importer_candidats(engine: Engine, debug: bool = False) -> None:
DATASET_URL = "https://www.data.gouv.fr/fr/datasets/r/483cd4bd-1b0e-4b52-a923-aadadf9c8f62"
file = get_file(DATASET_URL, "candidats_2024.csv")
with file.open('r') as f:
with Session(engine) as session:
reader = csv.DictReader(f, delimiter=';')
for line in reader:
line: dict[str, str]
numero_panneau = line.get("Numéro de panneau")
liste = session.execute(select(Liste).filter_by(numero=numero_panneau)).scalar_one()
liste.nom_majuscules = line['Libellé de la liste']
if candidat := session.execute(select(Candidat).filter_by(liste_id=liste.id, ordre=line['Ordre'])) \
.scalar_one_or_none():
candidat.liste_id = liste.id
candidat.nom = line['Nom sur le bulletin de vote']
candidat.prenom = line['Prénom sur le bulletin de vote']
candidat.sexe = Candidat.Genre(line['Sexe'])
candidat.date_naissance = datetime.datetime.strptime(line['Date de naissance'], "%d/%m/%Y").date()
candidat.profession = line['Profession']
candidat.code_personnalite = Candidat.Personnalite(line['Code personnalité'])
candidat.sortant = line['Sortant'] == "OUI"
else:
candidat = Candidat(
liste_id=liste.id,
ordre=int(line['Ordre']),
nom=line['Nom sur le bulletin de vote'],
prenom=line['Prénom sur le bulletin de vote'],
sexe=Candidat.Genre(line['Sexe']).name,
date_naissance=datetime.datetime.strptime(line['Date de naissance'], "%d/%m/%Y").date(),
profession=line['Profession'],
code_personnalite=Candidat.Personnalite(line['Code personnalité']).name,
sortant=line['Sortant'] == "OUI",
)
session.add(candidat)
session.commit()
def run(engine: Engine, debug: bool = False) -> None:
creer_blocs(engine, debug)
creer_nuances(engine, debug)
importer_listes(engine, debug)
importer_candidats(engine, debug)