Limit ALLOWED HOSTS for CMS

appsembler · Jan 29, 2024 · 88217c3 · 88217c3
1 parent 196c141
commit 88217c3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cms/envs/production.py b/cms/envs/production.py
@@ -192,9 +192,9 @@ def get_env_setting(setting):
 SITE_NAME = ENV_TOKENS['SITE_NAME']
 
 ALLOWED_HOSTS = [
-    # TODO: bbeggs remove this before prod, temp fix to get load testing running
-    "*",
-    CMS_BASE,
+    ENV_TOKENS.get('LMS_BASE', ''),
+    ENV_TOKENS.get('CMS_BASE', ''),
+    FEATURES.get('PREVIEW_LMS_BASE', '')
 ]
 
 LOG_DIR = ENV_TOKENS['LOG_DIR']

diff --git a/lms/envs/production.py b/lms/envs/production.py
@@ -212,6 +212,7 @@ def get_env_setting(setting):
 
 ALLOWED_HOSTS = [
     ENV_TOKENS.get('LMS_BASE', ''),
+    ENV_TOKENS.get('CMS_BASE', ''),
     FEATURES.get('PREVIEW_LMS_BASE', '')
 ]