Nécessité du mode debug pour un contenu verbeux

This commit is contained in:
2024-06-09 17:45:56 +02:00
parent aabd3f0153
commit dec9f414ac
5 changed files with 68 additions and 54 deletions

View File

@ -8,7 +8,7 @@ from nupes.cache import get_file
from nupes.models.europeennes2024 import Bloc, Nuance, Liste, Candidat
def creer_blocs(engine: Engine) -> None:
def creer_blocs(engine: Engine, debug: bool = False) -> None:
blocs = [
{"id": 1, "nom": "Gauche", "couleur": "#BB1840"},
{"id": 2, "nom": "Droite libérale", "couleur": "#FFEB00"},
@ -28,7 +28,7 @@ def creer_blocs(engine: Engine) -> None:
session.commit()
def creer_nuances(engine: Engine) -> None:
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"},
@ -57,7 +57,7 @@ def creer_nuances(engine: Engine) -> None:
session.commit()
def importer_listes(engine: Engine) -> None:
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 !",
@ -126,7 +126,7 @@ def importer_listes(engine: Engine) -> None:
session.commit()
def importer_candidats(engine: Engine) -> None:
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")
@ -166,8 +166,8 @@ def importer_candidats(engine: Engine) -> None:
session.commit()
def run(engine: Engine) -> None:
creer_blocs(engine)
creer_nuances(engine)
importer_listes(engine)
importer_candidats(engine)
def run(engine: Engine, debug: bool = False) -> None:
creer_blocs(engine, debug)
creer_nuances(engine, debug)
importer_listes(engine, debug)
importer_candidats(engine, debug)