From 66b8fb8103d6273ee4cd99ba79cb4e1cb0d55f2e Mon Sep 17 00:00:00 2001
From: Emmy D'Anello <ynerant@emy.lu>
Date: Tue, 18 Jun 2024 14:37:06 +0200
Subject: [PATCH] =?UTF-8?q?D=C3=A9placement=20des=20scripts?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py                                       | 21 ++++++++++++-------
 nupes/scripts/europeennes2024/__init__.py     |  1 +
 .../export_resultats.py}                      |  0
 .../import_candidats.py}                      |  0
 .../import_resultats.py}                      |  0
 .../import_resultats_web.py}                  |  0
 6 files changed, 15 insertions(+), 7 deletions(-)
 create mode 100644 nupes/scripts/europeennes2024/__init__.py
 rename nupes/scripts/{export_resultats_2024.py => europeennes2024/export_resultats.py} (100%)
 rename nupes/scripts/{import_candidats_2024.py => europeennes2024/import_candidats.py} (100%)
 rename nupes/scripts/{import_resultats_2024.py => europeennes2024/import_resultats.py} (100%)
 rename nupes/scripts/{import_resultats_web_2024.py => europeennes2024/import_resultats_web.py} (100%)

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