Déplacement des scripts

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")
 

nupes/scripts/europeennes2024/__init__.py

@@ -0,0 +1 @@
+from . import export_resultats, import_candidats, import_resultats, import_resultats_web