Merge pull request #1 from eillarra/patch-1

Add a "Vary" header for DNT, useful for caching.

dnt/middleware.py

@@ -1,8 +1,21 @@
+from django.utils.cache import patch_vary_headers
+
+
 class DoNotTrackMiddleware(object):
-    """Sets request.dnt to True or False based on the presence of the
-    DNT HTTP header."""
+
     def process_request(self, request):
+        """
+        Sets request.DNT to True or False based on the presence of the DNT HTTP header.
+        """
         if 'HTTP_DNT' in request.META and request.META['HTTP_DNT'] == '1':
             request.DNT = True
         else:
             request.DNT = False
+
+    def process_response(self, request, response):
+        """
+        Adds a "Vary" header for DNT, useful for caching.
+        """
+        patch_vary_headers(response, ['DNT'])
+
+        return response