Ajout données de géométrie administrative

main.py

@@ -6,7 +6,7 @@ import os
 from dotenv import load_dotenv
 from sqlalchemy import create_engine
 
-from nupes.scripts import import_candidats_2024
+from nupes.scripts import import_candidats_2024, import_geographie
 
 
 def parse_args():
@@ -15,7 +15,8 @@ def parse_args():
                         choices=["présidentielle", "législatives", "sénatoriales", "européennes",
                                  "régionales", "départementales", "municipales"], default="européennes")
     parser.add_argument('--year', '-y', type=int, help="Année de l'élection", default=2024)
-    parser.add_argument('action', help="Action à réaliser", choices=["import_candidats"])
+    parser.add_argument('action', help="Action à réaliser",
+                        choices=["import_candidats", "import_geographie"])
     parser.add_argument('--debug', '-d', action='store_true', help="Mode debug")
     return parser.parse_args()
 
@@ -28,6 +29,9 @@ def main():
     engine = create_engine(os.getenv("SQLALCHEMY_URL"), echo=args.debug)
 
     match args.action:
+        case "import_geographie":
+            print(f"Import des données géographiques pour les élections {args.type} {args.year}")
+            import_geographie.run(engine)
         case "import_candidats":
             print(f"Import des candidats pour les élections {args.type} {args.year}")
             import_candidats_2024.run(engine)