diff --git a/main.py b/main.py index a924780..c7447db 100755 --- a/main.py +++ b/main.py @@ -6,8 +6,8 @@ import os from dotenv import load_dotenv from sqlalchemy import create_engine -from nupes.scripts import import_candidats_2024, import_geographie, import_resultats_2024, \ - import_resultats_web_2024, export_resultats_2024 +from nupes.scripts import import_geographie +from nupes.scripts import europeennes2024 def parse_args(): @@ -32,7 +32,14 @@ def main(): debug = args.debug verbose = args.verbose - engine = create_engine(os.getenv("SQLALCHEMY_URL"), echo=args.debug) + match args.type, args.year: + case "européennes", 2024: + election_module = europeennes2024 + case _: + print(f"Type d'élection {args.type} {args.year} non reconnu") + return + + engine = create_engine(os.getenv("SQLALCHEMY_URL"), echo=debug) match args.action: case "import_geographie": @@ -42,19 +49,19 @@ def main(): case "import_candidats": if verbose: print(f"Import des candidats pour les élections {args.type} {args.year}") - import_candidats_2024.run(engine, verbose) + election_module.import_candidats.run(engine, verbose) case "import_resultats": if verbose: print(f"Import des résultats pour les élections {args.type} {args.year}") - import_resultats_2024.run(engine, verbose) + election_module.import_resultats.run(engine, verbose) case "import_resultats_web": if verbose: print(f"Import des résultats depuis le web pour les élections {args.type} {args.year}") - import_resultats_web_2024.run(engine, verbose) + election_module.import_resultats_web.run(engine, verbose) case "export_resultats": if verbose: print(f"Export des résultats pour les élections {args.type} {args.year}") - export_resultats_2024.run(engine, verbose) + election_module.export_resultats.run(engine, verbose) case _: print(f"Action {args.action} non reconnue") diff --git a/nupes/scripts/europeennes2024/__init__.py b/nupes/scripts/europeennes2024/__init__.py new file mode 100644 index 0000000..4d0dd39 --- /dev/null +++ b/nupes/scripts/europeennes2024/__init__.py @@ -0,0 +1 @@ +from . import export_resultats, import_candidats, import_resultats, import_resultats_web diff --git a/nupes/scripts/export_resultats_2024.py b/nupes/scripts/europeennes2024/export_resultats.py similarity index 100% rename from nupes/scripts/export_resultats_2024.py rename to nupes/scripts/europeennes2024/export_resultats.py diff --git a/nupes/scripts/import_candidats_2024.py b/nupes/scripts/europeennes2024/import_candidats.py similarity index 100% rename from nupes/scripts/import_candidats_2024.py rename to nupes/scripts/europeennes2024/import_candidats.py diff --git a/nupes/scripts/import_resultats_2024.py b/nupes/scripts/europeennes2024/import_resultats.py similarity index 100% rename from nupes/scripts/import_resultats_2024.py rename to nupes/scripts/europeennes2024/import_resultats.py diff --git a/nupes/scripts/import_resultats_web_2024.py b/nupes/scripts/europeennes2024/import_resultats_web.py similarity index 100% rename from nupes/scripts/import_resultats_web_2024.py rename to nupes/scripts/europeennes2024/import_resultats_web.py