log response in separate middleware

Author: copelco

nc/middleware.py

@@ -16,7 +16,19 @@ def __init__(self, get_response):
     def __call__(self, request: HttpRequest):
         headers = {"_type": "request", "_path": request.get_full_path(), "_now": now().isoformat()}
         headers.update(request.headers)
+        for header in ("wsgi.input", "wsgi.errors", "wsgi.file_wrapper"):
+            headers.pop(header, None)
         print(json.dumps(headers))
+        return self.get_response(request)
+
+
+class ResponseLoggingMiddleware:
+    """Log response headers"""
+
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request: HttpRequest):
         response: HttpResponse = self.get_response(request)
         headers = {"_type": "response", "_path": request.get_full_path(), "_now": now().isoformat()}
         headers.update(response.headers)

traffic_stops/settings/base.py

@@ -153,6 +153,7 @@ def __init__(self, tz_name=None):
 ]
 
 MIDDLEWARE = [
+    "nc.middleware.ResponseLoggingMiddleware",
     "django.middleware.security.SecurityMiddleware",
     "django.contrib.sessions.middleware.SessionMiddleware",
     "django.middleware.locale.LocaleMiddleware",

traffic_stops/settings/dev.py

@@ -28,7 +28,7 @@
 NC_AUTO_IMPORT_MONITORS = ("nc-monitor@example.com",)
 
 ALLOWED_HOSTS = os.getenv("ALLOWED_HOSTS", "")
-ALLOWED_HOSTS = ALLOWED_HOSTS.split(",") if ALLOWED_HOSTS else []
+ALLOWED_HOSTS = ALLOWED_HOSTS.split(",") if ALLOWED_HOSTS else ["*"]
 
 # Special test settings
 if "test" in sys.argv: