[z3-checkins] r35755 - z3/deliverance/trunk/deliverance

[ianb at codespeak.net]

Author: ianb
Date: Thu Dec 14 17:44:21 2006
New Revision: 35755

Modified:
   z3/deliverance/trunk/deliverance/relocateresponse.py
Log:
update content-length in link rewriting code

Modified: z3/deliverance/trunk/deliverance/relocateresponse.py
==============================================================================
--- z3/deliverance/trunk/deliverance/relocateresponse.py	(original)
+++ z3/deliverance/trunk/deliverance/relocateresponse.py	Thu Dec 14 17:44:21 2006
@@ -5,7 +5,7 @@
 import urlparse
 import re
 from paste.request import construct_url
-from paste.response import header_value
+from paste.response import header_value, replace_header
 import fixuplinks
 
 def relocate_response(headers, content, base_href, old_href, new_href):
@@ -77,7 +77,6 @@
         app_iter = self.app(environ, repl_start_response)
         if skipped:
             return app_iter
-        start_response(*stat_headers)
         try:
             for chunk in app_iter:
                 written.append(chunk)
@@ -86,4 +85,7 @@
                 app_iter.close()
         content = ''.join(written)
         content = relocate_content(content, base_href, self.old_href, new_href)
+        headers = stat_headers[1]
+        replace_header(headers, 'content-length', str(len(content)))
+        start_response(*stat_headers)
         return [content]