Allow custom configuration of additional hosts via ADDITIONAL_HOSTS e…

…nvironment variable.
i-dot-ai · Jul 16, 2024 · 27420d0 · 27420d0
1 parent 545c227
commit 27420d0
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/django_app/redbox_app/setting_enums.py b/django_app/redbox_app/setting_enums.py
@@ -1,5 +1,8 @@
+import os
 from enum import StrEnum
 
+ADDITIONAL_HOSTS = os.environ.get("ADDITIONAL_HOSTS", "").split(";")
+
 
 class Environment(StrEnum):
     def __new__(cls, value: str, is_test: bool, hosts=list[str]):
@@ -17,11 +20,11 @@ def is_local(self) -> bool:
     def uses_minio(self) -> bool:
         return self.is_test
 
-    LOCAL = ("LOCAL", True, ["localhost", "127.0.0.1", "0.0.0.0"])  # noqa: S104 nosec: B104: Not in prod
-    INTEGRATION = ("INTEGRATION", True, ["localhost", "127.0.0.1", "0.0.0.0"])  # noqa: S104 nosec: B104: Not in prod
-    DEV = ("DEV", False, ["redbox-dev.ai.cabinetoffice.gov.uk"])
-    PREPROD = ("PREPROD", False, ["redbox-preprod.ai.cabinetoffice.gov.uk"])
-    PROD = ("PROD", False, ["redbox.ai.cabinetoffice.gov.uk"])
+    LOCAL = ("LOCAL", True, ["localhost", "127.0.0.1", "0.0.0.0", *ADDITIONAL_HOSTS])  # noqa: S104 nosec: B104: Not in prod
+    INTEGRATION = ("INTEGRATION", True, ["localhost", "127.0.0.1", "0.0.0.0", *ADDITIONAL_HOSTS])  # noqa: S104 nosec: B104: Not in prod
+    DEV = ("DEV", False, ["redbox-dev.ai.cabinetoffice.gov.uk", *ADDITIONAL_HOSTS])
+    PREPROD = ("PREPROD", False, ["redbox-preprod.ai.cabinetoffice.gov.uk", *ADDITIONAL_HOSTS])
+    PROD = ("PROD", False, ["redbox.ai.cabinetoffice.gov.uk", *ADDITIONAL_HOSTS])
 
 
 class Classification(StrEnum):