Add 'patch_vary_headers' to middleware.py

mishbahr · Sep 28, 2014 · 7fdc37a · 7fdc37a
1 parent f3af95f
commit 7fdc37a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/responsive/middleware.py b/responsive/middleware.py
@@ -4,6 +4,7 @@
 import re
 
 from django.template.loader import render_to_string
+from django.utils.cache import patch_vary_headers
 from django.utils.encoding import force_text
 
 from .conf import settings
@@ -56,6 +57,8 @@ def process_response(self, request, response):
             pattern = re.compile(b'<head>', re.IGNORECASE)
             response.content = pattern.sub(b'<head>' + snippet, force_text(response.content))
 
-        if response.get('Content-Length', None):
-            response['Content-Length'] = len(response.content)
+            if response.get('Content-Length', None):
+                response['Content-Length'] = len(response.content)
+
+        patch_vary_headers(response, ('Cookie', ))
         return response