Changes between Version 9 and Version 10 of StripWhitespaceMiddleware

Timestamp:

May 28, 2012, 3:40:32 AM (13 years ago)

Author:

anonymous

Comment:

changed for supporting conditional remove

StripWhitespaceMiddleware

@@ -19 +19 @@
 import re
 
-class StripWhitespaceMiddleware:
+class StripWhitespaceMiddleware(object):
     """
     Strips leading and trailing whitespace from response content.
@@ -26 +26 @@
     def __init__(self):
         self.whitespace = re.compile('^\s*\n', re.MULTILINE)
-        self.whitespace_lead = re.compile('^\s+', re.MULTILINE)
-        self.whitespace_trail = re.compile('\s+$', re.MULTILINE)
+        # self.whitespace_lead = re.compile('^\s+', re.MULTILINE)
+        # self.whitespace_trail = re.compile('\s+$', re.MULTILINE)
 
 
     def process_response(self, request, response):
         if "text" in response['Content-Type']:
-            new_content = self.whitespace_lead.sub('', response.content)
-            new_content = self.whitespace_trail.sub('\n', new_content)
+            if hasattr(self, 'whitespace_lead'):
+                response.content = self.whitespace_lead.sub('', response.content)
+            if hasattr(self, 'whitespace_trail'):
+                response.content = self.whitespace_trail.sub('\n', response.content)
             # Uncomment the next line to remove empty lines
-            # new_content = self.whitespace.sub('', new_content)
-            response.content = new_content
+            if hasattr(self, 'whitespace'):
+                response.content = self.whitespace.sub('', response.content)
             return response
         else:
-            return response
+            return response    
 
 }}}