Add constant for EONI import Scheme

DemocracyClub · Dec 11, 2024 · 291fefa · 291fefa
1 parent 656c96b
commit 291fefa
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/polling_stations/settings/constants/importers.py b/polling_stations/settings/constants/importers.py
@@ -1,8 +1,18 @@
 import os
+from django.db import models
+
 
 if SERVER_ENVIRONMENT := os.environ.get("DC_ENVIRONMENT"):
     S3_DATA_BUCKET = f"pollingstations.elections.{SERVER_ENVIRONMENT}"
 else:
     S3_DATA_BUCKET = os.environ.get(
         "S3_DATA_BUCKET", "pollingstations.elections.development"
     )
+
+
+class EONIImportScheme(models.TextChoices):
+    NATIONAL = ("NATIONAL", "National (NI Assembly and Westminster Elections)")
+    LOCAL = ("LOCAL", "Local (Council Elections)")
+
+
+EONI_IMPORT_SCHEME = EONIImportScheme.NATIONAL