Merge pull request #212 from sevdog/xaccel-headers

Allow XResponses to keep original headers provided to base response

Author: Natim

demo/demoproject/apache/tests.py

@@ -43,3 +43,19 @@ def test_response(self):
             basename="hello-world.txt",
             file_path="/apache-optimized-by-decorator/hello-world.txt",
         )
+
+
+class ModifiedHeadersTestCase(django.test.TestCase):
+    def test_response(self):
+        """'apache:modified_headers' returns X-Sendfile response."""
+        setup_file()
+        url = reverse("apache:modified_headers")
+        response = self.client.get(url)
+        assert_x_sendfile(
+            self,
+            response,
+            content_type="text/plain; charset=utf-8",
+            basename="hello-world.txt",
+            file_path="/apache-modified-headers/hello-world.txt",
+        )
+        self.assertEqual(response['X-Test'], 'header')

demo/demoproject/apache/urls.py

@@ -15,4 +15,9 @@
         views.optimized_by_decorator,
         name="optimized_by_decorator",
     ),
+    path(
+        "modified_headers/",
+        views.modified_headers,
+        name="modified_headers",
+    ),
 ]

demo/demoproject/apache/views.py

@@ -22,3 +22,17 @@
     source_url=storage.base_url,
     destination_dir="/apache-optimized-by-decorator/",
 )
+
+
+def _modified_headers(request):
+    view = StorageDownloadView.as_view(storage=storage, path="hello-world.txt")
+    response = view(request)
+    response["X-Test"] = 'header'
+    return response
+
+
+modified_headers = x_sendfile(
+    _modified_headers,
+    source_url=storage.base_url,
+    destination_dir="/apache-modified-headers/",
+)

demo/demoproject/lighttpd/tests.py

@@ -43,3 +43,19 @@ def test_response(self):
             basename="hello-world.txt",
             file_path="/lighttpd-optimized-by-decorator/hello-world.txt",
         )
+
+
+class ModifiedHeadersTestCase(django.test.TestCase):
+    def test_response(self):
+        """'lighttpd:modified_headers' returns X-Sendfile response."""
+        setup_file()
+        url = reverse("lighttpd:modified_headers")
+        response = self.client.get(url)
+        assert_x_sendfile(
+            self,
+            response,
+            content_type="text/plain; charset=utf-8",
+            basename="hello-world.txt",
+            file_path="/lighttpd-modified-headers/hello-world.txt",
+        )
+        self.assertEqual(response['X-Test'], 'header')

demo/demoproject/lighttpd/urls.py

@@ -15,4 +15,9 @@
         views.optimized_by_decorator,
         name="optimized_by_decorator",
     ),
+    path(
+        "modified_headers/",
+        views.modified_headers,
+        name="modified_headers",
+    ),
 ]

demo/demoproject/lighttpd/views.py

@@ -22,3 +22,17 @@
     source_url=storage.base_url,
     destination_dir="/lighttpd-optimized-by-decorator/",
 )
+
+
+def _modified_headers(request):
+    view = StorageDownloadView.as_view(storage=storage, path="hello-world.txt")
+    response = view(request)
+    response["X-Test"] = 'header'
+    return response
+
+
+modified_headers = x_sendfile(
+    _modified_headers,
+    source_url=storage.base_url,
+    destination_dir="/lighttpd-modified-headers/",
+)

demo/demoproject/nginx/tests.py

@@ -51,3 +51,23 @@ def test_response(self):
             with_buffering=None,
             limit_rate=None,
         )
+
+
+class ModifiedHeadersTestCase(django.test.TestCase):
+    def test_response(self):
+        """'nginx:modified_headers' returns X-Sendfile response."""
+        setup_file()
+        url = reverse("nginx:modified_headers")
+        response = self.client.get(url)
+        assert_x_accel_redirect(
+            self,
+            response,
+            content_type="text/plain; charset=utf-8",
+            charset="utf-8",
+            basename="hello-world.txt",
+            redirect_url="/nginx-modified-headers/hello-world.txt",
+            expires=None,
+            with_buffering=None,
+            limit_rate=None,
+        )
+        self.assertEqual(response['X-Test'], 'header')

demo/demoproject/nginx/urls.py

@@ -16,4 +16,9 @@
         views.optimized_by_decorator,
         name="optimized_by_decorator",
     ),
+    path(
+        "modified_headers/",
+        views.modified_headers,
+        name="modified_headers",
+    ),
 ]

demo/demoproject/nginx/views.py

@@ -22,3 +22,17 @@
     source_url=storage.base_url,
     destination_url="/nginx-optimized-by-decorator/",
 )
+
+
+def _modified_headers(request):
+    view = StorageDownloadView.as_view(storage=storage, path="hello-world.txt")
+    response = view(request)
+    response["X-Test"] = 'header'
+    return response
+
+
+modified_headers = x_accel_redirect(
+    _modified_headers,
+    source_url=storage.base_url,
+    destination_url="/nginx-modified-headers/",
+)

django_downloadview/apache/middlewares.py

@@ -32,4 +32,5 @@ def process_download_response(self, request, response):
             content_type=response["Content-Type"],
             basename=response.basename,
             attachment=response.attachment,
+            headers=response.headers,
         )