🎨 Format Python code with psf/black

DSGT-DLP · Feb 7, 2024 · 5f04b42 · 5f04b42
1 parent 3b8eabc
commit 5f04b42
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 11 deletions.
diff --git a/training/training/middleware/health_check_middleware.py b/training/training/middleware/health_check_middleware.py
@@ -7,6 +7,6 @@ def __init__(self, get_response):
         self.get_response = get_response
 
     def __call__(self, request):
-        if request.path == '/health':
-            return HttpResponse('ok')
-        return self.get_response(request)
+        if request.path == "/health":
+            return HttpResponse("ok")
+        return self.get_response(request)
diff --git a/training/training/settings.py b/training/training/settings.py
@@ -27,15 +27,17 @@
 
 import requests, os
 
-ALLOWED_HOSTS = ["backend-load-balancer-296304048.us-east-1.elb.amazonaws.com"]
-ELB_HEALTHCHECK_HOSTNAMES = [
-    ip
-    for network in requests.get(os.environ["ECS_CONTAINER_METADATA_URI"]).json()[
-        "Networks"
+ALLOWED_HOSTS = []
+if "ECS_CONTAINER_METADATA_URI" in os.environ:
+    ELB_HEALTHCHECK_HOSTNAMES = [
+        ip
+        for network in requests.get(os.environ["ECS_CONTAINER_METADATA_URI"]).json()[
+            "Networks"
+        ]
+        for ip in network["IPv4Addresses"]
     ]
-    for ip in network["IPv4Addresses"]
-]
-ALLOWED_HOSTS += ELB_HEALTHCHECK_HOSTNAMES
+    ALLOWED_HOSTS += ELB_HEALTHCHECK_HOSTNAMES
+    ALLOWED_HOSTS.append("backend-load-balancer-296304048.us-east-1.elb.amazonaws.com")
 
 # Application definition